26 lines
704 B
Python
26 lines
704 B
Python
import asyncio
|
|
from fastapi import WebSocket
|
|
|
|
|
|
class WebSocketConnectionManager:
|
|
def __init__(self) -> None:
|
|
self.__active_connections: list[WebSocket] = []
|
|
|
|
async def connect(self, ws: WebSocket):
|
|
await ws.accept()
|
|
self.__active_connections.append(ws)
|
|
|
|
def disconnect(self, ws: WebSocket):
|
|
self.__active_connections.remove(ws)
|
|
|
|
async def send_text(self, ws: WebSocket, msg: str):
|
|
try:
|
|
await ws.send_text(msg)
|
|
except:
|
|
self.__active_connections.remove(ws)
|
|
|
|
async def broadcast(self, msg: str):
|
|
await asyncio.gather(
|
|
*[self.send_text(conn, msg) for conn in self.__active_connections]
|
|
)
|