python里使用Future对象进行回调别的函数

最新推荐文章于 2024-07-10 03:22:38 发布

caimouse

最新推荐文章于 2024-07-10 03:22:38 发布

阅读量3.8k

点赞数 3

分类专栏： milang(小语）文章标签： python tensorflow asyncio

本文链接：https://blog.csdn.net/caimouse/article/details/77869636

版权

milang(小语）专栏收录该内容

389 篇文章 7 订阅

订阅专栏

Future对象也可以像协程一样，当它设置完成结果时，就可以立即进行回调别的函数，例子如下：

import asyncio
import functools


def callback(future, n):
    print('{}: future done: {}'.format(n, future.result()))


async def register_callbacks(all_done):
    print('registering callbacks on future')
    all_done.add_done_callback(functools.partial(callback, n=1))
    all_done.add_done_callback(functools.partial(callback, n=2))


async def main(all_done):
    await register_callbacks(all_done)
    print('setting result of future')
    all_done.set_result('the result')


event_loop = asyncio.get_event_loop()
try:
    all_done = asyncio.Future()
    event_loop.run_until_complete(main(all_done))
finally:
    event_loop.close()

输出结果如下：

registering callbacks on future
setting result of future
1: future done: the result
2: future done: the result

在这个例子里，先调用函数add_done_callback（）来注册一个回调函数，由于只支持一个参数，使用functools.partial来作一个封装。当set_result（）函数调用之后，就立即进行回调函数的运行。