diff options
| -rw-r--r-- | bot.py | 33 |
1 files changed, 18 insertions, 15 deletions
@@ -110,21 +110,24 @@ _main_ws = None # kept open across messages instead of reconnecting every time async def call_main(payload: str) -> tuple[bool, str]: global _main_ws - try: - if _main_ws is None or _main_ws.closed: - _main_ws = await websockets.connect(MAIN_URI, open_timeout=None, ping_timeout=None) - await _main_ws.send(payload) - raw = await _main_ws.recv() - data = json.loads(raw) - if "error" in data: - return False, data["error"] - return True, data["ok"] - except websockets.exceptions.ConnectionClosedOK: - _main_ws = None - return False, f"[{now_str()}] websocket closed cleanly (no reply)" - except Exception as e: - _main_ws = None - return False, f"[{now_str()}] websocket error: {e}" + for attempt in range(2): # retry once with a fresh connection if the cached one turned out dead + try: + if _main_ws is None: + _main_ws = await websockets.connect(MAIN_URI, open_timeout=None, ping_timeout=None) + await _main_ws.send(payload) + raw = await _main_ws.recv() + data = json.loads(raw) + if "error" in data: + return False, data["error"] + return True, data["ok"] + except websockets.exceptions.ConnectionClosedOK: + _main_ws = None + return False, f"[{now_str()}] websocket closed cleanly (no reply)" + except Exception as e: + _main_ws = None + if attempt == 0: + continue # one retry with a fresh connection before giving up + return False, f"[{now_str()}] websocket error: {e}" async def handle_error(error_msg: str, notify_channel=None, notify_user=None): print(error_msg) |