46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
"""Gestionnaire de connexions WebSocket."""
|
|
|
|
import json
|
|
import logging
|
|
from typing import Any
|
|
|
|
from fastapi import WebSocket
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class WSManager:
|
|
def __init__(self):
|
|
self.active: list[WebSocket] = []
|
|
|
|
async def connect(self, ws: WebSocket):
|
|
await ws.accept()
|
|
self.active.append(ws)
|
|
logger.debug("WS connecté (%d total)", len(self.active))
|
|
|
|
def disconnect(self, ws: WebSocket):
|
|
self.active.discard(ws) if hasattr(self.active, "discard") else None
|
|
if ws in self.active:
|
|
self.active.remove(ws)
|
|
logger.debug("WS déconnecté (%d restants)", len(self.active))
|
|
|
|
async def broadcast(self, data: dict[str, Any]):
|
|
if not self.active:
|
|
return
|
|
msg = json.dumps(data, default=str)
|
|
dead = []
|
|
for ws in self.active:
|
|
try:
|
|
await ws.send_text(msg)
|
|
except Exception:
|
|
dead.append(ws)
|
|
for ws in dead:
|
|
self.disconnect(ws)
|
|
|
|
async def send(self, ws: WebSocket, data: dict[str, Any]):
|
|
try:
|
|
await ws.send_text(json.dumps(data, default=str))
|
|
except Exception as e:
|
|
logger.debug("Erreur envoi WS: %s", e)
|
|
self.disconnect(ws)
|