FastAPI WebSocket 基本使用二分组发送Json数据

最新推荐文章于 2025-09-26 21:23:27 发布

原创

最新推荐文章于 2025-09-26 21:23:27 发布 · 1.6k 阅读

4 ·

CC 4.0 BY-SA版权

更多博文在我的个人网站 https://www.charmcode.cn

文章标签：

#websocket json #fastapi 即时通讯

本文介绍如何使用FastAPI和WebSocket实现群聊功能，包括私信和广播消息。通过FastAPI服务端代码和客户端HTML页面，实现了用户之间的即时通讯。

FastAPI WebSocket 分组发送Json数据

用户1和用户2 可以互相发送私信消息
用户1 2 3之间相当于一个群，可以发送广播消息

效果

展示效果

代码

FastAPI 服务端代码

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2020/8/19 13:38
# @Author  : CoderCharm
# @File    : main.py
# @Software: PyCharm
# @Github  : github/CoderCharm
# @Email   : wg_python@163.com
# @Desc    :
"""

https://stackoverflow.com/questions/15219858/how-to-store-a-complex-object-in-redis-using-redis-py

obj = ExampleObject()
pickled_object = pickle.dumps(obj)
r.set('some_key', pickled_object)
unpacked_object = pickle.loads(r.get('some_key'))
obj == unpacked_object



typing.Dict[key_type, value_type]

"""

from typing import List, Dict

from fastapi import FastAPI, WebSocket, WebSocketDisconnect

app = FastAPI()


class ConnectionManager:
    def __init__(self):
        # 存放激活的链接
        self.active_connections: List[Dict[str, WebSocket]] = []

    async def connect(self, user: str, ws: WebSocket):
        # 链接
        await ws.accept()
        self.active_connections.append({
   
   "user": user, "ws": ws})

    def disconnect(self,user: str, ws: WebSocket):
        # 关闭时 移除ws对象
        self.active_connections.remove({
   
   "user": user, "ws": ws})

    @staticmethod
    async def send_personal_message(message: dict, ws: WebSocket):
        # 发送个人消息
        await ws.send_json(message)

    async def send_other_message(self, message: dict, user: str):
        # 发送个人消息
        for connection in self.active_connections:
            if connection["user"] == user:
                await connection['ws'].send_json(message)

    async def broadcast(self, data: dict):
        # 广播消息
        for connection in self.active_connections:
            await connection['ws'].send_json(data)


manager = ConnectionManager()


@app.websocket("/ws/{user}")
async def websocket_endpoint(ws: WebSocket, user: str):

    await manager.connect(user, ws)

    await manager.broadcast({
   
   "user": user, "message": "进入聊天"})

    try:
        while True:
            data = await ws.receive_json()
            print(data, type(data))

            send_user = data.get("send_user")
            if send_user:
                await manager.send_personal_message(data, ws)
                await manager.send_other_message(data, send_user)