Python并发编程之asyncio、async/await

asyncio是Python 3.4版本引入的标准库，直接内置了对异步IO的支持。asyncio的编程模型就是一个消息循环。咱们从asyncio模块中直接获取一个EventLoop的引用，而后把须要执行的协程扔到EventLoop中执行，就实现了异步IO。例如：

# -*- coding: utf-8 -*-
import threading
import asyncio

@asyncio.coroutine
def hello():
    print('Hello world! (%s)' % threading.currentThread())
    yield from asyncio.sleep(1)
    print('Hello again! (%s)' % threading.currentThread())

loop = asyncio.get_event_loop()
tasks = [hello(), hello()]
loop.run_until_complete(asyncio.wait(tasks))
loop.close()

运行结果：

Hello world! (<_MainThread(MainThread, started 4589409728)>)
Hello world! (<_MainThread(MainThread, started 4589409728)>)
Hello again! (<_MainThread(MainThread, started 4589409728)>)
Hello again! (<_MainThread(MainThread, started 4589409728)>)

@asyncio.coroutine把一个generator标记为coroutine类型，而后，咱们就把这个coroutine扔到EventLoop中执行。

用asyncio提供的@asyncio.coroutine能够把一个generator标记为coroutine类型，而后在coroutine内部用yield from调用另外一个coroutine实现异步操做。

hello()会首先打印出Hello world!，而后，yield from语法能够让咱们方便地调用另外一个generator。因为asyncio.sleep()也是一个coroutine，因此线程不会等待asyncio.sleep()，而是直接中断并执行下一个消息循环。当asyncio.sleep()返回时，线程就能够从yield from拿到返回值（此处是None），而后接着执行下一行语句。

把asyncio.sleep(1)当作是一个耗时1秒的IO操做，在此期间，主线程并未等待，而是去执行EventLoop中其余能够执行的coroutine了，所以能够实现并发执行。

为了简化并更好地标识异步IO，从Python 3.5开始引入了新的语法async和await，其中：

把@asyncio.coroutine替换为async；

把yield from替换为await。   

上述例子能够改为：

# -*- coding: utf-8 -*-
import threading
import asyncio

async def hello():
    print("Hello world!")
    r = await asyncio.sleep(1)
    print("Hello again!")

loop = asyncio.get_event_loop()
tasks = [hello(), hello()]
loop.run_until_complete(asyncio.wait(tasks))
loop.close()