Rust+Rocket使用NamedFile返回文件扩展名错误及文件删除问题

1. 解决文件扩展名显示为.dms的问题 #

浏览器误判文件格式，核心原因是未通过响应头明确告知浏览器文件的类型与名称。默认情况下NamedFile无法正确识别xlsx的标准MIME类型，导致浏览器将文件识别为.dms格式。

需要给响应添加两个关键头：

• Content-Type: 设置为application/vnd.openxmlformats-officedocument.spreadsheetml.sheet（xlsx文件的标准MIME类型）
• Content-Disposition: 设置为attachment; filename="simple1.xlsx"，强制浏览器以附件形式下载并使用指定文件名

2. 解决下载后自动删除服务器文件的问题 #

直接在函数末尾删除文件会失败，因为NamedFile是异步发送的，函数执行到删除逻辑时文件仍处于被占用状态。最适合新手的方案是使用临时文件：

• 通过tempfile crate创建临时文件，文件会在临时文件对象被销毁时自动删除
• 无需手动管理文件路径与删除逻辑，避免文件残留

3. 完整优化后的代码 #

首先在Cargo.toml中添加tempfile依赖：

[package]
name = "demo_download_xlsx_rust"
version = "0.1.0"
edition = "2021"

[dependencies]
xlsxwriter = "0.4.0"
tempfile = "3.8.0"

[dependencies.rocket]
version = "0.5.0-rc.2"

修改后的Rust代码：

use std::path::Path;
use rocket::{
    response::{status::NotFound, Responder},
    http::Header,
    fs::NamedFile,
};
use xlsxwriter::Workbook;
use tempfile::NamedTempFile;

#[macro_use] extern crate rocket;

// 创建xlsx文件到临时文件，返回临时文件对象
fn create_workbook() -> Result<NamedTempFile, xlsxwriter::XlsxError> {
    // 创建带.xlsx后缀的临时文件，适配xlsxwriter的格式要求
    let temp_file = NamedTempFile::new()?;
    let temp_path = temp_file.path().to_path_buf();
    
    let workbook = Workbook::new(&temp_path);
    let mut sheet1 = workbook.add_worksheet(None)?;
    sheet1.write_string(2, 1, "Hello, world!", None)?;
    workbook.close()?;
    
    // 重新打开临时文件（xlsxwriter关闭后需重新获取可读取句柄）
    let temp_file = temp_file.reopen()?;
    Ok(temp_file)
}

#[get("/")]
async fn index() -> impl Responder<'static> {
    // 处理工作簿创建错误，返回NotFound响应
    let temp_file = match create_workbook() {
        Ok(file) => file,
        Err(e) => return NotFound(e.to_string()),
    };
    
    // 创建NamedFile并设置响应头
    let path = temp_file.path();
    let mut named_file = match NamedFile::open(path).await {
        Ok(file) => file,
        Err(e) => return NotFound(e.to_string()),
    };
    
    // 指定下载文件名
    named_file = named_file.with_header(Header::new(
        "Content-Disposition",
        "attachment; filename=\"simple1.xlsx\""
    ));
    // 指定文件MIME类型
    named_file = named_file.with_header(Header::new(
        "Content-Type",
        "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
    ));
    
    // 返回响应，临时文件会在发送完成后自动销毁
    named_file
}

#[launch]
fn rocket() -> _ {
    rocket::build()
        .mount("/", routes![index])
}

关键优化说明 #

• 错误处理: 替换expect为match，将错误转换为Rocket标准的NotFound响应，避免程序panic
• 临时文件管理: 用tempfile自动处理文件生命周期，无需手动删除文件
• 响应头明确化: 通过设置标准头彻底解决浏览器格式识别错误问题
• 逻辑简化: 直接操作临时文件，避免固定路径文件的重复读写操作

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