Compare commits

...

1 Commits

Author SHA1 Message Date
Alexander Whitestone
571475a749 fix: closes #716
Some checks failed
CI / test (pull_request) Failing after 9s
CI / validate (pull_request) Failing after 18s
Review Approval Gate / verify-review (pull_request) Failing after 3s
2026-04-12 12:44:37 -04:00

View File

@@ -52,19 +52,20 @@ async def broadcast_handler(websocket: websockets.WebSocketServerProtocol):
continue continue
disconnected = set() disconnected = set()
# Create broadcast tasks for efficiency # Create broadcast tasks, tracking which client each task targets
tasks = [] task_client_pairs = []
for client in clients: for client in clients:
if client != websocket and client.open: if client != websocket and client.open:
tasks.append(asyncio.create_task(client.send(message))) task = asyncio.create_task(client.send(message))
task_client_pairs.append((task, client))
if tasks:
if task_client_pairs:
tasks = [pair[0] for pair in task_client_pairs]
results = await asyncio.gather(*tasks, return_exceptions=True) results = await asyncio.gather(*tasks, return_exceptions=True)
for i, result in enumerate(results): for i, result in enumerate(results):
if isinstance(result, Exception): if isinstance(result, Exception):
# Find the client that failed target_client = task_client_pairs[i][1]
target_client = [c for c in clients if c != websocket][i] logger.error(f"Failed to send to client {target_client.remote_address}: {result}")
logger.error(f"Failed to send to a client {target_client.remote_address}: {result}")
disconnected.add(target_client) disconnected.add(target_client)
if disconnected: if disconnected: