Rust 非异步场景下调用异步方法设计拉取器trait的方案咨询

这里提供三种落地思路，可根据你的现有系统改造成本选择：

方案1：全异步化重构（长期最优） #

如果可以接受小范围重构现有调用链路，直接把trait升级为异步实现是最稳妥的方案，Rust 1.75+已经原生支持trait中的异步方法，低于该版本可以使用async-trait宏兼容。

// 统一异步trait定义
trait GraphPuller {
    async fn graph() -> Result<Vec<i64>, Box<dyn std::error::Error>>;
}

struct GitPuller;
struct TcpPuller;
struct AwsPuller;

impl GraphPuller for GitPuller {
    async fn graph() -> Result<Vec<i64>, Box<dyn std::error::Error>> {
        // 原有同步逻辑直接保留即可，同步代码在异步上下文中可正常执行
        Ok(vec![])
    }
}

impl GraphPuller for TcpPuller {
    async fn graph() -> Result<Vec<i64>, Box<dyn std::error::Error>> {
        Ok(vec![])
    }
}

impl GraphPuller for AwsPuller {
    async fn graph() -> Result<Vec<i64>, Box<dyn std::error::Error>> {
        // 直接调用AWS官方异步SDK即可
        let aws_config = aws_config::load_from_env().await;
        let client = aws_sdk_s3::Client::new(&aws_config);
        let resp = client.get_object()
            .bucket("your-bucket")
            .key("your-data-path")
            .send()
            .await?;
        // 处理返回结果得到目标Vec<i64>
        Ok(vec![])
    }
}

方案2：兼容现有同步调用链路（零重构老代码） #

如果不能修改原有同步调用逻辑，可在AWS实现侧把异步代码做阻塞执行，前提是你的上层调用链路全是同步环境，没有嵌套异步运行时。

// 完全保留原有同步trait不变
trait GraphPuller {
    fn graph() -> Result<Vec<i64>, Box<dyn std::error::Error>>;
}

struct AwsPuller;

impl GraphPuller for AwsPuller {
    fn graph() -> Result<Vec<i64>, Box<dyn std::error::Error>> {
        // 初始化异步运行时，建议做成全局单例避免重复初始化开销
        let rt = tokio::runtime::Runtime::new()?;
        // 阻塞执行异步逻辑
        let result = rt.block_on(async {
            let aws_config = aws_config::load_from_env().await;
            let client = aws_sdk_s3::Client::new(&aws_config);
            let resp = client.get_object()
                .bucket("your-bucket")
                .key("your-data-path")
                .send()
                .await?;
            Ok::<_, Box<dyn std::error::Error>>(vec![])
        })?;
        Ok(result)
    }
}

可选方案：拆分同步/异步trait，灵活调度 #

如果需要同时保留独立的同步、异步调用路径，可以拆分两个子trait，上层根据数据源类型自动适配：

use thiserror::Error;

// 自定义统一错误类型，避免动态分发开销
#[derive(Error, Debug)]
enum PullError {
    #[error("IO error: {0}")]
    Io(#[from] std::io::Error),
    #[error("AWS error: {0}")]
    Aws(#[from] aws_sdk_s3::error::SdkError<aws_sdk_s3::operation::get_object::GetObjectError>),
}

trait SyncGraphPuller {
    fn graph() -> Result<Vec<i64>, PullError>;
}

trait AsyncGraphPuller {
    async fn graph() -> Result<Vec<i64>, PullError>;
}

• 禁止在异步运行时的worker线程中调用block_on：会直接触发死锁，只要上层存在异步上下文，必须选择全异步化方案
• 运行时复用：如果使用方案2，不要每次调用都新建tokio运行时，做成全局单例可大幅降低性能损耗
• 命名规范：Rust官方约定方法使用蛇形命名，建议将原有的Graph方法改为graph符合语言规范

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