Rust中如何向多线程按值传递worker id？编译错误解决

问题原因 #

编译错误的核心是循环变量i被闭包借用，Rust编译器无法保证i的生命周期能覆盖线程执行周期。虽然thread::scope会等待所有线程结束，但编译器需要明确的所有权转移才能确认安全性。另外，原代码中的PtrWrapper未实现Copy/Clone，导致多线程无法同时访问data_ptr。

解决方案 #

1. 为PtrWrapper实现Copy和Clone #

PtrWrapper仅包含一个裸指针（本身实现Copy），直接派生这两个trait即可让每个线程获取data_ptr的独立副本：

#[derive(Copy, Clone)]
struct PtrWrapper(*const u8);

unsafe impl Sync for PtrWrapper {}
unsafe impl Send for PtrWrapper {}

2. 使用move闭包捕获Worker ID #

给scope.spawn的闭包添加move关键字，每个闭包会复制并获取当前迭代的i的所有权（usize是Copy类型，无所有权转移风险），彻底解决生命周期冲突：

if bytes_per_worker >= page_size {
    thread::scope(|scope| {
        for i in 0..n_workers {
            // 添加move关键字，让闭包获取当前i的所有权
            scope.spawn(move || {
                let i_first = bytes_per_worker * i;
                let i_limit = cmp::min(total_bytes, bytes_per_worker * (i+1));
                for j in (i_first..i_limit).step_by(page_size) {
                    unsafe {
                        std::ptr::read_volatile(data_ptr.0.offset(j.try_into().unwrap()));
                    }
                }
            });
        }
    });
}

完整修改后的代码 #

#[derive(Copy, Clone)]
struct PtrWrapper(*const u8);

unsafe impl Sync for PtrWrapper {}
unsafe impl Send for PtrWrapper {}

// ...

let data = &mmap[begin_offset..end_offset];

let n_workers = 32;
let total_bytes = data.len();
let bytes_per_worker = (total_bytes + n_workers - 1) / n_workers;
let page_size = 512;
let data_ptr = PtrWrapper(data.as_ptr());
if bytes_per_worker >= page_size {
    thread::scope(|scope| {
        for i in 0..n_workers {
            scope.spawn(move || {
                let i_first = bytes_per_worker * i;
                let i_limit = cmp::min(total_bytes, bytes_per_worker * (i+1));
                for j in (i_first..i_limit).step_by(page_size) {
                    unsafe {
                        std::ptr::read_volatile(data_ptr.0.offset(j.try_into().unwrap()));
                    }
                }
            });
        }
    });
}

补充说明 #

• thread::scope是Rust 1.63+引入的安全多线程工具，会等待所有派生线程完成，因此可安全借用栈上变量，无需额外同步原语（如Arc）。
• 为PtrWrapper实现Copy/Clone是关键，确保每个线程拥有独立的data_ptr副本，避免所有权冲突。

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