Fixed the websocket

app/bybit/web_socket.py

@@ -11,6 +11,13 @@ logging.config.dictConfig(LOGGING_CONFIG)
 logger = logging.getLogger("web_socket")
 
 
+class CustomWebSocket(WebSocket):
+    def _on_error(self, error):
+        logger.error(f"WebSocket error: {error}")
+        # Здесь можно добавить собственную логику, например уведомления, метрики и т.д.
+        return super()._on_error(error)
+
+
 class WebSocketBot:
     """
     Class to handle WebSocket connections and messages.
@@ -43,9 +50,9 @@ class WebSocketBot:
                     continue
 
                 if tg_id in self.user_sockets:
-                    self.user_sockets.clear()
-                    self.user_messages.clear()
-                    self.user_keys.clear()
+                    self.user_sockets.pop(tg_id, None)
+                    self.user_messages.pop(tg_id, None)
+                    self.user_keys.pop(tg_id, None)
                     logger.info(
                         "Closed old websocket for user %s due to key change", tg_id
                     )
@@ -73,12 +80,13 @@ class WebSocketBot:
     async def try_connect_user(self, api_key, api_secret, tg_id):
         """Try to connect a user to the WebSocket."""
         try:
-            self.ws_private = WebSocket(
+            self.ws_private = CustomWebSocket(
                 demo=True,
                 testnet=False,
                 channel_type="private",
                 api_key=api_key,
                 api_secret=api_secret,
+                ping_interval=20,
             )
 
             self.user_sockets[tg_id] = self.ws_private
@@ -98,7 +106,7 @@ class WebSocketBot:
             return True
         except Exception as e:
             logger.error("Error connecting user %s: %s", tg_id, e)
-            return False
+            await asyncio.sleep(5)
 
     async def handle_order_update(self, message, tg_id):
         """Handle order updates."""