wasm-mt

文档 | GitHub | Crate

Rust 和 WebAssembly 的多线程库。

wasm-mt 帮助您创建和执行基于 Web Worker 的线程。您可以使用 Rust 闭包简单编程线程，并使用 async/await 进行编排。

示例

• wasm-mt-pool - 基于 wasm-mt 的线程池库。 [ crate | 源码 ]

您可以在浏览器中运行以下所有应用！

• exec - 如何使用 wasm_mt。 [ 实时 | 源码 ]
• fib - 使用嵌套线程计算斐波那契数列。 [ 实时 | 源码 ]
• executors - 使用 wasm_mt 的最小串行/并行执行器。 [ 实时 | 源码 ]
• parallel - 串行/并行执行器的 Julia 集基准测试。 [ 实时 | 源码 ]
• arraybuffers - 使用 WasmMt::new_with_arraybuffers() 的演示。 [ 实时 | 源码 ]

背景和实现

前一篇名为《"Rust和Wasm的多线程编程"》的开创性作品由@alexcrichton撰写，主要围绕Web Workers、共享内存和WebAssembly线程提案展开。共享内存建立在SharedArrayBuffer之上，但其在主要浏览器中的可用性一直受到限制。此外，rust-wasm线程实现工作以及线程提案似乎仍在进行中。

相反，基于Web Worker的JavaScript多线程长期以来一直得到了良好的支持。经过实验，我们提出了一种Rust易用多线程解决方案，它不依赖于SharedArrayBuffer。如今，它可以在所有主要浏览器中正常工作，我们将其命名为wasm-mt。

在内部，我们使用postMessage Web Worker API（通过wasm-bindgen提供的绑定）来初始化创建的线程。并且，重要的是，我们继续使用postMessage来动态发送Rust闭包（通过serde_traitobject序列化）到创建的线程。通过这种方式，父线程可以等待在创建的线程中执行的闭包的结果。我们发现这种方法对于扩展也非常灵活。例如，将WasmMt::Thread扩展以支持更多自定义的线程间通信模式非常简单。

然而，需要注意的是，与由wasm-bindgen领导的基于共享内存的多线程工作相比，wasm-mt存在一些显著的局限性。与共享内存的多线程工作相比，wasm-mt在效率上并不高，因为它不包含对标准线程原语操作的支持

• 基于共享内存的消息传递和互斥锁，
• 原子指令和按线程提案的内存管理。

谢谢

• wasm-bindgen的开发者
• @alecmocatta为serde-traitobject crate
• swc-project，它简化了wasm-mt-test crate的使用

入门

要求

• rustc (nightly)
• wasm-pack build带有--target no-modules选项

Cargo.toml

wasm-mt = "0.1"
serde = { version = "1.0", features = ["derive"] }
serde_closure = "0.3"

创建线程

首先，使用new创建一个WasmMt线程构建器并初始化它

use wasm_mt::prelude::*;

let pkg_js = "./pkg/exec.js"; // path to `wasm-bindgen`'s JS binding
let mt = WasmMt::new(pkg_js).and_init().await.unwrap();

然后，使用thread函数创建一个wasm_mt::Thread并初始化它

let th = mt.thread().and_init().await.unwrap();

执行线程

使用 exec! 宏，您可以在线程中执行闭包并 await 结果

// fn add(a: i32, b: i32) -> i32 { a + b }

let a = 1;
let b = 2;
let ans = exec!(th, move || {
    let c = add(a, b);

    Ok(JsValue::from(c))
}).await?;
assert_eq!(ans, JsValue::from(3));

您还可以使用 exec! 来执行 异步闭包

// use wasm_mt::utils::sleep;
// async fn sub(a: i32, b: i32) -> i32 {
//    sleep(1000).await;
//    a - b
// }

let a = 1;
let b = 2;
let ans = exec!(th, async move || {
    let c = sub(a, b).await;

    Ok(JsValue::from(c))
}).await?;
assert_eq!(ans, JsValue::from(-1));

在线程中执行 JavaScript

使用 exec_js! 宏，您可以在线程中执行 JavaScript

let ans = exec_js!(th, "
    const add = (a, b) => a + b;
    return add(1, 2);
").await?;
assert_eq!(ans, JsValue::from(3));

类似地，使用 exec_js_async! 来运行异步 JavaScript

let ans = exec_js_async!(th, "
    const sub = (a, b) => new Promise(resolve => {
        setTimeout(() => resolve(a - b), 1000);
    });
    return await sub(1, 2);
").await?;
assert_eq!(ans, JsValue::from(-1));

创建执行器

通过使用 [Vec<wasm_mt::Thread>][Thread]，您可以轻松创建自定义执行器。一个这样的例子是 wasm-mt-pool 包。它提供了一个基于 工作窃取 调度策略的 线程池。

在这里，为了简单起见，我们展示了更直接的执行器的实现：顺序执行器和并行执行器。

首先，准备一个包含初始化线程的 Vec<wasm_mt::Thread>

let mut v: Vec<wasm_mt::Thread> = vec![];
for i in 0..4 {
    let th = mt.thread().and_init().await?;
    v.push(th);
}

console_ln!("🚀 serial executor:");
for th in &v {
    console_ln!("starting a thread");
    let ans = exec!(th, move || Ok(JsValue::from(42))).await?;
    console_ln!("ans: {:?}", ans);
}

console_ln!("🚀 parallel executor:");
for th in v {
    spawn_local(async move {
        console_ln!("starting a thread");
        let ans = exec!(th, move || Ok(JsValue::from(42))).await.unwrap();
        console_ln!("ans: {:?}", ans);
    });
}

🚀 serial executor:
starting a thread
ans: JsValue(42)
starting a thread
ans: JsValue(42)
starting a thread
ans: JsValue(42)
starting a thread
ans: JsValue(42)
🚀 parallel executor:
(4) starting a thread
(4) ans: JsValue(42)