等待为 asyncio.Event 设置超时或事件

Question

我有一个类，其方法如下所示：

# self.stop_event -> threading.Event
def run(self):
    while not self.stop_event.wait(3):  # i.e. repeat every 3 sec
        pass  # do stuff

想法是其中几个在它们自己的线程中运行，并且在某些时候一个线程执行stop_event.set()，这自然会停止所有其他线程。我想为此切换到 asyncio，因为run 中的任务主要是在睡觉和做 IO。因此，我必须：

# self.stop_event -> asyncio.Event
async def run(self):
    while not self.stop_event.is_set():
        await asyncio.sleep(3)
        pass  # do stuff

问题是asyncio.Event是不能等待的，所以设置的时候，在方法完成之前最多有3秒的等待时间。这是一个问题，因为睡眠时间可能是几分钟。目前，我正在通过将run 包装在asyncio.Task 中，然后像event_loop.call_soon(the_task.cancel) 一样取消它来解决这个问题。

我想问是否有更好的方法来实现上述？有没有办法让asyncio.Event 以某种方式超时，类似于threading.Event？

Answer 1

有没有一种方法可以让我以某种方式等待asyncio.Event 超时，类似于threading.Event？

asyncio.wait_for 支持方便地向任何等待的协程添加超时。 asyncio.Event 的 threading.Event.wait 超时功能的模拟可能如下所示：

async def event_wait(evt, timeout):
    # suppress TimeoutError because we'll return False in case of timeout
    with contextlib.suppress(asyncio.TimeoutError):
        await asyncio.wait_for(evt.wait(), timeout)
    return evt.is_set()

这允许run 与使用threading.Event 的几乎完全相同：

async def run(self):
    while not await event_wait(self.stop_event, 3):
        pass  # do stuff