Neon 项目中共享 Rust 变量的几种实现方式

在开发 Node.js 原生模块时，我们经常需要在多个 JavaScript 调用之间共享 Rust 变量。本文将介绍在 Neon 项目中实现这一需求的几种方法，帮助开发者根据具体场景选择最合适的方案。

全局共享变量（OnceLock）

当我们需要在整个进程生命周期内共享一个单例变量时，可以使用 Rust 标准库中的 OnceLock。这种方法适用于那些初始化成本高且不需要多实例的场景，比如数据库连接池或运行时环境。

use std::sync::OnceLock;

fn global_xclient() -> &'static XClient {
    static GLOBAL: OnceLock<XClient> = OnceLock::new();
    
    GLOBAL.get_or_init(|| {
        let config = XConfig::default();
        XClient::new("http://xxxx", config).unwrap()
    })
}

使用时只需在需要的地方调用 global_xclient() 函数即可获取共享实例。注意初始化代码必须放在 get_or_init 的闭包内，确保只执行一次。

线程局部存储（LocalKey）

对于需要在不同 JavaScript 线程（如 Web Worker）中保持独立实例的场景，Neon 提供了 LocalKey 机制。这种方式类似于 OnceLock，但会为每个 JavaScript 线程维护独立的变量副本。

闭包捕获变量

当我们需要在导出的 JavaScript 函数中访问特定变量时，可以使用闭包捕获的方式：

#[neon::main]
fn main(mut cx: ModuleContext) -> NeonResult<()> {
    let shared_data = Arc::new(String::from("共享数据"));
    
    cx.export_function("getData", {
        let data = shared_data.clone();
        move |mut cx| Ok(cx.string(&data))
    })?;
    
    Ok(())
}

这种方法利用了 Rust 的所有权机制，通过 Arc 实现线程安全的引用计数，适合需要传递复杂数据结构的情况。

JavaScript 对象封装（JsBox）

当需要在 JavaScript 和 Rust 之间传递复杂对象时，可以使用 JsBox 将 Rust 结构体封装为 JavaScript 对象：

#[neon::main]
fn main(mut cx: ModuleContext) -> NeonResult<()> {
    cx.export_function("createClient", |mut cx| {
        let config = XConfig::default();
        let client = XClient::new("http://xxxx", config)?;
        Ok(cx.boxed(client))
    })?;
    
    Ok(())
}

这种方式允许 JavaScript 代码持有 Rust 对象的引用，并通过其他函数调用操作这些对象，非常适合面向对象的编程模式。

性能与安全性考虑

1. OnceLock 提供了最佳的性能，但牺牲了灵活性
2. 闭包捕获 在灵活性和性能之间取得了平衡
3. JsBox 提供了最大的灵活性，但引入了额外的内存管理开销
4. 在多线程环境下，必须确保共享变量的线程安全性

开发者应根据具体需求选择最合适的方案，平衡性能、灵活性和代码复杂度。