如何在Rust异步代码中以取消安全的方式检测stdin输入？

你遇到的核心问题是tokio的异步stdin读取（next_line）在select!中未被正确取消，导致当异步任务fut完成后，程序仍被阻塞等待用户输入。这是因为AsyncBufReadExt::next_line底层依赖后台线程的阻塞IO操作，直接使用时无法在分支完成后立即终止该操作。

解决方案：用spawn_blocking实现可取消的stdin读取 #

通过将阻塞式stdin读取包装到tokio::spawn_blocking中，配合JoinHandle的终止能力，确保异步任务完成时能立即停止stdin读取，避免阻塞executor，同时保证操作的取消安全性。

修改后的代码如下：

use std::future::Future;
use std::io::{self, BufRead};

use console::style;
use spinoff::{spinners, Spinner};
use tokio::{select, task};

pub async fn interact_with_cancel<R>(prompt: &str, fut: impl Future<Output = R>) -> Option<R> {
    let mut spinner = Spinner::new(
        spinners::Moon, 
        format!("{} {}", prompt, style("| hit enter to cancel").dim()), 
        None
    );

    // 将stdin读取包装为后台阻塞任务，返回可终止的JoinHandle
    let stdin_task = task::spawn_blocking(move || {
        let stdin = io::stdin();
        let mut reader = stdin.lock();
        let mut line = String::new();
        // 阻塞读取一行，任务可被外部abort终止
        let _ = reader.read_line(&mut line);
    });
    
    select! {
        _ = stdin_task => {
            spinner.clear();
            None
        },
        result = fut => {
            spinner.clear();
            // 异步任务完成时，主动终止stdin读取任务
            stdin_task.abort();
            Some(result)
        }
    }
}

#[cfg(test)]
mod test {
    use std::time::Duration;

    use super::*;

    #[tokio::test]
    async fn test_interact_with_cancel() {
        let _ = interact_with_cancel("testing", tokio::time::sleep(Duration::from_secs(2)))
            .await;
    }
}

关键改动说明 #

• 替换异步stdin实现：用spawn_blocking包装原生阻塞式stdin读取，避免异步API的隐式后台线程无法及时终止的问题。
• 主动终止任务：当异步任务fut完成时，调用stdin_task.abort()直接终止后台的stdin读取操作，确保程序立即返回，无需等待用户输入。
• 取消安全保障：spawn_blocking返回的JoinHandle在被终止或丢弃时，会正确清理后台线程资源，不会造成泄漏，完全符合取消安全要求。

如果需要检测单个按键（无需等待回车），可以结合crossterm或termion等终端库的非阻塞读取功能，同样包装到spawn_blocking中即可实现类似的取消安全逻辑。

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