From 867802b2a705cc329e0604c7f8eca1bdd3397fcc Mon Sep 17 00:00:00 2001 From: algizn97 Date: Mon, 15 Dec 2025 21:57:13 +0500 Subject: [PATCH] Fixed the websocket --- app/bybit/web_socket.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/app/bybit/web_socket.py b/app/bybit/web_socket.py index 225cae5..cd438c0 100644 --- a/app/bybit/web_socket.py +++ b/app/bybit/web_socket.py @@ -14,7 +14,6 @@ logger = logging.getLogger("web_socket") class CustomWebSocket(WebSocket): def _on_error(self, error): logger.error(f"WebSocket error: {error}") - # Здесь можно добавить собственную логику, например уведомления, метрики и т.д. return super()._on_error(error) @@ -72,6 +71,13 @@ class WebSocketBot: async def clear_user_sockets(self): """Clear the user_sockets and user_messages dictionaries.""" + for tg_id, ws in list(self.user_sockets.items()): + try: + if ws and hasattr(ws, 'close'): + await ws.close() + except Exception as e: + logger.error(f"Error closing WS for {tg_id}: {e}") + self.user_sockets.clear() self.user_messages.clear() self.user_keys.clear() @@ -85,8 +91,7 @@ class WebSocketBot: testnet=False, channel_type="private", api_key=api_key, - api_secret=api_secret, - ping_interval=20, + api_secret=api_secret ) self.user_sockets[tg_id] = self.ws_private