Rust如何将函数参数传递给闭包？线程逃逸问题求解

问题描述 #

编写了一个启动双线程的函数，其中一个线程闭包使用testdatapath参数时触发Borrowed data escapes outside of closure错误。核心问题是借用的参数生命周期仅覆盖函数体，但线程需要参数生命周期覆盖整个线程运行周期，导致数据逃逸。尝试克隆变量未解决问题，寻求正确实现方式。

错误代码：

pub fn measure_stats(testdatapath: &PathBuf, filenameprefix: &String) {
    let (tx, rx) = mpsc::channel();
   
    let filename = format!("test.txt");
    let measure_thread = thread::spawn(move || {
        let stats = sar();
        fs::write(filename, stats).expect("failed to write output to file");

        // Send a signal that we're done.
        let _ = tx.send(());
    });

    thread::spawn(move || {
        let mut n = 0;
        loop {
            // Break if the measure thread is done.
            match rx.try_recv() {
                Ok(_) | Err(TryRecvError::Disconnected) => break,
                Err(TryRecvError::Empty) => {}
            }
            let filename = format!("{:04}.img", n);
            let filepath = Path::new(testdatapath).join(&filename);
            random_file_write(&filepath).unwrap();
            random_file_read(&filepath).unwrap();
            fs::remove_file(&filepath).expect("failed to remove file");
            n += 1;
        }
    });
    measure_thread.join().expect("joining measure thread panicked");
}

解决方案 #

1. 正确转移PathBuf所有权到线程 #

之前克隆无效是因为仅克隆了引用，而非底层PathBuf实例。需要克隆PathBuf并将所有权转移给线程闭包：

pub fn measure_stats(testdatapath: &PathBuf, filenameprefix: &String) {
    let (tx, rx) = mpsc::channel();
    let filename = format!("test.txt");
    let measure_thread = thread::spawn(move || {
        let stats = sar();
        fs::write(filename, stats).expect("failed to write output to file");
        let _ = tx.send(());
    });

    // 克隆PathBuf获取所有权，转移给线程闭包
    let testdatapath_clone = testdatapath.clone();
    thread::spawn(move || {
        let mut n = 0;
        loop {
            match rx.try_recv() {
                Ok(_) | Err(TryRecvError::Disconnected) => break,
                Err(TryRecvError::Empty) => {}
            }
            let filename = format!("{:04}.img", n);
            // 使用克隆后的PathBuf，无需借用
            let filepath = testdatapath_clone.join(&filename);
            random_file_write(&filepath).unwrap();
            random_file_read(&filepath).unwrap();
            fs::remove_file(&filepath).expect("failed to remove file");
            n += 1;
        }
    });
    measure_thread.join().expect("joining measure thread panicked");
}

如果调用方可以转移所有权，也可以直接让函数接收PathBuf而非引用，省去克隆步骤：

pub fn measure_stats(testdatapath: PathBuf, filenameprefix: &String) {
    // 后续代码直接使用testdatapath即可
}

2. 优化线程同步逻辑 #

当前使用try_recv轮询效率较低，推荐改用阻塞等待或更高效的同步方式：

pub fn measure_stats(testdatapath: &PathBuf, filenameprefix: &String) {
    let (tx, rx) = mpsc::channel();
    let filename = format!("test.txt");
    let measure_thread = thread::spawn(move || {
        let stats = sar();
        fs::write(filename, stats).expect("failed to write output to file");
        tx.send(()).expect("failed to send completion signal");
    });

    let testdatapath_clone = testdatapath.clone();
    let io_thread = thread::spawn(move || {
        let mut n = 0;
        // 阻塞等待完成信号，无需轮询
        while rx.try_recv().is_err() {
            let filename = format!("{:04}.img", n);
            let filepath = testdatapath_clone.join(&filename);
            random_file_write(&filepath).unwrap();
            random_file_read(&filepath).unwrap();
            fs::remove_file(&filepath).expect("failed to remove file");
            n += 1;
        }
    });

    measure_thread.join().expect("joining measure thread panicked");
    // 不要忘记等待IO线程结束
    io_thread.join().expect("joining IO thread panicked");
}

3. 关于filenameprefix的处理 #

如果后续需要在线程中使用filenameprefix，同样遵循所有权转移原则：要么让函数接收String所有权，要么克隆filenameprefix后转移给闭包，避免借用逃逸。

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