【FastAPI】实现服务器向客户端发送SSE(Server-Sent Events)广播

news/2024/11/15 0:40:23/

在FastAPI中实现服务器向客户端发送SSE(Server-Sent Events)广播,可以通过以下步骤实现。SSE是一种服务器推送技术,允许服务器发送实时数据到客户端,通常用于创建实时更新的应用程序。

1. 安装必要的依赖

首先,确保你已经安装了FastAPI和Uvicorn:

pip install fastapi uvicorn

2. FastAPI服务器实现SSE广播

FastAPI使用StreamingResponse可以实现SSE。以下是一个实现SSE广播的简单例子,服务器会定期向所有连接的客户端广播消息。

python">from fastapi import FastAPI, Request
from fastapi.responses import StreamingResponse
from typing import List
import asyncioapp = FastAPI()# 用于保存所有的客户端连接
clients: List[asyncio.Queue] = []# SSE生成器,用于向客户端发送事件流
async def event_generator():queue = asyncio.Queue()  # 每个客户端拥有自己的消息队列clients.append(queue)try:while True:data = await queue.get()  # 等待新消息yield f"data: {data}\n\n"except asyncio.CancelledError:clients.remove(queue)  # 客户端断开连接时移除其队列raise# SSE endpoint,用于客户端连接
@app.get("/sse")
async def sse(request: Request):# 返回SSE流return StreamingResponse(event_generator(), media_type="text/event-stream")# 广播消息到所有客户端
async def broadcast_message(message: str):for queue in clients:await queue.put(message)  # 将消息放入所有客户端的队列中# 定期发送广播消息
@app.on_event("startup")
async def startup_event():async def send_messages():count = 0while True:await asyncio.sleep(5)  # 每5秒发送一次消息await broadcast_message(f"Server message: {count}")count += 1asyncio.create_task(send_messages())

3. 客户端实现

客户端可以使用EventSource对象来接收SSE事件流。以下是一个简单的HTML客户端代码:

<!DOCTYPE html>
<html>
<head><title>SSE Demo</title>
</head>
<body><h1>Server-Sent Events</h1><div id="messages"></div><script>const eventSource = new EventSource("/sse");eventSource.onmessage = function(event) {const newMessage = document.createElement("div");newMessage.textContent = event.data;document.getElementById("messages").appendChild(newMessage);};eventSource.onerror = function() {console.log("Error occurred or connection closed.");};</script>
</body>
</html>

4. 运行FastAPI服务器

使用Uvicorn运行服务器

uvicorn main:app --reload

这样,服务器会每5秒向所有连接的客户端广播消息。客户端只需要访问/sse即可建立与服务器的SSE连接,并实时接收服务器推送的消息。

关键点说明:

  • SSE(Server-Sent Events)服务器通过text/event-stream向客户端推送数据,客户端通过EventSource接收。
  • 广播实现:将消息放入所有客户端的队列中,确保每个连接的客户端都能接收到消息。
  • StreamingResponse:用于持续向客户端发送数据流。

这样,你就可以在FastAPI中实现一个SSE广播功能了。


http://www.ppmy.cn/news/1527518.html

相关文章

python函数一:函数的概念、函数定义与调用、函数的参数、函数的返回值、说明文档以及函数的嵌套调用

文章目录 1. 函数介绍1.1 函数的概念1.2 函数定义与调用1.2 函数的参数1.3 函数的返回值1.4 说明文档 2. 函数的嵌套调用2.1 嵌套调用及执行流程2.2 嵌套调用的应用 1. 函数介绍 1.1 函数的概念 什么是函数&#xff1f; 函数:是一个被命名的、独立的、完成特定功能的代码段&am…

【爬虫软件】批量采集抖音主页已发布作品

一、背景介绍 以下xx代表你猜中的部分。 1.1 爬取目标 用python开发的xx爬虫采集软件&#xff0c;可自动按博主抓取其已发布视频。 为什么有了源码还开发界面软件呢&#xff1f;方便不懂编程代码的小白用户使用&#xff0c;无需安装python&#xff0c;无需改代码&#xff0c;…

el-input设置type=‘number‘和v-model.number的区别

el-input设置typenumber’与设置.number修饰符的区别 1. 设置type‘number’ 使用el-input时想收集数字类型的数据&#xff0c;我们首先会想到typenumber&#xff0c;设置完type为number时会限制我们输入的内容只能为数字&#xff0c;不能为字符/汉字等非数字类型的数值&…

大数据处理技术:分布式文件系统HDFS

目录 1 实验名称&#xff1a; 2 实验目的 3 实验内容 4 实验原理 5 实验过程或源代码 5.1 HDFS的基本操作 5.2 HDFS-JAVA接口之读取文件 5.3 HDFS-JAVA接口之上传文件 5.4 HDFS-JAVA接口之删除文件 6 实验结果 6.1 HDFS的基本操作 6.2 HDFS-JAVA接口之读取文件 6.…

php变量赋值javascipt变量

javascript变量可以通过document.getElementById()获取html变量值php变量 可以在不同php脚本中使用&#xff0c;最后一次使用变量的初始值位为上一次使用变量的值 <!DOCTYPE html> <html> <body><?php $a "Hello World!"; echo $a; ?> …

Java入门程序-HelloWorld

Java程序开发的三个步骤 1.编写代码得到 .java 源代码文件 2.使用javac编译得到 .class 字节码文件 3.使用java运行 注意事项 建议代码文件名全英文&#xff0c;首字母大写&#xff0c;满足驼峰命名法&#xff0c;源代码文件的后缀必须是.java 开发HelloWorld程序 &…

网络基础,协议,OSI分层,TCP/IP模型

网络的产生是数据交流的必然趋势&#xff0c;计算机之间的独立的个体&#xff0c;想要进行数据交互&#xff0c;一开始是使用磁盘进行数据拷贝&#xff0c;可是这样的数据拷贝效率很低&#xff0c;于是网络交互便出现了&#xff1b; 1.网络是什么 网络&#xff0c;顾名思义是…

SQL Server数据库简单的事务日志备份恢复

模拟数据库备份恢复过程 1.基础操作 1.创建TestDB数据库&#xff0c;并添加数据 USE [master] GO CREATE DATABASE TestDB CONTAINMENT NONE ON PRIMARY ( NAME NTestDB, FILENAME ND:\TestDB.mdf , SIZE 8192KB , MAXSIZE UNLIMITED, FILEGROWTH 65536KB ) LOG ON ( …