rust scoped thread

原创

寻找繁星 2022-10-14 15:05:24 博主文章分类：rust ©著作权

©著作权归作者所有：来自51CTO博客作者寻找繁星的原创作品，请联系作者获取转载授权，否则将追究法律责任

rust里，std::thread::spawn要求传入的闭包的生命周期必须是'static的，也就是说闭包中不能借用局部变量，因为rust不知道这个局部变量会不会突然就被释放了。但是一般情况下，spawn之后会调用join来确保函数返回前子线程会终止，也就是说实际上可以保证子线程在跑的时候，这个局部变量一直是有效的。

这种情况下怎么办呢？这就需要用到第三方crate：crossbeam。其中引入了scoped thread的概念，即通过强制join来确保子线程的生命周期局限在一个范围里，那么只要局部变量的生命周期比这个范围大，那子线程就可以借用这个局部变量了。

例子：

use crossbeam;
use std::sync::Mutex;

fn main() {
    let mut data = 233;
    crossbeam::scope(|s| {
        s.spawn(|_| {
            data += 1;
        });
    }).unwrap();
    println!("{}", data);
}

输出：

这里只可变借用（mutable borrow）了一次，所以编译可以通过。但是如果可变借用多次呢？

use crossbeam;
use std::sync::Mutex;

fn main() {
    let mut data = 233;
    crossbeam::scope(|s| {
        s.spawn(|_| {
            data += 1;
        });
        s.spawn(|_| {
            data += 1;
        });
    }).unwrap();
    println!("{}", data);
}

这时就会编译错误：

error[E0499]: cannot borrow `data` as mutable more than once at a time
  --> src/main.rs:18:17
   |
15 |         s.spawn(|_| {
   |                 --- first mutable borrow occurs here
16 |             data += 1;
   |             ---- first borrow occurs due to use of `data` in closure
17 |         });
18 |         s.spawn(|_| {
   |           ----- ^^^ second mutable borrow occurs here
   |           |
   |           first borrow later used by call
19 |             data += 1;
   |             ---- second borrow occurs due to use of `data` in closure

error: aborting due to previous error

For more information about this error, try `rustc --explain E0499`.
error: could not compile `scoped_thread`

To learn more, run the command again with --verbose.

要解决这个问题，就需要用到std::sync::Mutex了。

use crossbeam;
use std::sync::Mutex;

fn main() {
    let data = Mutex::new(233);
    crossbeam::scope(|s| {
        s.spawn(|_| {
            *data.lock().unwrap() += 1;
        });
        s.spawn(|_| {
            *data.lock().unwrap() += 1;
        });
    }).unwrap();
    println!("{}", *data.lock().unwrap());
}