Django频道websocket通信不是真正的异步?

2024-09-22 20:39:23 发布

您现在位置:Python中文网/ 问答频道 /正文

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

频道代码:

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()

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


Tags: 数据textselfjsonmessagedataasyncqueue
1条回答
网友
1楼 · 发布于 2024-09-22 20:39:23

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

如果您想要异步请求,请参阅以下答案:

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

To use requests (or any other blocking libraries) with asyncio, you can use BaseEventLoop.run_in_executor to run a function in another thread and yield from it to get the result.

或者是http get异步的替代方法

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

相关问题 更多 >