Question

我编写了django通道代码，以通过websocket从两个不同的源异步发送api数据。不同的源需要几秒钟到一分钟的时间来计算和发送回数据。我设法使用asyncio事件循环异步调用它们。但是问题在于他们没有异步地将响应发送回去。该代码只是等待所有数据到达并同时发送所有内容。

渠道代码：

class SearchHotelConsumer(AsyncWebsocketConsumer):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.source1 = Source1()
        self.source2 = Source2()

    async def connect(self):
        await self.accept()

    def disconnect(self, close_code):
        pass

    async def _source1_handler(self, request, queue):
        source1_response = await self.source1.async_post(request)

        await queue.put(source1_response.data)

    async def _source2_handler(self, request, queue):
        source2_response = await self.source2.async_post(request)

        await queue.put(source2_response.data)

    async def _send_msg(self, queue):
        while True:
            message = await queue.get()
            if message is None:
                break
            print('got the message')
            await self.send(text_data=json.dumps({
                'message': message
            }, cls=DjangoJSONEncoder))

            queue.task_done()

    def receive(self, text_data):
        text_data_json = json.loads(text_data)
        message = text_data_json['message']

        request = HttpRequest()
        request.method = 'POST'
        request.session = self.scope["session"]
        request = Request(request)
        for key, val in message.items():
            request.data[key] = val

        queue = asyncio.Queue()
        sender = asyncio.ensure_future(self._send_msg(queue))

        await self._source1_handler(request, queue)
        await self._source2_handler(request, queue)

        await queue.join()

        sender.cancel()

如何使消息发送部分真正异步？

Answer 1

您未显示的async_post代码可能是同步的。

如果您想要异步请求，请参见以下答案：

https://stackoverflow.com/a/22414756/10840818

要将请求（或任何其他阻止库）与asyncio一起使用，可以使用BaseEventLoop.run_in_executor在另一个线程中运行一个函数，并从另一个线程中产生一个函数来获取结果。

或者是实现HTTP获取异步方式的另一种方法

async with aiohttp.ClientSession() as session:
        async with session.get(url, headers=HEADERS, params=params) as resp:
            data = await resp.json()

Django Channels Websocket通信不是真正异步的吗？

1 个答案: