Axum+Tokio下载文件：如何设计兼容AsyncRead的函数签名？

我正在用Axum实现文件下载功能，现有代码运行正常。希望将文件读取逻辑抽离为独立函数，同时兼容tokio::fs::File与rust-s3的文件读取，计划以AsyncRead作为公共返回类型，但尝试的函数签名出现类型不匹配、返回临时值引用等错误，请问正确的函数签名应该如何设计？

尝试的代码示例 #

use tokio::io::AsyncRead;

pub struct Player {
    db: Arc<DB>
}

impl Handler {
    pub async fn player_pdf(
        &self,
        id: &str,
    ) -> Result<&(dyn AsyncRead)> {
        //...use id here...

        let file = &tokio::fs::File::open("player.pdf").await?;

        Ok(file)
    }
}

遇到的错误 #

• 类型不匹配错误：期望&dyn tokio::io::AsyncRead，实际为tokio::fs::File
• 返回临时值引用错误：返回的引用指向函数内创建的临时值

解决方案 #

核心是避免返回临时值的引用，同时让不同类型的异步读取器统一到AsyncRead trait对象下，推荐以下两种实现方式：

方式1：返回装箱的动态trait对象（推荐动态分发场景） #

将返回值改为Box<dyn AsyncRead + Unpin + Send>，持有具体读取器的所有权，同时满足异步运行时的约束：

use tokio::io::{AsyncRead};
use std::sync::Arc;
use std::error::Error;

pub struct Handler {
    db: Arc<DB>
}

impl Handler {
    pub async fn player_pdf(
        &self,
        id: &str,
    ) -> Result<Box<dyn AsyncRead + Unpin + Send>, Box<dyn Error>> {
        // 根据业务逻辑选择读取本地文件或S3对象
        if self.needs_local_file(id).await {
            let local_file = tokio::fs::File::open("player.pdf").await?;
            Ok(Box::new(local_file))
        } else {
            // 假设rust-s3返回的对象实现了AsyncRead + Unpin + Send
            let s3_object = self.s3_client.get_object("bucket_name", id).await?;
            Ok(Box::new(s3_object))
        }
    }

    // 示例辅助方法：判断是否需要读取本地文件
    async fn needs_local_file(&self, id: &str) -> bool {
        // 实际逻辑：从db查询id对应的存储位置
        true
    }
}

• Box<dyn AsyncRead>：将具体类型装箱为trait对象，解决类型不匹配问题
• Unpin：Axum处理响应时要求异步读取器实现该trait，绝大多数异步IO类型默认支持
• Send：满足Axum多线程运行时的跨线程发送要求

方式2：使用泛型约束（适合类型确定的场景） #

如果调用处能明确返回类型，可通过泛型约束实现静态分发，避免动态装箱开销：

use tokio::io::AsyncRead;
use std::sync::Arc;
use std::error::Error;

pub struct Handler {
    db: Arc<DB>
}

impl Handler {
    pub async fn player_pdf<R>(
        &self,
        id: &str,
    ) -> Result<R, Box<dyn Error>>
    where
        R: AsyncRead + Unpin + Send,
    {
        // 根据业务逻辑返回对应类型的实例，需确保类型转换合法
        let file = tokio::fs::File::open("player.pdf").await?;
        Ok(file as R)
    }
}

这种方式灵活性较弱，仅适合返回类型固定的场景。

结合Axum生成下载响应 #

抽离后的函数可直接用于Axum的响应处理，比如返回StreamingBody：

use axum::body::StreamingBody;
use tokio_util::io::ReaderStream;
use axum::extract::Path;

// Axum路由handler
pub async fn download_pdf(
    handler: Arc<Handler>,
    Path(id): Path<String>
) -> Result<StreamingBody, Box<dyn Error>> {
    let reader = handler.player_pdf(&id).await?;
    let stream = ReaderStream::new(reader);
    Ok(StreamingBody::new(stream))
}

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