python - 存储事件后，异步事件等待函数不会继续

Question

我正在开发一个功能，该功能应该在通过 websocket 给出响应后继续进行。为了实现这一点，我使用了一个异步事件，该事件将在给出响应后设置。

涉及三个函数:

async def send(self, message):
    await self.channel(message.toJSON())
    if (message.method == 'get' or message.method == 'post'):
        event = asyncio.Event()
        self._queueMessage(message, event)
        await event.wait()
    print('continue')

def _queueMessage(self, message, event):
    self.queue.append([message, event])

def _process_response_message(self, message):
    for entry in self.queue:
        if (message['_id'] == entry[0]._id):
            print(entry[1])
            entry[1].set()
            print(entry[1])
            return

返回:

<asyncio.locks.Event object at 0x7f3a1ff20da0 [unset,waiters:1]>
<asyncio.locks.Event object at 0x7f3a1ff20da0 [set,waiters:1]>

在这个例子中， print('continue') 函数从未被调用，我也不知道为什么，因为 .set() 函数实际上被调用，并且 .set() 似乎工作正常，如果我在调用之前使用它等待事件.wait()。

我有什么遗漏吗？

Answer 1

根据您收到的消息_process_response_message似乎正在另一个线程中运行。 asyncio.Event 不是线程安全对象 should use loop.call_soon_threadsafe 函数从其他线程调用它的方法。尝试像这样更改您的代码:

async def send(self, message):
    await self.channel(message.toJSON())
    if (message.method == 'get' or message.method == 'post'):
        loop = asyncio.get_event_loop()
        event = asyncio.Event()
        self._queueMessage(message, loop, event)
        await event.wait()
    print('continue')

def _queueMessage(self, message, loop, event):
    self.queue.append([message, loop, event])

def _process_response_message(self, message):
    for entry in self.queue:
        qmsg, loop, event = entry
        if (message['_id'] == qmsg._id):
            loop.call_soon_threadsafe(event.set)
            return