3f4165fe78
核心实现：搭建 Monorepo 架构，完成 shared 共享包、gateway、user-service 基础框架开发 技术落地：严格匹配指定技术栈版本，完成 Docker、gRPC、FastAPI、PostgreSQL、Redis 等配置，实现服务间基础连通 配套文件：生成 Makefile、环境变量模板、数据库 / 脚本初始化文件及启动验证文档 架构定位：仅搭建基础架构骨架，无任何业务逻辑、业务规则及业务相关字段，为后续业务开发提供支撑
25 lines
864 B
Python
25 lines
864 B
Python
from fastapi import WebSocket
|
|
from typing import Dict, List
|
|
|
|
class ConnectionManager:
|
|
def __init__(self):
|
|
self.active_connections: Dict[str, WebSocket] = {}
|
|
|
|
async def connect(self, websocket: WebSocket, client_id: str):
|
|
await websocket.accept()
|
|
self.active_connections[client_id] = websocket
|
|
|
|
def disconnect(self, client_id: str):
|
|
if client_id in self.active_connections:
|
|
del self.active_connections[client_id]
|
|
|
|
async def send_personal_message(self, message: str, client_id: str):
|
|
if client_id in self.active_connections:
|
|
await self.active_connections[client_id].send_text(message)
|
|
|
|
async def broadcast(self, message: str):
|
|
for connection in self.active_connections.values():
|
|
await connection.send_text(message)
|
|
|
|
manager = ConnectionManager()
|