FastAPI-websocket, two channels

main.py

@@ -0,0 +1,74 @@
+import logging
+
+from fastapi import FastAPI, WebSocket, WebSocketDisconnect
+from fastapi.responses import HTMLResponse
+
+logging.basicConfig(
+    level=logging.DEBUG,  # Use DEBUG or INFO for dev
+    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
+    datefmt="%Y-%m-%d %H:%M:%S",
+)
+logger = logging.getLogger(__name__)
+app = FastAPI()
+
+
+class ConnectionManager:
+    def __init__(self):
+        self.active_connections: list[WebSocket] = []
+        self.logger = logging.getLogger(self.__class__.__name__)
+
+    async def connect(self, websocket: WebSocket):
+        await websocket.accept()
+        self.logger.info("connect")
+        self.active_connections.append(websocket)
+
+    def disconnect(self, websocket: WebSocket):
+        self.logger.info("disconnect")
+        self.active_connections.remove(websocket)
+
+    async def send_personal_message(self, message: str, websocket: WebSocket):
+        self.logger.info("send_personal_message")
+        print(f"send_personal_message: {message}")
+        await websocket.send_text(f"personal: {message}")
+
+    async def broadcast(self, message: str):
+        self.logger.info("broadcast")
+        for connection in self.active_connections:
+            await connection.send_text(message)
+
+
+manager = ConnectionManager()
+manager2 = ConnectionManager()
+
+@app.get("/")
+async def get():
+    with open("html/index.html") as html_index:
+        html = html_index.read()
+        return HTMLResponse(html)
+
+
+@app.websocket("/ws/{client_id}")
+async def websocket_endpoint(websocket: WebSocket, client_id: int):
+    await manager.connect(websocket)
+    try:
+        while True:
+            data = await websocket.receive_text()
+            logger.info(f"channel 1: {data}")
+            await manager.send_personal_message(f"You wrote: {data}", websocket)
+            await manager.broadcast(f"Client #{client_id} says: {data}")
+    except WebSocketDisconnect:
+        manager.disconnect(websocket)
+        await manager.broadcast(f"Client #{client_id} left the chat")
+
+@app.websocket("/ws1/{client_id}")
+async def websocket_endpoint1(websocket: WebSocket, client_id: int):
+    await manager2.connect(websocket)
+    try:
+        while True:
+            data = await websocket.receive_text()
+            logger.info(f"channel 2: {data}")
+            await manager2.send_personal_message(f"You wrote: {data}", websocket)
+            await manager2.broadcast(f"Client #{client_id} says: {data}")
+    except WebSocketDisconnect:
+        manager2.disconnect(websocket)
+        await manager2.broadcast(f"Client #{client_id} left the chat")