优化代码

comsumer/order.py

@@ -16,11 +16,11 @@ async def start_bn_future_order_consumer(account_id, api_key, api_secret, testne
     client = await AsyncClient.create(api_key, api_secret, testnet=testnet)
     bm = BinanceSocketManager(client, max_queue_size=100000)
     ws = bm.futures_user_socket()
-    while True:
-        msg = await ws.recv()
-        msg['x-account-id'] = account_id
-        asyncio.create_task(send_msg(msg))
-                    
+    async with ws as stream:
+        while True:
+            msg = await ws.recv()
+            msg['x-account-id'] = account_id
+            asyncio.create_task(send_msg(msg))
 
 async def send_msg(msg):
     logger.info(f"send to kafka: {msg}")