我是异步编程和python的新手,但我想在状态机中实现异步函数 . 要求如下:
异步任务必须在任何时间点都可取消
异步任务不得阻止执行的主线程
异步任务必须在完成后调用回调 .
为了实现目标1和2,我已经成功地实现了我的异步例程,以便在执行程序内运行,如下所述:asyncio: Is it possible to cancel a future been run by an Executor? . 在调试时,我看到我也完成了#2,因为原始线程继续成功 . 但是,我很难实现回调函数 . 请注意,此回调函数必须在原始线程上运行,因为它将更改分配给它的对象的状态 . 我怎么能做到这一点?
我的代码:
class Free(State):
def write(self, manager):
write_future = self.write_async(manager)
self.set_write_future(write_future)
self.change_state(manager, busy_state)
def write_async(self, manager):
event = threading.Event()
write_future = asyncio.get_event_loop().run_in_executor(None, self.async_write, 10, event)
write_future.add_done_callback(functools.partial(self.async_write_callback, manager))
return event
def async_write(self, seconds_to_block, event):
for i in range(seconds_to_block):
if event.is_set():
return
print('writing {}/{}'.format(i, seconds_to_block))
time.sleep(1)
print('done writing {}'.format(seconds_to_block))
def async_write_callback(self, manager):
#never gets called
self.terminate_future()
self.change_state(manager, free_state)