[ADD] more info if websocket failed

Author: shurshilov

routers/checkup.py

@@ -115,14 +115,13 @@ async def checkup(req: Request):
             "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,
             "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,
         }
         if not req.app.state.websocket_client.connected:
             result["status"]["checkup_websocket"] = "error"
-        if req.app.state.websocket_client.disconnected_reason:
-            result["info"]["checkup_websocket"][
-                "disconnected_reason"
-            ] = req.app.state.websocket_client.disconnected_reason
     except Exception as exc:
         result["info"]["checkup_websocket"]["error"] = str(exc)
         result["status"]["checkup_websocket"] = "error"

services/websocket.py

@@ -28,8 +28,10 @@ class WebsocketEvents:
     no_answer_last_message_time = ""
     no_answer_last_message = {}
     last_connected_time = ""
+    last_try_connected_time = ""
     connected = False
     disconnected_reason = ""
+    disconnected_time = ""
 
     def __init__(
         self,
@@ -187,6 +189,7 @@ async def start_consumer(self):
             }
         """
         try:
+            self.last_try_connected_time = str(datetime.datetime.now())
             async with websockets.connect(self.websocket_url) as websocket:
                 self.connected = True
                 self.last_connected_time = str(datetime.datetime.now())
@@ -218,5 +221,6 @@ async def start_consumer(self):
             log.exception("Unknown start_consumer error: %s", exc)
             self.connected = False
             self.disconnected_reason = str(exc)
+            self.disconnected_time = str(datetime.datetime.now())
             await asyncio.sleep(self.timeout)
             await self.start_consumer()