优化代码

comsumer/order.py

@@ -16,12 +16,12 @@ 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()
+    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}")
     await kafka_service.send_to_kafka("binance_future_order_update", json.dumps(msg),key="DEFAULT",send_now=True)