[ADD] more info if websocket failed

Author: shurshilov

routers/checkup.py

@@ -114,11 +114,12 @@ async def checkup(req: Request):
             "answer_last_message": req.app.state.websocket_client.answer_last_message,
             "no_answer_last_message_time": req.app.state.websocket_client.no_answer_last_message_time,
             "no_answer_last_message": req.app.state.websocket_client.no_answer_last_message,
-            "last_connected_time": req.app.state.websocket_client.last_connected_time,
             "last_try_connected_time": req.app.state.websocket_client.last_try_connected_time,
+            "last_connected_time": req.app.state.websocket_client.last_connected_time,
             "connected": req.app.state.websocket_client.connected,
             "disconnected_time": req.app.state.websocket_client.disconnected_time,
             "disconnected_reason": req.app.state.websocket_client.disconnected_reason,
+            "disconnect_count": req.app.state.websocket_client.disconnect_count,
         }
         if not req.app.state.websocket_client.connected:
             result["status"]["checkup_websocket"] = "error"

services/websocket.py

@@ -32,6 +32,7 @@ class WebsocketEvents:
     connected = False
     disconnected_reason = ""
     disconnected_time = ""
+    disconnect_count = 0
 
     def __init__(
         self,
@@ -220,6 +221,7 @@ async def start_consumer(self):
         except Exception as exc:
             log.exception("Unknown start_consumer error: %s", exc)
             self.connected = False
+            self.disconnect_count += 1
             self.disconnected_reason = str(exc)
             self.disconnected_time = str(datetime.datetime.now())
             await asyncio.sleep(self.timeout)