tokio中loop+select!与spawn双任务的差异及优势对比

问题 #

我是异步编程新手，对不同实现方式的行为差异存在困惑。以tokio官方GitHub仓库中的chat.rs示例为例：

// snip
    loop {
        tokio::select! {
            // A message was received from a peer. Send it to the current user.
            Some(msg) = peer.rx.recv() => {
                // do something
            }
            result = peer.lines.next() => match result {
                // A message was received from the current user, we should
                // broadcast this message to the other users.
                Some(Ok(msg)) => {
                    // do something
                }
                Some(Err(e)) => {
                    // handle error
                }
                // The stream has been exhausted.
                None => break,
            },
        }
    }
    // do something

相较于以下使用两个tokio::spawn的实现：

let handle_1 = tokio::spawn(async move {
        while let Some(msg) = peer.rx.recv() {
            // do something
        }
    });
    
    let handle_2 = tokio::spawn (async move {
        loop {
            let result = peer.lines.next();
            match result {
                Some(Ok(msg)) => {
                    // do something
                },
                Some(Err(e)) => {
                    // handle error
                },
                None => break,
            };
        }
    });
    
    handle_1.await;
    handle_2.await;
// do something

请问使用loop+select!的实现方式相比spawn双任务有何优势？

回答 #

• 资源开销更低：每次tokio::spawn都会创建新任务，任务本身会产生调度元数据等内存开销。loop+select!在单个任务内处理两个异步操作，避免了额外的任务创建成本，在连接量大的场景（比如聊天服务）中，这种开销的累积差异会很显著。
• 状态访问更安全简洁：如果peer包含共享状态，双spawn任务需要使用互斥锁等同步原语保证线程安全；而loop+select!在同一任务内执行，所有操作是协作式调度下的单线程逻辑，可直接访问状态，无需额外同步，代码更简洁。
• 退出逻辑更统一：原示例中当lines.next()返回None时，直接break退出整个循环，两个异步操作会同时停止。双spawn实现则需要处理任务间通知——比如handle_2退出时要告知handle_1停止，否则handle_1可能一直阻塞在recv()上，还得额外实现任务取消逻辑，复杂度更高。
• 调度效率更高：Tokio任务调度需要切换上下文，单任务内的select!在同一上下文等待多个异步事件，减少了任务切换的开销，调度效率更高。

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