如何用Tokio将依赖局部变量的CPU密集任务从异步函数分离？

问题根源 #

你的代码中heavy_computation是阻塞函数，直接在Tokio异步任务中调用会占用Tokio的工作线程。Tokio的多线程运行时依赖工作线程调度异步任务，一旦工作线程被阻塞，其他任务（比如task2）就无法获得调度资源，导致整个运行时卡住。

你之前尝试的方案无效，主要有两个原因：

1. 使用了标准库的同步通道（std::sync::mpsc），这类通道没有异步await方法，无法在Tokio异步任务中等待结果。
2. 嵌套tokio::spawn完全多余，反而会让代码逻辑混乱，无法正确传递计算结果。

正确解决方案（无需克隆局部变量） #

利用std::thread::scope安全借用栈上变量，搭配Tokio的异步oneshot通道传递结果，将阻塞任务放到独立系统线程中执行，不占用Tokio工作线程：

use tokio::sync::oneshot;

fn func() {
    let task1 = tokio::spawn(async move {
        let local_variable = ...; // 栈上的局部变量，无需克隆

        // 创建一次性异步通道，用于传递计算结果
        let (sender, receiver) = oneshot::channel();

        // 使用thread::scope创建临时线程，安全借用local_variable
        std::thread::scope(|s| {
            s.spawn(move || {
                // 在独立线程中执行阻塞计算
                let res = heavy_computation(&local_variable);
                // 发送结果，忽略发送失败（比如接收方提前销毁）
                let _ = sender.send(res);
            });
        });

        // 异步等待计算结果，不阻塞Tokio工作线程
        let res = receiver.await.expect("计算线程未正常发送结果");
        // 执行异步IO操作
        io(res).await;
    });

    let task2 = tokio::spawn(...);
}

方案说明 #

1. std::thread::scope的作用：它会保证所有在scope内创建的线程，在scope结束前全部完成，因此可以安全地借用当前栈上的local_variable，无需克隆或使用Arc。
2. Tokio异步通道：oneshot::channel是专门用于一次性结果传递的异步通道，支持await等待，完美适配Tokio的异步模型。
3. 隔离阻塞任务：heavy_computation运行在独立的系统线程中，完全不占用Tokio的工作线程，task2可以正常被调度执行。

针对CPU密集型任务的替代方案 #

如果heavy_computation是CPU密集型任务，推荐使用Rayon的并行线程池搭配rayon::scope，同样无需克隆变量：

use tokio::sync::oneshot;
use rayon::scope as rayon_scope;

fn func() {
    let task1 = tokio::spawn(async move {
        let local_variable = ...;

        let (sender, receiver) = oneshot::channel();

        // 使用Rayon的scope执行CPU密集型任务
        rayon_scope(|s| {
            s.spawn(move |_| {
                let res = heavy_computation(&local_variable);
                let _ = sender.send(res);
            });
        });

        let res = receiver.await.expect("计算线程未正常发送结果");
        io(res).await;
    });

    let task2 = tokio::spawn(...);
}

Rayon会自动管理线程池，适合处理大量CPU密集型任务，比手动创建临时线程更高效。

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