FastAPI中间件读取响应体时如何避免等待后台任务完成？

你的问题核心在于：Starlette（FastAPI基于它）的BackgroundTasks是在响应完全发送给客户端之后才触发执行的，但你在中间件里提前异步遍历并消费了整个body_iterator，这会导致Starlette误以为响应已经发送完成，提前触发后台任务，而中间件的协程会等待任务执行完毕才返回响应给客户端，最终造成响应延迟。

另外，后台任务的错误会传播到接口，是因为提前触发的任务运行在当前请求的上下文里，异常会被请求流程捕获，而非在响应发送后的独立上下文里隔离处理。

要实现读取响应体（比如日志记录）同时不阻塞后台任务执行，你需要避免提前消费整个body_iterator，而是通过包装迭代器的方式，在响应体逐步发送给客户端的过程中读取内容，保留原有的后台任务触发逻辑。

修改后的代码如下：

from fastapi import FastAPI, BackgroundTasks, Request, Response
import asyncio
import logging
from starlette.background import BackgroundTask

app = FastAPI()

class LoggingAsyncIterator:
    def __init__(self, iterator):
        self.iterator = iterator
        self.body_chunks = []

    def __aiter__(self):
        return self

    async def __anext__(self):
        try:
            chunk = await self.iterator.__anext__()
            self.body_chunks.append(chunk)
            return chunk
        except StopAsyncIteration:
            # 响应体全部发送完成后再处理日志，此时后台任务已开始执行
            full_body = b''.join(self.body_chunks).decode('utf-8')
            logging.info(f"Response body: {full_body}")
            raise StopAsyncIteration

@app.middleware('http')
async def some_middleware(request: Request, call_next):
    response = await call_next(request)
    # 包装原迭代器，不提前消费响应内容
    response.body_iterator = LoggingAsyncIterator(response.body_iterator)
    return response

async def any_task():
    await asyncio.sleep(5)
    print("Background task finished")

@app.get("/")
async def root(bg: BackgroundTasks):
    bg.add_task(any_task)
    return {"message": "OK"}

• LoggingAsyncIterator会在每一次读取响应体片段时记录内容，直到迭代完成（响应体全部发送给客户端），此时再拼接完整响应体做日志处理。
• 这种方式不会提前触发后台任务，Starlette会在响应完全发送给客户端后正常执行任务，响应不会被延迟。
• 后台任务的异常不会传播到请求上下文，因为任务运行在响应发送后的独立流程中。
• 避免直接操作响应对象的__dict__，直接访问response.body_iterator属性更符合Starlette的设计规范。

内容的提问来源于stack exchange，提问作者João Pedro Zimmermann