首页 > 后端开发 > Python教程 > 为什么我的 FastAPI StreamingResponse 无法使用生成器函数进行流式传输?

为什么我的 FastAPI StreamingResponse 无法使用生成器函数进行流式传输?

Mary-Kate Olsen
发布: 2024-11-08 12:37:02
原创
906 人浏览过

Why is My FastAPI StreamingResponse Failing to Stream with a Generator Function?

FastAPI StreamingResponse 无法使用生成器函数进行流式处理

FastAPI 的 StreamingResponse 是一种将数据增量发送回客户端的便捷方法,但有时它会出现问题可能不会按预期运行,尤其是在使用生成器函数时。在这里,我们将深入探讨潜在原因及其各自的解决方案。

常见原因和解决方案:

1.不正确的 HTTP 方法和凭据处理:

避免使用 POST 请求进行数据检索。相反,选择 GET 请求。另外,强烈建议使用标头或 cookie 作为凭据而不是查询参数,以增强安全性并避免 URL 参数污染。

2.生成器函数中的阻塞操作:

如果您的生成器函数包含阻塞 I/O 或 CPU 密集型操作,请使用 def 而不是 async def 来防止潜在的死锁和事件循环中断。或者,如果使用 async def,请在单独的 ThreadPool 或 ProcessPool 中执行阻塞操作。

3.不完整的换行符:

如果您使用请求的 iter_lines() 迭代响应数据,请考虑它逐行读取响应。为了确保数据到达时显示,请修改您的响应以包含换行符或使用具有指定块大小的 iter_content()。

4.媒体类型和 MIME 嗅探:

浏览器可能会缓冲文本/纯文本响应来检测内容类型。要避免这种情况,请使用不同的媒体类型(例如 application/json 或 text/event-stream)或通过将 X-Content-Type-Options 标头设置为 nosniff 来禁用 MIME 嗅探。

示例解决方案:

下面是 FastAPI 应用程序的一个工作实现,它可以流式传输虚假数据并解决上述问题:

from fastapi import FastAPI
from fastapi.responses import 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():
    headers = {'X-Content-Type-Options': 'nosniff'}
    return StreamingResponse(fake_data_streamer(), headers=headers, media_type='text/plain')
登录后复制

请记住,处理流式响应可能会有所不同,具体取决于客户端(Web 浏览器、HTTP 客户端等)及其各自的功能。

以上是为什么我的 FastAPI StreamingResponse 无法使用生成器函数进行流式传输?的详细内容。更多信息请关注PHP中文网其他相关文章!

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