Python3.10?Generator生成器Coroutine原生協(xié)程詳解

更新時間：2022年12月28日 09:04:47 作者：劉悅的技術博客

這篇文章主要為大家介紹了Python3.10?Generator生成器Coroutine原生協(xié)程詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪

引言

普遍意義上講，生成器是一種特殊的迭代器，它可以在執(zhí)行過程中暫停并在恢復執(zhí)行時保留它的狀態(tài)。而協(xié)程，則可以讓一個函數(shù)在執(zhí)行過程中暫停并在恢復執(zhí)行時保留它的狀態(tài)，在Python3.10中，原生協(xié)程的實現(xiàn)手段，就是生成器，或者說的更具體一些：協(xié)程就是一種特殊的生成器，而生成器，就是協(xié)程的入門心法。

協(xié)程底層實現(xiàn)

我們知道，Python3.10中可以使用async和await關鍵字來實現(xiàn)原生協(xié)程函數(shù)的定義和調(diào)度，但其實，我們也可以利用生成器達到協(xié)程的效果，生成器函數(shù)和普通函數(shù)的區(qū)別在于，生成器函數(shù)使用 yield 語句來暫停執(zhí)行并返回結果。例如，下面是一個使用生成器函數(shù)實現(xiàn)的簡單協(xié)程：

def my_coroutine():  
    while True:  
        x = yield  
        print(x)  
# 使用生成器函數(shù)創(chuàng)建協(xié)程  
coroutine = my_coroutine()  
# 啟動協(xié)程  
next(coroutine)  
# 在協(xié)程中傳入數(shù)據(jù)  
coroutine.send(1)  
coroutine.send(2)  
coroutine.send(3)

程序返回：

? mydemo git:(master) ? /opt/homebrew/bin/python3.10 "/Users/liuyue/wodfan/work/mydemo/src/test.py"
1
2
3

在上面的代碼中，生成器函數(shù) my_coroutine 使用了一個無限循環(huán)來實現(xiàn)協(xié)程的邏輯。每當調(diào)用 send 方法時，協(xié)程就會從 yield 語句處恢復執(zhí)行，并將傳入的參數(shù)賦值給變量 x。

如此，就完成了協(xié)程執(zhí)行-》阻塞-》切換-》回調(diào)的工作流模式。

當然，作為事件循環(huán)機制，協(xié)程服務啟動可能無限期地運行，要關閉協(xié)程服務，可以使用生成器的close()方法。當一個協(xié)程被關閉時，它會生成GeneratorExit異常，該異常可以用生成器的方式進行捕獲：

def my_coroutine():  
    try :  
        while True:  
            x = yield  
            print(x)  
    except GeneratorExit:  
            print("協(xié)程關閉")  
# 使用生成器函數(shù)創(chuàng)建協(xié)程  
coroutine = my_coroutine()  
# 啟動協(xié)程  
next(coroutine)  
# 在協(xié)程中傳入數(shù)據(jù)  
coroutine.send(1)  
coroutine.send(2)  
coroutine.send(3)  
coroutine.close()

程序返回：

? mydemo git:(master) ? /opt/homebrew/bin/python3.10 "/Users/liuyue/wodfan/work/mydemo/src/test.py"
1
2
3
協(xié)程關閉

業(yè)務場景

在實際業(yè)務場景中，我們也可以使用生成器來模擬協(xié)程流程，主要體現(xiàn)在數(shù)據(jù)的IO流操作中，假設我們需要從本地往服務器傳輸數(shù)據(jù)，首先建立鏈接對象：

class Connection:  
    def __init__(self, addr):  
        self.addr = addr  
    def transmit(self, data):  
        print(f"X: {data[0]}, Y: {data[1]} sent to {self.addr}")

隨后建立生成器函數(shù)：

def send_to_server(conn):  
    while True:  
        try:  
            raw_data = yield  
            raw_data = raw_data.split(' ')  
            coords = (float(raw_data[0]), float(raw_data[1]))  
            conn.transmit(coords)  
        except ConnectionError:  
            print("鏈接丟失，進行回調(diào)")  
            conn = Connection("重新連接v3u.cn")

利用生成器調(diào)用鏈接類的transmit方法進行數(shù)據(jù)的模擬傳輸，如果鏈接斷開，則會觸發(fā)回調(diào)重新連接，執(zhí)行邏輯：

if __name__ == '__main__':  
    conn = Connection("v3u.cn")  
    sender = send_to_server(conn)  
    sender.send(None)  
    for i in range(1, 6):  
        sender.send(f"{100/i} {200/i}")  
    # 模擬鏈接斷開  
    conn.addr = None  
    sender.throw(ConnectionError)   
    for i in range(1, 6):  
        sender.send(f"{100/i} {200/i}")

程序返回：

X: 100.0, Y: 200.0 sent to v3u.cn  
X: 50.0, Y: 100.0 sent to v3u.cn  
X: 33.333333333333336, Y: 66.66666666666667 sent to v3u.cn  
X: 25.0, Y: 50.0 sent to v3u.cn  
X: 20.0, Y: 40.0 sent to v3u.cn  
鏈接丟失，進行回調(diào)  
X: 100.0, Y: 200.0 sent to 重新連接v3u.cn  
X: 50.0, Y: 100.0 sent to 重新連接v3u.cn  
X: 33.333333333333336, Y: 66.66666666666667 sent to 重新連接v3u.cn  
X: 25.0, Y: 50.0 sent to 重新連接v3u.cn  
X: 20.0, Y: 40.0 sent to 重新連接v3u.cn

如此，我們就可以利用生成器的“狀態(tài)保留”機制來控制網(wǎng)絡鏈接突然斷開的回調(diào)補救措施了。

所以說，協(xié)程就是一種特殊的生成器：

async def test():  
    pass  
print(type(test()))

您猜怎么著？

<class 'coroutine'>

結語

誠然，生成器和協(xié)程也并非完全是一個概念，與生成器不同的是，協(xié)程可以被另一個函數(shù)（稱為調(diào)用方）恢復執(zhí)行，而不是只能由生成器本身恢復執(zhí)行。這使得協(xié)程可以用來實現(xiàn)更復雜的控制流，因為它們可以在執(zhí)行時暫停并在任意時刻恢復執(zhí)行。

以上就是Python3.10 Generator生成器Coroutine原生協(xié)程詳解的詳細內(nèi)容，更多關于Python生成器原生協(xié)程的資料請關注腳本之家其它相關文章！

您可能感興趣的文章: