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)即可。上面的函数在同一个程序中可以多次使用。

    原文作者:远飞的大雁2010
    原文地址: https://www.jianshu.com/p/c245214ed213
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