首页  >  文章  >  后端开发  >  如何使用Python的asyncio常用函数?

如何使用Python的asyncio常用函数?

WBOY
WBOY转载
2023-04-26 20:10:061639浏览

协程的定义

需要使用 async def 语句

协程可以做哪些事:

1、等待一个future结果

2、等待另一个协程(产生一个结果或引发一个异常)

3、产生一个结果给正在等它的协程

4、引发一个异常给正在等它的协程

协程的运行

调用协程函数,协程不会开始运行,只是返回一个协程对象

要让协程对象运行有两种方式:

1、在另一个已经运行的协程中用await等待它

2、通过ensure_future函数计划它的执行

只有某线程的loop运行了,协程才可能运行

下面的例子:

先拿到当前线程缺省的loop,然后将协程对象交给loop.run_until_complete,协程对象随后会在loop里得到运行

loop = asyncio.get_event_loop()
loop.run_until_complete(do_some_work(3))

run_until_complete是一个阻塞调用,知道协程运行结束才返回

它的参数是一个future,但是我们传给它的却是协程对象,它在内部做了检查,通过ensure_future函数把协程对象包装成了future

我们可以这样写:

loop.run_until_complete(asyncio.ensure_future(do_some_work(3)))

多个协程运行

多个协程在一个loop里运行,为了把多个协程交给loop,需要借助asyncio.gathre函数

loop.run_until_complete(asyncio.gather(do_some_work(1), do_some_work(3)))

或者先将协程对象存到list中,这种比较常见

loop = asyncio.get_event_loop() #获取当前线程loop
coros_list = []
for i in range(2000):
    coros_list.append(main(i))
loop.run_until_complete(asyncio.gather(*coros_list))

gather 起聚合的作用,把多个 futures 包装成单个 future,因为 loop.run_until_complete 只接受单个 future。

关于loop.close()

简单来说,loop 只要不关闭,就还可以再运行。:

loop = asyncio.get_event_loop() #获取当前线程loop
loop.run_until_complete(do_some_work(loop, 1))
loop.run_until_complete(do_some_work(loop, 3))
loop.close()

但是如果关闭了,就不能再运行了:

loop = asyncio.get_event_loop() #获取当前线程loop
loop.run_until_complete(do_some_work(loop, 1))
loop.close()
loop.run_until_complete(do_some_work(loop, 3))  # 此处异常

回调

加入协程是一个IO的读操作,等他读完数据后,我们希望得到通知,以便下一步数据的处理。这个可以向future添加回调实现

def done_callback(futu):
    print('Done')
futu = asyncio.ensure_future(do_some_work(3))
futu.add_done_callback(done_callback)
loop.run_until_complete(futu)

事件循环

事件循环会运行异步任务和回调,执行网络IO操作,以及运行子进程

从asyncio event loop policy文档,我们得知, event loop policy是一个进程全局对象,控制对该进程内所有event loop的管理。

进程的全局policy定义了该policy管控的context的含义,在每个context中管理分开独立的event loop. 默认的policy定义的context就是当前的线程, 也就是说不同的线程是不同的context,因此有不同的event loop。

获取事件循环

asyncio.get_running_loop() # 返回当前os线程中正在运行的事件循环
asyncio.get_event_loop() # 获取当前事件循环
asyncio.set_event_loop(loop) # 获取当前事件循环
asyncio.new_event_loop() # 创建并返回一个新的事件循环对象

asyncio.get_event_loop()

若:

  • 当前线程为主线程

  • 当前线程没有启动event loop

调用asyncio.get_event_loop()方法会生成一个新的默认event loop,并设置为当前线程的事件循环。

此时,get_event_loop()相当于:

loop = asyncio.new_event_loop()
asyncio.set_event_loop(loop)

以上是如何使用Python的asyncio常用函数?的详细内容。更多信息请关注PHP中文网其他相关文章!

声明:
本文转载于:yisu.com。如有侵权,请联系admin@php.cn删除