Python 协程使用心得

基本概念

协程：又称微线程，纤程。英文名Coroutine。协程是一种子程序，它在执行过程中可以中断，然后转而执行别的子程序，在适当的时候再返回来接着执行。

注意：如程序内不需要中断，则不要定义成协程，以免增加调用不必要的麻烦。

协程的定义

定义协程有两种方式，一种是使用async def ，另一种是使用asyncio.coroutine修饰器，两种方法基本上一样。只不过在async def中需要使用await，而在asyncio.coroutine 中需要使用yield from。推荐使用第一种方法。以下两种方法是等价的：

import asyncio
async def hello():
    await asyncio.sleep(1)

@asyncio.coroutine
def hello():
    yield from asyncio.sleep(1)

调用协程

调用协程有两种方法，一种是使用await来调用，另一种是使用ensure_future来调用。

使用await调用的，马上中止当前线程，待await后面的线程执行完成后，再执行await后面的语句。使用ensure_future调用的，调用后立即返回，并执行后面的程序。

并发执行

如果需要多个协程并发执行，就需要使用wait函数。wait函数，可以接收多个协程，并返回一个协程。以下是多个协程同时执行的例子：

import asyncio
async def hello(name):
    print('%s starting'%(name))
    await asyncio.sleep(1)
    print('%s end.'%(name))

async def main():
    await wait(hello('a'),hello('b'))

try:
    loop=asyncio.get_event_loop()
    loop.run_until_complete(main())
finally:
    loop.close()

上面的示例可以并发执行两个协程hello('a')和hello('b')。

协程的启动

在程序中启动协程，可以采用上面的方法。但上面的方法有个缺点，只能使用一次。如果在程序中需要多次启动协程，则可以采用如下方法：

import asyncio
def start(*coros):
    # 执行协程，如coro为多个，则全部并发执行。
    try:
        loop=asyncio.new_event_loop()
        asyncio.set_event_loop(loop)
        if len(coros)==1:
            loop.run_until_complete(coros[0])
        else:
            loop.run_until_complete(wait(coros))
    finally:
        loop.close()

启动协程时，执行start(coro)即可。上面的函数在同一个程序中可以多次使用。