用Python3的async/await做异步编程

前年我曾写过一篇《初探 Python 3 的异步 IO 编程》,当时只是初步接触了一下 yield from 语法和 asyncio 标准库。前些日子我在 V2EX 看到一篇《为什么只有基于生成器的协程可以真正的暂停执行并强制性返回给事件循环?》,激起了我再探 Python 3 异步编程的兴趣。然而看了很多文章和,才发现极少提到 async 和 await 实际意义的,绝大部分仅止步于对 asyncio 库的使用,真正有所帮助的只有《How the heck does async/await work in Python 3.5?》和《A tale of event loops》这两篇。

在接着写下去之前,我先列举一些 PEPs 以供参考:

  • PEP 255 — Simple Generators
  • PEP 342 — Coroutines via Enhanced Generators
  • PEP 380 — Syntax for Delegating to a Subgenerator
  • PEP 492 — Coroutines with async and await syntax
  • PEP 525 — Asynchronous Generators

从这些 PEPs 中可以看出 Python 生成器 / 协程的发展历程:先是 PEP 255 引入了简单的生成器;接着 PEP 342 赋予了生成器 send() 方法,使其可以传递数据,协程也就有了实际意义;接下来,PEP 380 增加了 yield from 语法,简化了调用子生成器的语法;然后,PEP 492 将协程和生成器区分开,使得其更不易被用错;最后,PEP 525 提供了异步生成器,使得编写异步的数据产生器得到简化。

本文将简单介绍一下这些 PEPs,着重深入的则是 PEP 492。

首先提一下生成器(generator)。

Generator function 是函数体里包含 yield 表达式的函数,它在调用时生成一个 generator 对象(以下将其命名为 gen)。第一次调用 next(gen) 或 gen.send(None) 时,将进入它的函数体:在执行到 yield 表达式时,向调用者返回数据;当函数返回时,抛出 StopIteration 异常。在该函数未执行完之前,可再次调用 next(gen) 进入函数体,也可调用 gen.send(value) 向其传递参数,以供其使用(例如提供必要的数据,或者控制其行为等)。

由于它主要的作用是产生一系列数据,所以一般使用 for … in gen 的语法来遍历它,以简化 next() 的调用和手动捕捉 StopIteration 异常。

即:

用Python3的async/await做异步编程

可以简化为:

for value in gen: process(value)

由于生成器提供了再次进入一个函数体的机制,其实它已经可以当成协程来使用了。

写个很简单的例子:

用Python3的async/await做异步编程

这里的 coroutine 函数用于处理连接和收发数据,而 main 函数则等待读写和传递参数。虽然看上去和同步的调用没啥区别,但其实在 main 函数中可以同时执行多个 coroutine,以实现并发执行。

再说一下 yield from。

如果一个生成器内部需要遍历另一个生成器,并将数据返回给调用者,你需要遍历它并处理所遇到的异常;而用了 yield from 后,则可以一行代码解决这些问题。具体例子就不列出了,PEP 380 里有详细的代码。

这对于协程而言也是一个利好,这使得它的调用也得到了简化:

用Python3的async/await做异步编程

接下来就该轮到协程(coroutine)登场了。

从上文也可看出,调用 yield from gen 时,我无法判定我是遍历了一个生成器,还是调用了一个协程,这种混淆使得接口的设计者和使用者需要花费额外的工夫来约定和检查。

于是 Python 又先后添加了 asyncio.coroutine 和 types.coroutine 这两个装饰器来标注协程,这样就使得需要使用协程时,不至于误用了生成器。顺带一提,前者是 asyncio 库的实现,需要保持向下兼容,本文暂不讨论;后者则是 Python 3.5 的语言实现,实际上是给函数的 code.co_flags 设置 CO_ITERABLE_COROUTINE 标志。随后,async def 也被引入以用于定义协程,它则是设置 CO_COROUTINE 标志。

至此,协程和生成器就得以区分,其中以 types.coroutine 定义的协程称为基于生成器的协程(generator-based coroutine),而以 async def 定义的协程则称为原生协程(native coroutine)。

这两种协程之间的区别其实并不大,非要追究的话,主要有这些:

  • 原生协程里不能有 yield 或 yield from 表达式。
  • 原生协程被垃圾回收时,如果它从来没被使用过(即调用 await coro 或 coro.send(None)),会抛出 RuntimeWarning。
  • 原生协程没有实现 iternext 方法。
  • 简单的生成器(非协程)不能 yield from 原生协程
  • 对原生协程及其函数分别调用 inspect.isgenerator() 和 inspect.isgeneratorfunction() 将返回 False。

实际使用时,如果不考虑向下兼容,可以都用原生协程,除非这个协程里用到了 yield 或 yield from 表达式。

定义了协程函数以后,就可以调用它们了。

PEP 492 也引入了一个 await 表达式来调用协程,它的用法和 yield from 差不多,但是它只能在协程函数内部使用,且只能接 awaitable 的对象。

所谓 awaitable 的对象,就是其 await 方法返回一个迭代器的对象。原生协程和基于生成器的协程都是 awaitable 的对象。

另一种调用协程的方法则和生成器一样,调用其 send 方法,并自行迭代。这种方式主要用于在非协程函数里调用协程。

举例来说,调用的代码会类似这样:

用Python3的async/await做异步编程

其中 generator_coroutine 函数里因为用到了 yield 表达式,所以只能定义成基于生成器的协程;native_coroutine 函数由于自身是协程,可以直接用 await 表达式调用其他协程;main 函数由于不是协程,因而需要用 native_coroutine().send(None) 这种方式来调用协程。

这个例子其实也解释了 V2EX 里提到的那个问题,即为什么原生协程不能「真正的暂停执行并强制性返回给事件循环」。

