Fastapi集成SSE服务后端主动推送消息到前端

sse技术简介

SSE(Server-Sent Events)是一种允许服务器向客户端浏览器推送信息的技术。它是 HTML5 的一部分,专门用于建立一个单向的从服务器到客户端的通信连接。SSE的使用场景非常广泛,包括实时消息推送、实时通知更新等。
严格地说,HTTP 无法做到服务器主动推送信息。但是,有一种变通方法,就是服务器向客户端声明,接下来要发送的是流信息(streaming)。

也就是说,发送的不是一次性的数据包,而是一个数据流,会连续不断地发送过来。这时,客户端不会关闭连接,会一直等着服务器发过来的新的数据流,视频播放就是这样的例子。本质上,这种通信就是以流信息的方式,完成一次用时很长的下载。

SSE 就是利用这种机制,使用流信息向浏览器推送信息。

FastAPI集成SSE功能

配置环境
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple sse-starlette fastapi[all]

代码实现:

try:
    from sse_starlette.sse import EventSourceResponse
    import asyncio
except:
    import os
    os.system('pip install -i https://pypi.tuna.tsinghua.edu.cn/simple sse-starlette')
    from sse_starlette.sse import EventSourceResponse
    import asyncio


@router.get("/sse")
async def root(request: Request):
    async def event_generator(request: Request):
        while True:
            if await request.is_disconnected():
                print("连接已中断")
                break
            sse_test_data = gl.gl_Redis.hgetall('sse:test:data')
            sse_test_data['warn_site'] = sse_test_data['warn_site'].split(',') if sse_test_data.get('warn_site') else []
            yield {
                "event": "message",
                # "retry": 15000,
                "data": demjson.encode(sse_test_data)
            }

            await asyncio.sleep(10)
    g = event_generator(request)
    return EventSourceResponse(g)

SEE客户端测试脚本

import aiohttp
import asyncio

async def test():
    headers = {'Content-Type': 'text/event-stream'}
    sseresp = aiohttp.request("GET", r"http://127.0.0.1:9000/api/sse", headers=headers)
    async with sseresp as r:
        async for chunk in r.content.iter_any():
            print(chunk.decode())

if __name__ == '__main__':
    loop = asyncio.get_event_loop()
    loop.run_until_complete(test())

【特别提示】在现在大多数webviews框架打包APP时,sse服务无效!!!

本文由博客一文多发平台 OpenWrite 发布!

热门相关:帝少夜宠:小甜妻,乖!   重生之大文豪   娘娘每天都在洗白   大周仙吏   极品医圣