Python asyncio常用函数使用详解

目录

• 协程的定义
• 协程的运行
• 多个协程运行
• 关于loop.close()
• 回调
• 事件循环

协程的定义

需要使用 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)

原文地址：https://hanhandi.blog.csdn.net/article/details/129051033