Rust wasm-bindgen开发中如何安全实现WASM全局状态共享

WASM 默认运行在单线程执行环境中，不需要使用 unsafe 裸指针，也不需要强制让自定义结构体实现 Send/Sync 或序列化 trait，使用标准库的 thread_local! 配合 RefCell、Option 即可实现完全安全的全局状态管理，同时覆盖未初始化、重复初始化等边界场景。

你当前使用的 unsafe 裸指针实现存在几个明确隐患：

• 重复调用 initialize_state 时，之前通过 Box::leak 创建的 State 实例会永久泄漏，无法被回收
• 未执行初始化就调用读写方法会直接触发空指针解引用，导致WASM实例无提示崩溃
• 裸指针操作绕开了Rust的编译期安全检查，后续代码迭代很容易引入难以排查的内存错误

可直接使用的安全实现代码 #

use wasm_bindgen::prelude::*;
use std::cell::RefCell;

thread_local! {
    // Option标记初始化状态，RefCell提供单线程场景下的内部可变性
    static STATE: RefCell<Option<State>> = RefCell::new(None);
}

#[wasm_bindgen]
pub fn initialize_state(value: i32) {
    STATE.with(|state| {
        // 重复初始化时旧State实例会被自动drop，无内存泄漏
        *state.borrow_mut() = Some(State { a_value: value });
    });
}

#[wasm_bindgen]
pub fn increment_value() {
    STATE.with(|state| {
        let mut state_guard = state.borrow_mut();
        let target_state = state_guard.as_mut().expect("调用方法前请先执行initialize_state初始化");
        target_state.a_value += 1;
    });
}

#[wasm_bindgen]
pub fn get_value() -> i32 {
    STATE.with(|state| {
        let state_guard = state.borrow();
        let target_state = state_guard.as_ref().expect("调用方法前请先执行initialize_state初始化");
        target_state.a_value
    })
}

// 自定义结构体不需要实现任何额外trait，即使不满足线程安全、无法序列化也可正常使用
struct State {
    a_value: i32,
}

方案特性 #

• 整段实现无任何unsafe代码，所有内存安全规则由Rust编译器在编译期保证
• 适配非线程安全结构体：thread_local! 存储的变量仅在当前WASM线程内访问，不存在跨线程共享要求，不需要结构体实现Send/Sync trait
• 无内存泄漏：所有State实例的生命周期受Rust所有权系统管理，重复初始化、重置状态时旧实例会被自动回收
• 错误提示明确：未初始化调用读写方法时会抛出明确的panic信息，不会出现无征兆的内存崩溃
• 性能开销极低：单线程场景下RefCell的运行时检查开销可以忽略，远低于锁实现的开销

如果后续需要开启WASM多线程支持，只需要将RefCell<Option<State>>替换为Mutex<Option<State>>，配合Arc使用即可，整体逻辑结构不需要调整。

内容的提问来源于stack exchange，提问作者CoolElectronics