关于Tokio多线程运行时任务与OS线程绑定的技术问询

嘿，这个问题问到点子上了，刚好是Tokio调度机制里大家常纠结的点，我给你梳理清楚：

首先回答第一个问题：默认情况下，Tokio多线程运行时的任务不会固定绑定到某一个OS线程。

Tokio的多线程调度器用的是「工作窃取」（work-stealing）模式：每个worker线程都有自己的任务队列，当某个线程的队列里没有可执行的任务（比如所有任务都在await IO、定时器这类异步操作），它就会去其他线程的队列里“偷”任务来运行。哪怕是同一个任务，在不同的poll阶段（比如await结束被唤醒后），也可能被调度到不同的OS线程上——毕竟异步future的poll是可中断的，每次唤醒后调度器会把它放到任意空闲的worker线程队列里。

然后是你补充的问题：完全可以配置让任务只在单个OS线程上运行，有几种常用方式：

• 用tokio::task::spawn_local替代普通的tokio::spawn。这个方法创建的任务会被绑定到创建它的那个worker线程，不会被其他线程窃取调度。不过要注意，spawn_local必须在当前线程的Tokio runtime上下文里调用，而且你的runtime得是多线程模式（单线程runtime本来就只有一个线程，自然不用绑定）。
• 创建独立的单线程runtime来运行任务。如果你的需求是某一批任务完全在单个OS线程里执行，可以单独起一个单线程runtime，所有任务都在这个runtime里运行：

use tokio::runtime::Builder;

fn main() -> Result<(), Box<dyn std::error::Error>> {
    let single_thread_rt = Builder::new_current_thread()
        .enable_all()
        .build()?;

    single_thread_rt.block_on(async {
        // 这里面的所有任务都会固定在这个单线程runtime对应的OS线程上
        tokio::task::spawn_local(async {
            // 任务逻辑
        }).await?;
        Ok(())
    })
}

• 用tokio::task::LocalSet来管理一组绑定线程的任务。如果你需要在多线程runtime里，让某一部分任务固定在单个线程，可以用LocalSet把它们包起来，然后在某个worker线程上运行这个LocalSet：

use tokio::task::LocalSet;

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let local_set = LocalSet::new();

    local_set.run_until(async {
        // 这里面用spawn_local创建的任务，都会绑定到运行这个LocalSet的worker线程
        tokio::task::spawn_local(async {
            // 任务逻辑
        }).await?;
        Ok(())
    }).await
}

备注：内容来源于stack exchange，提问作者ababo