我有一个类,其方法如下:
# 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
?
答案 0 :(得分:6)
有没有办法可以在
asyncio.Event
上等待超时,类似于threading.Event
?
asyncio.wait_for
可以方便地为任何协同程序添加超时。使用超时模拟threading.Event.wait
可能如下所示:
async def event_wait(evt, timeout):
try:
await asyncio.wait_for(evt.wait(), timeout)
except asyncio.TimeoutError:
pass
return evt.is_set()
这允许run
几乎与使用threading.Event
:
async def run(self):
while not await event_wait(self.stop_event, 3):
pass # do stuff
完全相同
{{1}}