Rust Tokio场景下如何让读取用户输入的任务同时监听shutdown信号

你当前的代码存在两个核心问题：

• 不应在async函数中直接调用std::io::Stdin::read_line这类阻塞IO接口，该调用会阻塞当前执行的tokio工作线程，导致同线程下的其他异步任务无法调度，违背tokio的异步运行时设计规则
• 阻塞读操作本身会一直等待用户输入，在等待过程中没有办法插入shutdown信号的检测逻辑

你需要按照官方建议，拆分阻塞IO和异步逻辑两部分实现：

1. 衍生独立的标准库线程处理用户输入，通过异步通道将读到的内容传递到异步任务侧
2. 在异步任务中通过tokio::select!同时监听输入消息和shutdown信号，任意一个触发都可以终止流程

示例代码 #

use std::io::{self, BufRead};
use std::thread;
use tokio::sync::{mpsc, watch};

async fn read_input(mut shutdown_rx: watch::Receiver<&str>) {
    // 创建通道传递用户输入，容量设为1足够处理单行输入
    let (input_tx, mut input_rx) = mpsc::channel(1);
    
    // 衍生独立线程处理阻塞stdin读取
    thread::spawn(move || {
        let stdin = io::stdin();
        let mut lines = stdin.lock().lines();
        while let Some(Ok(line)) = lines.next() {
            // 发送输入到异步侧，如果通道关闭说明已经触发shutdown，直接退出线程
            if input_tx.blocking_send(line).is_err() {
                break;
            }
        }
    });

    loop {
        tokio::select! {
            // 收到shutdown信号直接退出
            _ = shutdown_rx.changed() => {
                break;
            }
            // 收到用户输入处理逻辑
            Some(line) = input_rx.recv() => {
                let op = line.trim_end();
                if op == "EXIT" {
                    break;
                } else if op == "send" {
                    // send_stuff();
                }
            }
        }
    }
}

逻辑说明 #

• 输入处理线程完全使用阻塞IO，不会影响异步运行时的调度
• 当shutdown信号触发时，select会直接退出循环，此时input_rx会被销毁，输入线程调用blocking_send时会发现通道断开，自动退出
• 如果用户输入EXIT，也可以主动终止整个逻辑，符合原有交互需求

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