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

DDD
发布: 2024-11-08 09:51:01
原创
906 人浏览过

Why Isn't My FastAPI StreamingResponse Working with a Generator Function?

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

来源:php.cn
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板