FastAPI通过SSE进行流式输出

服务端推送

在服务器推送技术中，服务器在消息可用后立即主动向客户端发送消息。其中，有两种类型的服务器推送：SSE和 WebSocket。

SSE（Server-Send Events）

SSE 是一种在基于浏览器的 Web 应用程序中仅从服务器向客户端发送文本消息的技术。SSE基于 HTTP 协议中的持久连接，具有由 W3C 标准化的网络协议和 EventSource 客户端接口，作为 HTML5 标准套件的一部分。

那么，在FastAPI中怎么实现呢？

下面将介绍其实现过程：

pip install asyncio
pip install sse-starlette

代码：

import jsonfrom fastapi import FastAPI, Query, Path
import uvicorn
from starlette.requests import Request
import asyncio
from sse_starlette import EventSourceResponseapp = FastAPI()@app.get("/stream")
async def flush_stream(request: Request):async def event_generator(request: Request):res_str = "双天至尊真是一部好的电视剧！！！"for idx, word in enumerate(res_str):if await request.is_disconnected():print("连接已中断")breakdata = json.dumps({"id": idx, "message": word}, ensure_ascii=False)yield dataawait asyncio.sleep(1)return EventSourceResponse(event_generator(request))if __name__ == '__main__':uvicorn.run("Run:app", host="0.0.0.0", port=8080)

效果：