假设事件循环在 main 函数里,原生协程是 native_coroutine 函数,那要怎么才能让它暂停并返回 main 函数呢?

很显然 await generator_coroutine() 是不行的,这会进入 generator_coroutine 的函数体,而不是回到 main 函数;如果 yield 一个值,又会遇到之前提到的一个限制,即原生协程里不能有 yield 表达式;最后仅剩 return 或 raise 这两种选择了,但它们虽然能回到 main 函数,却也不是「暂停」,因为再也没法「继续」了。

所以一般而言,如果要用 Python 3.5 来做异步编程的话,最外层的事件循环需要调用协程的 send 方法,里面大部分的异步方法都可以用原生协程来实现,但最底层的异步方法则需要用基于生成器的协程。

为了有个更直观的认识,再来举个例子,抓取 10 个百度搜索的页面:

用Python3的async/await做异步编程

其他的函数都没什么好说的,主要解释下 until_readable、until_writable 和 main 函数。

其实 until_readable 和 until_writable 函数都是 yield 一个 (fileobj, event) 元组,用于告知事件循环,这个 fileobj 的 event 事件需要被监听。

而在 main 函数中,事件循环遍历并执行 tasks 里包含的协程。这些协程在等待 IO 时返回事件循环,由事件循环注册事件及其对应的协程。到下一个事件循环时,取出所有就绪的事件,继续执行其对应的协程,就完成了整个的异步执行过程。

如果关注到 fetch_url 函数,就会发现业务逻辑用到的代码其实挺简单,只是 await 异步函数而已。这虽然简化了大部分的开发工作,但其实也限制了它的表达能力,因为在一个协程内,不能同时 await 多个异步函数——它实际上是顺序执行的,只是不同协程之间可以异步执行而已。

考虑一个 HTTP/2 的客户端,它和服务器之间的连接是多路复用的,也就是可以在一个连接里同时发出和接收多份数据,而这些数据的传输是乱序的。如果一份 JavaScript 资源已经下载完毕,没必要再等其他的图片资源下载完毕才能执行。要做到这点,就需要协程有并发执行多个子协程,共同完成任务的能力。这在使用多线程或回调函数时是很容易做到的,但使用 await 就显得捉襟见肘了。倒也不是不能做,只是需要拿之前的代码改下,yield 一些子协程,并在事件循环中判断一下类型就行了。

虽然仅用上述提到的东西,已经能做异步编程了,但我还是得补充 2 个漏掉的语法知识:

1. async with

先考虑普通的 with 语句,它的主要作用是在进入和退出一个区域时,做一些初始化和清理工作。

例如:

用Python3的async/await做异步编程

就可以改写为:

with open('1.txt') as f: content = f.read()

这里要求 open 函数返回的 f 对象带有 enterexit 方法。其中,enter 方法只需要返回一个文件对象就行了,exit 则需要调用这个文件对象的 close 方法。类似的,假设这个 open 函数和 close 方法变成了异步的,你的代码可能是这样的:

用Python3的async/await做异步编程

你就可以用 async with 来改写它:

async with async_open('1.txt') as f: content = await f.read()

相应的,async_open 函数返回的 f 对象需要实现 aenteraexit 这 2 个异步方法。

2.async for

这里也先考虑普通的 for 语句,它的主要作用是遍历一个迭代器。

例如:

用Python3的async/await做异步编程

可以改写为:

f = open('1.txt')for line in f: print(line)

这里要求 open 函数返回的 f 对象返回一个迭代器对象,即实现了 iter 方法,这个方法要返回一个实现了 next 方法的对象。而 next 方法在每次调用时,都返回下一行的文件内容,直到文件结束时抛出 StopIteration 异常。类似的,假如 next 方法变成了异步的,你的代码可能是这样的:

用Python3的async/await做异步编程

你可以用 async with 来改写它:

f = open('1.txt')async for line in f: print(line)

相应的,所需实现的方法分别变成了 aiteranext。其中,后者是异步方法。顺带一提,PEP 525 引入的异步生成器(asynchronous generator)就实现了这两个方法。在异步方法中使用 yield 表达式,会将它变成异步生成器函数(Python 3.6 以后可用,3.5 之前是语法错误)。值得注意的是,异步生成器没有实现 await 方法,因此它不是协程,也不能被 await。

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 216,372评论 6 498
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,368评论 3 392
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 162,415评论 0 353
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,157评论 1 292
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,171评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,125评论 1 297
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,028评论 3 417
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,887评论 0 274
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,310评论 1 310
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,533评论 2 332
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,690评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,411评论 5 343
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,004评论 3 325
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,659评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,812评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,693评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,577评论 2 353

推荐阅读更多精彩内容

  • 前年我曾写过一篇《初探 Python 3 的异步 IO 编程》,当时只是初步接触了一下 yield from 语法...
    星星在线阅读 2,811评论 0 2
  • 转一篇驹神的关于异步编程和Asyncio的文章。这是上篇,共三篇。原文地址:http://aju.space/20...
    SeanCheney阅读 9,793评论 2 99
  • 1 什么是异步编程 通过学习相关概念,我们逐步解释异步编程是什么。 1.1 阻塞 程序未得到所需计算资源时被挂起的...
    hugoren阅读 2,654评论 2 10
  • 上篇 中篇 下篇 1 什么是异步编程 1.1 阻塞 程序未得到所需计算资源时被挂起的状态。 程序在等待某个操作完成...
    秦时明星阅读 998评论 0 3
  • 爱车有我,一切无忧! 活动大优惠开始啦 到湘约酒家吃饭,老板送洗车,免费免费 车在洗对面洗,你在享受午餐晚餐 多一...
    爱洁士洗车店阅读 260评论 0 0