如何在Rust的move闭包外部访问闭包内生成的数据？

原始代码 #

let stream = device.build_input_stream(
    &config.into(),
    move |data, _: &_| {
        let b = write_input_data::<f32, f32>(data);
        println!("b: {}", b.len());
    },
    |err| {
        eprintln!("an error occurred on stream: {}", err);
    },
    None
).unwrap();

fn write_input_data<T, U>(input: &[T]) -> Vec<U>
where
    T: Sample,
    U: Sample + From<T> + hound::Sample + std::fmt::Debug,
{
    let mut buffer = Vec::new();
    for sample in input.iter() {
        buffer.push(U::from(*sample));
    }

    return buffer;
}

需求与问题 #

希望获取write_input_data生成的所有数据，但因回调闭包使用move关键字，无法在main函数中维护并访问一个共享向量来存储这些数据。

已尝试操作 #

• 移除move关键字
• 在main函数中创建向量，尝试将闭包内的b数据推入其中
• 向write_input_data传递引用（因move转移所有权而无效）

解决方案：可行，使用线程安全的共享容器 #

音频流的回调函数运行在独立后台线程，因此需要用线程安全的共享所有权容器跨线程传递数据，可通过Arc<Mutex<Vec<U>>>实现：

修改后的代码示例 #

use std::sync::{Arc, Mutex};
// 假设已导入所需的Sample等 trait

fn main() {
    // 创建线程安全的共享向量，用于存储所有采样数据
    let collected_data = Arc::new(Mutex::new(Vec::<f32>::new()));
    // 克隆Arc，将所有权转移到闭包中
    let data_clone = collected_data.clone();

    let stream = device.build_input_stream(
        &config.into(),
        move |data, _: &_| {
            let b = write_input_data::<f32, f32>(data);
            // 锁定Mutex，将新生成的数据追加到共享向量中
            if let Ok(mut data) = data_clone.lock() {
                data.extend(b);
            }
            println!("本次收集数据长度: {}", b.len());
        },
        |err| {
            eprintln!("an error occurred on stream: {}", err);
        },
        None
    ).unwrap();

    // 启动音频流（根据所用音频库API，可能需要调用stream.play()）
    stream.play().unwrap();

    // 示例：在main线程中读取收集到的数据
    std::thread::sleep(std::time::Duration::from_secs(5));
    if let Ok(data) = collected_data.lock() {
        println!("总收集数据长度: {}", data.len());
        // 此处可对data进行后续处理，比如写入文件
    }
}

fn write_input_data<T, U>(input: &[T]) -> Vec<U>
where
    T: Sample,
    U: Sample + From<T> + hound::Sample + std::fmt::Debug,
{
    let mut buffer = Vec::new();
    for sample in input.iter() {
        buffer.push(U::from(*sample));
    }
    buffer
}

原理说明 #

1. Arc：提供线程安全的共享所有权，多个线程可持有同一Arc实例，内部引用计数自动管理内存。
2. Mutex：保证同一时间仅一个线程能访问内部向量，避免多线程并发写入导致的数据竞争。
3. 克隆Arc并move进闭包：闭包获得共享容器的所有权，同时main线程仍持有原Arc，可随时锁定Mutex读取数据。

之前尝试无效的原因 #

• 移除move关键字：闭包会捕获环境变量的引用，但回调函数生命周期由音频库管理，可能长于main函数中的变量，编译器会因生命周期不匹配报错。
• 直接传递向量引用：普通引用无法跨线程安全使用，且同样存在生命周期问题，编译器会拒绝这种不安全操作。

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