为什么我的 FastAPI StreamingResponse 不能使用生成器函数？

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中文网其他相关文章！