FastAPI大文件高效下载实践：告别内存溢出，巧用FileResponse

import io
from fastapi import FastAPI, Response
from starlette.responses import StreamingResponse

app = FastAPI()

@app.get("/download-large-file-bad-practice")
async def download_large_file_bad_practice(filename: str = "example_large_file.bin"):
    """
    此方法尝试使用StreamingResponse分发大文件，但存在内存问题。
    它将整个文件内容读入内存（通过file.read()），导致内存溢出。
    """
    try:
        # 假设文件位于当前目录或指定路径
        file_path = f"./{filename}"
        with open(file_path, "rb") as f:
            # 严重问题：file.read() 会将整个文件加载到内存中
            file_content = f.read()
            headers = {'Content-Disposition': f'attachment; filename="{filename}"'}
            # 即使StreamingResponse本身是流式的，但io.BytesIO(file_content)已经加载了整个文件
            return StreamingResponse(content=io.BytesIO(file_content), media_type="application/octet-stream", headers=headers)
    except FileNotFoundError:
        return Response(status_code=404, content="File not found")
    except Exception as e:
        return Response(status_code=500, content=f"An error occurred: {str(e)}")

import os
from fastapi import FastAPI, HTTPException
from starlette.responses import FileResponse

app = FastAPI()

# 假设在项目根目录有一个名为 'files' 的文件夹存放待下载的文件
# 为了演示，我们先创建一个虚拟大文件
def create_dummy_large_file(filename: str, size_mb: int):
    """创建指定大小的虚拟文件用于测试"""
    file_path = os.path.join("files", filename)
    os.makedirs(os.path.dirname(file_path), exist_ok=True)
    with open(file_path, "wb") as f:
        # 写入随机字节，模拟大文件
        f.write(os.urandom(1024 * 1024 * size_mb)) # 1MB * size_mb
    print(f"Created dummy file: {file_path} ({size_mb} MB)")

# 在应用启动时创建一些测试文件
@app.on_event("startup")
async def startup_event():
    create_dummy_large_file("test_document.pdf", 5) # 5 MB
    create_dummy_large_file("huge_archive.zip", 100) # 100 MB

@app.get("/download-file/{filename}")
async def download_file(filename: str):
    """
    使用FileResponse高效分发大文件。
    只需要提供文件的完整路径，FileResponse会处理剩余的流式传输。
    """
    file_path = os.path.join("files", filename) # 假设文件存储在 'files' 目录下

    if not os.path.exists(file_path):
        raise HTTPException(status_code=404, detail="File not found")

    # FileResponse会自动设置Content-Type和Content-Disposition
    # filename参数用于指定下载时客户端显示的文件名
    return FileResponse(path=file_path, filename=filename, media_type="application/octet-stream")