StreamingResponse 不使用生成器函数进行流式处理
FastAPI 提供 StreamingResponse 类来将响应增量发送回客户端。然而,在某些情况下,此功能可能无法按预期工作。
根本原因调查
分析提供的 FastAPI 代码和问题描述后,我们已经确定几个潜在的原因:
1。使用POST请求进行数据请求:
使用POST请求不适合从服务器请求数据。相反,建议使用 GET 请求来实现此目的。
2.使用查询参数进行身份验证:
通过查询参数发送敏感凭据(例如 auth_key)是不安全的。考虑使用标头或 cookie 进行身份验证。
3.阻塞生成器函数:
StreamingResponse 中的生成器函数是使用 def(不是异步 def)定义的,这可能会导致 FastAPI 事件循环中出现阻塞问题。
4.基于行的分块:
请求的 iter_lines() 一次迭代一行响应数据。如果响应中不存在换行符,则不会增量打印数据。
5. MIME 嗅探:
某些浏览器(例如 Chrome)可能会缓冲文本/纯文本响应,以便在显示纯文本内容之前检查它们。这可能会阻碍流媒体传输。
建议修复:
1.使用 GET 请求:
重构代码以使用 GET 请求来获取数据。
2.安全身份验证:
使用标头或 cookie 安全地发送身份验证凭据。
3.异步生成器函数:
使用 async def 定义 StreamingResponse 的生成器函数。如果生成器内需要任何阻塞操作,请使用外部线程池来执行它们。
4.基于块的分块:
使用 iter_content() 而不是 iter_lines() 来迭代块中的响应数据。指定适当的块大小。
5.禁用 MIME 嗅探:
通过为 StreamingResponse 指定不同的媒体类型(例如 application/json 或 text/event-stream)或通过将 X-Content-Type-Options 标头设置为 nosniff 来禁用 MIME 嗅探。
工作示例:
以下代码演示了具有流功能的工作 FastAPI 应用程序:
from fastapi import FastAPI, StreamingResponse import asyncio app = FastAPI() async def fake_data_streamer(): for i in range(10): yield b'some fake data\n\n' await asyncio.sleep(0.5) @app.get('/') async def main(): return StreamingResponse(fake_data_streamer(), media_type='text/event-stream')
以上是为什么我的 FastAPI StreamingResponse 不能使用生成器函数?的详细内容。更多信息请关注PHP中文网其他相关文章!