多发送者场景下如何关闭Tokio mpsc发送端通道？

你的代码核心问题在于主函数始终持有原始的Sender实例，而Tokio的MPSC通道只有当所有Sender都被drop后，接收端的recv()才会返回None。只要主函数里的tx存在，通道就会一直保持开启状态，即便所有爬虫任务完成，程序也会卡在rx.recv().await处无法退出。

解决方案：使用WeakSender避免强引用持有 #

通过WeakSender（从Sender调用downgrade()获取），我们可以让主函数持有弱引用，不再保留强引用的Sender。只有当需要创建新任务时，才尝试将弱引用升级为Sender；当所有强引用的Sender都被drop后，通道会自动关闭，接收端退出循环。

修改后的代码如下：

use tokio::sync::mpsc;

async fn visit_page(url: u32) -> Option<u32>{
    if url < 10 {
        Some(url + 1) 
    } else {
        None
    }
}

#[tokio::main]
async fn main() {
    let (tx, mut rx) = mpsc::channel(100);
    // 获取Sender的弱引用
    let weak_tx = tx.downgrade();

    // 发送初始任务后，立即drop原始Sender，避免主函数持有强引用
    tx.send(Some(0)).await.unwrap();
    drop(tx);

    while let Some(recv) = rx.recv().await {
        if let Some(url) = recv {
            println!("Got = {}", url);
            // 尝试将弱引用升级为Sender，若失败说明通道已关闭，不再创建新任务
            if let Some(tx) = weak_tx.upgrade() {
                tokio::spawn(async move {
                    let result = visit_page(url).await;
                    // 忽略发送错误（比如通道已关闭），避免程序panic
                    let _ = tx.send(result).await;
                });
            }
        }
    }
}

代码解释 #

1. WeakSender的作用：它不会阻止通道关闭，只有当存在活跃的Sender强引用时，upgrade()才能成功获取到可用的Sender。
2. drop原始Sender：主函数不再持有强引用的Sender，通道的生命周期完全由爬虫任务中的Sender副本决定。
3. 忽略发送错误：当通道已经关闭时，send()会返回错误，这里用let _ =忽略错误，避免不必要的panic。

当最后一个爬虫任务完成并drop掉持有的Sender后，所有强引用消失，通道关闭，rx.recv().await返回None，主循环退出，程序正常结束。

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