python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > Python asyncio

Python asyncio常用函数使用详解

作者:拾牙慧者

Asyncio在经过一段时间的发展以及获取Curio等第三方库的经验来提供更多的功能,目前高级功能也基本完善,但是相对于其他语言,Python的Asyncio高级功能还是不够的,但好在Asyncio的低级API也比较完善

协程的定义

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

若:

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

此时,get_event_loop()相当于:

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

到此这篇关于Python asyncio常用函数使用详解的文章就介绍到这了,更多相关Python asyncio内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
阅读全文