Rust Axum Multipart大文件上传遇长度超限Panic问题求助

问题原因 #

Axum的Multipart组件默认对单个字段和整个请求的大小有严格限制，当上传大文件（如视频）时，一旦超过默认阈值就会抛出length limit exceeded错误，而原代码中使用unwrap()会直接触发线程panic。

解决方案 #

1. 自定义Multipart大小限制 #

通过MultipartConfig可以修改默认的大小限制，根据业务需求调整单个字段和总请求的最大允许值：

首先确保Cargo.toml中Axum启用multipart特性：

[dependencies]
axum = { version = "0.7", features = ["multipart"] }
tokio = { version = "1.0", features = ["full"] }

修改代码，注入自定义配置并处理上传逻辑：

use axum::{
    extract::{Multipart, MultipartConfig},
    routing::post,
    Router,
};

async fn upload(multipart: Multipart) -> Result<(), String> {
    while let Some(mut field) = multipart.next_field().await.map_err(|e| e.to_string())? {
        let name = field.name().ok_or("上传字段未命名")?.to_string();
        
        // 流式写入文件，避免大文件占用过多内存
        let mut file = tokio::fs::File::create(&name)
            .await
            .map_err(|e| format!("创建文件失败: {}", e))?;
        tokio::io::copy(&mut field, &mut file)
            .await
            .map_err(|e| format!("写入文件失败: {}", e))?;
        
        println!("文件 `{}` 上传成功", name);
    }
    Ok(())
}

#[tokio::main]
async fn main() {
    // 配置单个字段最大10GB，总请求最大10GB（可按需调整）
    let multipart_config = MultipartConfig::new()
        .max_field_size(10 * 1024 * 1024 * 1024)
        .max_total_size(10 * 1024 * 1024 * 1024);

    let app = Router::new()
        .route("/upload", post(upload))
        .with_state(multipart_config);

    let addr = ([127, 0, 0, 1], 3000).into();
    println!("服务启动，监听地址: {}", addr);
    axum::Server::bind(&addr)
        .serve(app.into_make_service())
        .await
        .unwrap();
}

2. 替换unwrap()，优雅处理错误 #

原代码中unwrap()会在出错时直接panic，改用map_err和?运算符处理错误，返回可读的错误信息，避免服务崩溃。

3. 流式处理大文件（推荐） #

原代码用field.bytes().await会将整个文件加载到内存，大视频文件可能导致内存溢出。改用tokio::io::copy将文件流直接写入磁盘，大幅降低内存占用。

关键说明 #

• max_field_size控制单个上传文件的最大大小，max_total_size控制整个multipart请求的总大小，需根据服务器资源合理设置。
• 确保Tokio启用fs特性，支持异步文件操作。

内容的提问来源于stack exchange，提问作者Ty Q.