INNER CODE UNIT · Python
_attempt_websocket_reconnect
runpod-workers/worker-comfyui · handler.py:116
def _attempt_websocket_reconnect(ws_url, max_attempts, delay_s, initial_error):
"""
Attempts to reconnect to the WebSocket server after a disconnect.
Args:
ws_url (str): The WebSocket URL (including client_id).
max_attempts (int): Maximum number of reconnection attempts.
delay_s (int): Delay in seconds between attempts.
initial_error (Exception): The error that triggered the reconnect attempt.
Returns:
websocket.WebSocket: The newly connected WebSocket object.
Raises:
websocket.WebSocketConnectionClosedException: If reconnection fails after all attempts.
"""
print(
f"worker-comfyui - Websocket connection closed unexpectedly: {initial_error}. Attempting to reconnect..."