基于当前运行环境获取Tokio runtime句柄的正确实现方式是什么？

现有get_runtime_handle实现的问题 #

• Runtime实例提前销毁：你在无Runtime场景下执行Runtime::new().unwrap().handle().clone()时，新创建的Runtime实例没有被持有，语句执行结束就会被Drop，对应的IO驱动、定时器、worker线程都会被终止，后续通过这个悬空Handle提交的异步任务（比如tokio::fs::read_dir）都会因为后台Runtime已销毁触发background task failed错误，这就是你fs调用崩溃的直接原因。你在main中直接新建Runtime并调用能正常运行，就是因为此时Runtime实例在main作用域内没有被提前释放。
• 阻塞调用违反Tokio约束：当代码已经运行在Tokio Runtime的worker线程上时，调用Handle::block_on会阻塞当前worker线程，Tokio明确禁止这种行为，会直接抛出你遇到的panic。
• 重复创建Runtime损耗性能：每次无Runtime时调用函数都会新建Runtime实例，会导致进程内存在多个Tokio Runtime，产生不必要的资源消耗和跨Runtime调度开销。
• 笔误问题：你代码中使用的handle.block_one是错误API，正确方法名为block_on。

正确实现方案 #

方案1：全链路异步化（优先推荐） #

Tokio官方最推荐的用法是全链路使用异步，从根源上避免同步调用异步的兼容问题：

use tokio::runtime::Runtime;
use std::io::Result;

async fn a_async() -> Result<()> {
    // 你的异步逻辑，例如 tokio::fs::read_dir 调用
    Ok(())
}

async fn a() -> Result<()> {
    a_async().await
}

#[tokio::main]
async fn main() -> Result<()> {
    a().await?;
    Ok(())
}

单元测试直接使用#[tokio::test]标记异步测试即可，不需要手动操作Runtime。

方案2：兼容同步调用入口的实现 #

如果确实需要对外暴露同步的a()函数，兼容同步调用场景，需要做两处核心修改：全局缓存自建的Runtime避免实例被销毁，检测上下文避免非法调用block_on：

use tokio::{runtime::{Handle, Runtime}, task};
use std::io::Result;
use once_cell::sync::Lazy;

// 全局缓存自建Runtime，仅当没有外部Runtime时使用
// Rust 1.70+可直接替换为std::sync::LazyLock，无需额外引入依赖
static GLOBAL_RUNTIME: Lazy<Runtime> = Lazy::new(|| Runtime::new().unwrap());

fn get_runtime_handle() -> Handle {
    Handle::try_current().unwrap_or_else(|_| GLOBAL_RUNTIME.handle().clone())
}

async fn a_async() -> Result<()> {
    // 你的异步逻辑
    Ok(())
}

fn a() -> Result<()> {
    match Handle::try_current() {
        // 已在Tokio Runtime中，用block_in_place规避worker线程阻塞panic
        Ok(handle) => task::block_in_place(move || handle.block_on(a_async())),
        // 无Runtime环境，直接用全局Runtime运行
        Err(_) => GLOBAL_RUNTIME.block_on(a_async())
    }
}

fn main() -> Result<()> {
    a()?;
    Ok(())
}

注意：block_in_place仅在Tokio多线程Runtime下可用，若你使用单线程current_thread Runtime，调用block_in_place依然会panic，这种场景下请避免在异步上下文中调用同步的a()函数。

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