快捷導(dǎo)航

Python協(xié)程的四種實現(xiàn)方式總結(jié)

更新時間：2023年01月14日 14:35:43 作者：永遠(yuǎn)是少年啊

今天繼續(xù)給大家介紹Python關(guān)知識，本文主要內(nèi)容是Python協(xié)程的四種實現(xiàn)方式。文中的示例代碼講解詳細(xì)，感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下

一、yield關(guān)鍵字實現(xiàn)方式

以yield關(guān)鍵字方式實現(xiàn)協(xié)程代碼如下所示：

def fun1():
    yield 1
    yield from fun2()
    yield 2

def fun2():
    yield 3
    yield 4

f1=fun1()
for item in f1:
    print(item)

在上述代碼中，一個Python函數(shù)中存在由yiled，就說明為生成器函數(shù)，yield類似于普通函數(shù)的return，區(qū)別在于yield返回后Python會“記住”返回的位置，在下次返回時就從這個位置處返回。yiled from關(guān)鍵字后面跟著也是一個生成器，表示從該生成器返回。

在上述代碼運(yùn)行后，for循環(huán)就是一個迭代的過程，隨著迭代，每次fun1()生成器就會執(zhí)行一次“yield”。因此，上述函數(shù)會輸出1——3——4——2。該端代碼執(zhí)行結(jié)果如下所示：

上述代碼，其實并沒有完全的實現(xiàn)協(xié)程，或者說實現(xiàn)的協(xié)程非常牽強(qiáng)。

二、greenlet實現(xiàn)方式

greenlet是一個Python的第三方模塊，使用時需要提前安裝。執(zhí)行命令：

pip install greenlet

即可完成安裝。

greenlet實現(xiàn)協(xié)程代碼如下所示：

from greenlet import greenlet

def fun1():
    print(1)
    gr2.switch()
    print(2)
    gr2.switch()

def fun2():
    print(3)
    gr1.switch()
    print(4)

gr1=greenlet(fun1)
gr2=greenlet(fun2)
gr1.switch()

在上述代碼中，greenlet()函數(shù)生成了一個greenlet對象，在該對象中調(diào)用switch()函數(shù)即可切換到指定greenlet對象對應(yīng)的函數(shù)中執(zhí)行，因此可以實現(xiàn)協(xié)程操作。

上述代碼執(zhí)行結(jié)果如下所示：

三、asyncio實現(xiàn)方式

在Python3.4以后，asyncio是Python的一個內(nèi)置模塊，因此無需安裝即可調(diào)用。使用asyncio模式實現(xiàn)的Python代碼如下所示：

import asyncio

@asyncio.coroutine
def fun1():
    print(1)
    yield from asyncio.sleep(2)
    print(2)

@asyncio.coroutine
def fun2():
    print(3)
    yield from asyncio.sleep(2)
    print(4)

tasks=[
    asyncio.ensure_future(fun1()),
    asyncio.ensure_future(fun2())
]
loop=asyncio.get_event_loop()
loop.run_until_complete(asyncio.wait(tasks))

在上述代碼中，@asyncio.coroutin為一個裝飾器，使用該裝飾器后，一個函數(shù)由普通函數(shù)成為了協(xié)程函數(shù)。與之前兩種方式相比，這段代碼實現(xiàn)了協(xié)程遇到I/O阻塞時自動的切換。

四、async和await關(guān)鍵字實現(xiàn)方式

使用async和await關(guān)鍵字的實現(xiàn)方式，是Python3.5版本以后引入的協(xié)程操作方式，這種方式與第三種本質(zhì)上是相同的，但是由于這兩個關(guān)鍵字的引入，因此協(xié)程的實現(xiàn)比起第三種更加簡介。這種實現(xiàn)方式Python代碼如下所示：

import asyncio

async def fun1():
    print(1)
    await asyncio.sleep(2)
    print(2)

async def fun2():
    print(3)
    await asyncio.sleep(2)
    print(4)

tasks=[
    asyncio.ensure_future(fun1()),
    asyncio.ensure_future(fun2())
]
loop=asyncio.get_event_loop()
loop.run_until_complete(asyncio.wait(tasks))

這種協(xié)程的實現(xiàn)方式，是Python官方推薦的實現(xiàn)方式，如果Python版本允許，盡量采取這種方式運(yùn)行。上述代碼執(zhí)行結(jié)果如下所示：