【Python】FastApi中使用websocket 简单实现多人聊天室

君凉夜

已于 2024-03-04 19:24:39 修改

阅读量3.5k

点赞数 11

CC 4.0 BY-SA版权

文章标签： fastapi websocket 网络协议 python

于 2024-01-25 19:00:06 首次发布

本文链接：https://blog.csdn.net/ygxyvip/article/details/135839508

本文详细指导如何在FastAPI框架中通过WebSocket实现服务器推送给用户消息的功能，包括WebSocket的安装、WebSocketManager类的创建，以及前端Vue中的WebSocket连接和维护逻辑。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

FastApi就不用说了，懂得都懂，不懂的请先懂了再来；

WebSocket使得客户端和服务器之间的数据交换变得更加简单，简单点讲就是后端可以实时推送消息给正在使用的用户，多的不说，直接进入正题。

1、安装Websocket

pip install websockets

2、创建websocketutils.py

from fastapi import WebSocket
from typing import List

"""
创建工具管理类 处理服务端和客户端的交互
"""
class WebsocketManager:
    def __init__(self):
        # 初始化参数 记录活跃的客户端
        self.active_clients: List[WebSocket] = []
 
    async def connect(self, websocket: WebSocket):
        # 创建客户与服务端之间的连接 并记录下客户端
        await websocket.accept()
        self.active_clients.append(websocket)
 
    def disconnect(self, websocket: WebSocket):
        # 断开某个客户端的连接
        self.active_clients.remove(websocket)
 
    async def send_message_to_client(self, message: str, websocket: WebSocket):
        # 给客户端发送消息
        await websocket.send_text(message)
 
    async def broadcast(self, message: str):
        # 给所有客户端发送消息 广播
        for connection in