Rust异步SleepSort程序mpsc通道无法终止问题求助

程序无法终止的核心原因是Tokio MPSC通道的发送端未全部销毁：

• 你创建的通道(tx, rx)中，主函数保留了原始的tx实例，同时给每个异步任务克隆了一份tx。
• Tokio的MPSC通道规则是：只有当所有Sender实例都被drop后，接收端rx.recv()才会返回None，从而结束while let循环。
• 当前主函数的tx一直存活到程序结束，通道永远不会触发“关闭”信号，导致rx.recv()一直阻塞等待，程序无法退出。

在所有异步任务启动完成后，手动销毁主函数中的原始tx，这样所有Sender就只剩下任务内的实例，当任务完成send操作后，任务内的tx会被自动销毁，最终所有Sender都消失，通道关闭，rx.recv()返回None，循环终止。

修改后的代码如下：

use std::time::Duration;

use text_io::read;
use tokio::sync::mpsc::channel;

#[tokio::main]
async fn main() {
    let n: u32 = read!();
    let (tx, mut rx) = channel::<u32>(n as usize);
    let mut before: Vec<u32> = Vec::new();
    for _ in 0..n {
        let a = read!();
        before.push(a);
    }

    for i in before.into_iter() {
        let tx = tx.clone();
        tokio::spawn(async move {
            tokio::time::sleep(Duration::from_millis((50 * i) as u64)).await;
            if let Err(_) = tx.send(i).await {
                panic!();
            }
        });
    }

    // 销毁主函数中的原始发送端，触发通道关闭条件
    drop(tx);

    while let Some(i) = rx.recv().await {
        println!("{}", i);
    }
}

另外，也可以选择收集所有异步任务的JoinHandle，等待全部任务完成后主动退出循环，但手动drop(tx)是更贴合MPSC通道设计的简洁方案。

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