![](/img/trans.png)
[英]How to modify the content-length post modification of response in a middleware in FastAPI?
[英]FastAPI - How to get the response body in Middleware
有没有办法在中间件中获取响应内容? 以下代码是从这里复制的。
@app.middleware("http")
async def add_process_time_header(request: Request, call_next):
start_time = time.time()
response = await call_next(request)
process_time = time.time() - start_time
response.headers["X-Process-Time"] = str(process_time)
return response
response
体是一个迭代器,一旦被迭代,就不能再次被迭代。 因此,您要么必须将所有迭代数据保存到list
(或bytes
变量)并使用它来返回自定义Response
,要么再次启动迭代器。 下面的选项演示了这两种方法。 如果您也想在middleware
中获取request
正文,请查看此答案。
将数据保存到list
并使用iterate_in_threadpool
再次启动迭代器,如此处所述 - 这是StreamingResponse
使用的,如此处所示。
from starlette.concurrency import iterate_in_threadpool
@app.middleware("http")
async def some_middleware(request: Request, call_next):
response = await call_next(request)
response_body = [chunk async for chunk in response.body_iterator]
response.body_iterator = iterate_in_threadpool(iter(response_body))
print(f"response_body={response_body[0].decode()}")
return response
注意 1:如果您的代码使用StreamingResponse
, response_body[0]
将仅打印出response
的第chunk
。 要获取整个response_body
,您应该加入该字节列表(块),如下所示( .decode()
返回bytes
对象的字符串表示形式):
print(f"response_body={(b''.join(response_body)).decode()}")
注意 2:如果您有一个StreamingResponse
流式传输不适合服务器 RAM 的主体(例如,30GB 的响应),则在迭代response.body_iterator
时可能会遇到 memory 错误。 这适用于此答案中列出的两个选项。
下面演示了另一种方法,其中响应正文存储在bytes
object 中(而不是列表,如上所示),并用于直接返回自定义Response
(连同原始响应的status_code
、 headers
和media_type
) .
@app.middleware("http")
async def some_middleware(request: Request, call_next):
response = await call_next(request)
response_body = b""
async for chunk in response.body_iterator:
response_body += chunk
print(f"response_body={response_body.decode()}")
return Response(content=response_body, status_code=response.status_code,
headers=dict(response.headers), media_type=response.media_type)
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.