diff --git a/comsumer/order.py b/comsumer/order.py index 85c03c1..3e17dbd 100644 --- a/comsumer/order.py +++ b/comsumer/order.py @@ -16,13 +16,13 @@ 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}") await kafka_service.send_to_kafka("binance_future_order_update", json.dumps(msg),key="DEFAULT",send_now=True) - logger.info(f"send to kafka success: {msg}") \ No newline at end of file + logger.info(f"send to kafka success: {msg}")