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() 的演示。[ 在线 | 源代码 ]

背景和实现

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

相反，基于 Web Worker 的 JavaScript 多线程已经得到了长期的良好支持。经过实验，我们提出了一种 Rust 人体工程学多线程解决方案，无需 SharedArrayBuffer。它今天可以在所有主流浏览器上运行，我们将其命名为 wasm-mt。

内部，我们使用 postMessage() Web Worker API（通过 wasm-bindgen 提供的绑定）来初始化生成的线程。而且，更重要的是，我们继续使用 postMessage() 来动态地将 Rust 闭包（由 serde_traitobject 序列化）发送到生成的线程。通过这种方式，父线程可以 await 在生成的线程中执行的闭包的结果。我们发现这种方法在扩展性方面也非常灵活。例如，将 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"

创建线程

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

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][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));

创建执行器

通过使用 [wasm_mt:Thread][Thread]，你可以轻松创建自定义执行器。一个例子是 wasm-mt-pool crate。它提供了一个基于 work stealing 调度策略的 线程池。

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

首先，准备一个包含初始化线程的 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);
}

然后，这里是执行器的实际操作。注意，在后一种情况下，我们使用 wasm_bindgen_futures::spawn_local 来并行调度线程。

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)

lib.rs:

使用 wasm-mt-pool 测试 crate 的实用工具。