如何处理tokio::try_join!中任务panic：匹配类型与干净终止

1. 定义自定义Panic类型 #

先创建枚举标识不同panic场景，必须实现Debug trait（panic payload要求兼容Any + Send，且Debug用于基础打印）：

use std::any::Any;
use tokio::sync::Notify;
use std::panic::{self, PanicInfo};

#[derive(Debug)]
enum CustomPanic {
    InvalidX,
    InvalidY,
    InvalidZ,
}

2. 修改异步函数抛出自定义Panic #

把原函数中抛出字符串panic的逻辑，改为抛出自定义枚举值：

async fn func1() {
    // 模拟触发X无效的panic场景
    panic!(CustomPanic::InvalidX);
}

async fn func2() {
    // 模拟触发Y无效的panic场景
    panic!(CustomPanic::InvalidY);
}

3. 配置Panic Hook区分类型+优雅终止Runtime #

用downcast_ref精准匹配panic类型，同时通过线程安全通知器触发Runtime优雅关闭，替代粗暴的process::exit：

// 全局通知器，用于触发Runtime优雅关闭
static SHUTDOWN_NOTIFY: Notify = Notify::const_new();

fn main() {
    panic::set_hook(Box::new(|panic_info: &PanicInfo<'_>| {
        // 尝试将panic payload转换为自定义类型，实现精准匹配
        match panic_info.payload().downcast_ref::<CustomPanic>() {
            Some(CustomPanic::InvalidX) => {
                println!("X was invalid, please try using valid X");
            }
            Some(CustomPanic::InvalidY) => {
                println!("Y was invalid, please try using valid Y");
            }
            Some(CustomPanic::InvalidZ) => {
                println!("Z was invalid, please try using valid Z");
            }
            None => {
                println!("Something else happened: {}", panic_info);
            }
        }
        // 通知主线程启动关闭流程
        SHUTDOWN_NOTIFY.notify_one();
    }));

    // 初始化并启动Tokio Runtime
    let rt = tokio::runtime::Runtime::new().unwrap();
    rt.block_on(async {
        // 执行异步任务
        let _ = tokio::join!(func1(), func2());
        // 等待关闭通知
        SHUTDOWN_NOTIFY.notified().await;
    });

    // 优雅清理Runtime资源
    rt.shutdown_background();
}

核心细节说明 #

• 类型匹配逻辑：panic_info.payload()返回&dyn Any，downcast_ref::<T>()可尝试将其转为具体类型引用，成功则返回Some(T)，失败返回None，完全避免字符串正则匹配的不可靠性。
• 优雅终止原理：用tokio::sync::Notify在panic线程中发送信号，主线程收到后结束异步任务块，再调用shutdown_background()让Runtime清理未完成的任务和资源，避免直接终止进程导致的资源泄漏。
• 类型约束：自定义panic类型默认实现Send（只要成员都兼容Send），满足Tokio多线程环境下跨线程传递panic payload的要求。

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