| 1234567891011121314151617181920212223242526272829303132 |
- from models import WebSocket
- from logger import logger
- import json
- class ConnectionManager:
- def __init__(self):
- self.active_connections: list[WebSocket] = []
- def jsonMessage(self, code=0, msg="", data: object = None):
- """json字符串数据"""
- jsonData = {"code": code, "msg": msg, "data": data}
- return json.dumps(jsonData)
- async def connect(self, websocket: WebSocket):
- '''连接事件'''
- await websocket.accept()
- self.active_connections.append(websocket)
- logger.info("socket 已连接")
- def disconnect(self, websocket: WebSocket):
- '''断开连接事件'''
- self.active_connections.remove(websocket)
- logger.info("socket 连接断开")
- async def send_personal_message(self, message: str, websocket: WebSocket):
- '''向用户发送消息'''
- await websocket.send_json(message)
- async def broadcast(self, message: str):
- """广播消息"""
- for connection in self.active_connections:
- await connection.send_json(message)
|
