fix: closes #730

server.py

@@ -48,24 +48,21 @@ async def broadcast_handler(websocket: websockets.WebSocketServerProtocol):
                 pass
 
             # Broadcast to all OTHER clients
-            if not clients:
-                continue
-                
             disconnected = set()
-            # Create broadcast tasks for efficiency
+            # Create broadcast tasks paired with their target client
-            tasks = []
+            task_client_pairs = []
             for client in clients:
                 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 = [t for t, _ in task_client_pairs]
                 results = await asyncio.gather(*tasks, return_exceptions=True)
-                for i, result in enumerate(results):
+                for (task, client), result in zip(task_client_pairs, results):
                     if isinstance(result, Exception):
-                        # Find the client that failed
+                        logger.error(f"Failed to send to a client {client.remote_address}: {result}")
-                        target_client = [c for c in clients if c != websocket][i]
+                        disconnected.add(client)
-                        logger.error(f"Failed to send to a client {target_client.remote_address}: {result}")
-                        disconnected.add(target_client)
             
             if disconnected:
                 clients.difference_update(disconnected)