fix trade ws reconnecting error

jackhh66 · Jul 25, 2022 · 3e6715f · 3e6715f
1 parent b58ea07
commit 3e6715f
Show file tree

Hide file tree

Showing 5 changed files with 16 additions and 1 deletion.
diff --git a/howtrader/__init__.py b/howtrader/__init__.py
@@ -1 +1 @@
-__version__ = "3.2.3"
+__version__ = "3.2.4"
diff --git a/howtrader/api/websocket/websocket_client.py b/howtrader/api/websocket/websocket_client.py
@@ -141,6 +141,9 @@ def on_error(
         except Exception:
             traceback.print_exc()
 
+    def on_exit_loop(self):
+        pass
+
     def exception_detail(
         self,
         exception_type: type,
@@ -193,6 +196,9 @@ async def _run(self):
             except Exception:
                 et, ev, tb = sys.exc_info()
                 self.on_error(et, ev, tb)
+                break
+
+        self.on_exit_loop()
 
     def _record_last_sent_text(self, text: str):
         """record the last send text for debugging"""

diff --git a/howtrader/gateway/binance/binance_inverse_gateway.py b/howtrader/gateway/binance/binance_inverse_gateway.py
@@ -1053,6 +1053,9 @@ def on_packet(self, packet: dict) -> None:
         elif packet['e'] == 'listenKeyExpired':
             self.gateway.rest_api.start_user_stream()
 
+    def on_exit_loop(self):
+        self.gateway.rest_api.start_user_stream()
+
     def on_account(self, packet: dict) -> None:
         """account data update"""
         for acc_data in packet["a"]["B"]:

diff --git a/howtrader/gateway/binance/binance_spot_gateway.py b/howtrader/gateway/binance/binance_spot_gateway.py
@@ -905,6 +905,9 @@ def on_packet(self, packet: dict) -> None:
         elif packet['e'] == 'listenKeyExpired':
             self.gateway.rest_api.start_user_stream()
 
+    def on_exit_loop(self):
+        self.gateway.rest_api.start_user_stream()
+
     def on_account(self, packet: dict) -> None:
         """account data update"""
         for d in packet["B"]:

diff --git a/howtrader/gateway/binance/binance_usdt_gateway.py b/howtrader/gateway/binance/binance_usdt_gateway.py
@@ -1062,6 +1062,9 @@ def on_packet(self, packet: dict) -> None:
         elif packet['e'] == 'listenKeyExpired':
             self.gateway.rest_api.start_user_stream()
 
+    def on_exit_loop(self):
+        self.gateway.rest_api.start_user_stream()
+
     def on_account(self, packet: dict) -> None:
         """account data update"""
         for acc_data in packet["a"]["B"]: