Go程式設計技巧 Goroutine的優雅控制

原文：go程式設計技巧--goroutine的優雅控制

goroutine是go語言最重要的機制，goroutine將複雜的需要非同步的io呼叫抽象成同步呼叫，符合人類正常的順序思維，極大的簡化了io程式設計的難度。如同執行緒一樣，對goroutine既要掌握基本的用法，更要很好的控制goroutine的退出機制。本文介紹一種goroutine的退出思路。

通常goroutine會因為兩種情況阻塞：

io操作，比如對socket的read。

channel操作。對乙個chan的讀寫都有可能阻塞goroutine。

對於情況1，只需要關閉對應的描述符，阻塞的goroutine自然會被喚醒。

重點討論情況2。併發程式設計，goroutine提供一種channel機制，channel類似管道，寫入者向裡面寫入資料，讀取者從中讀取資料。如果channel裡面沒有資料，讀取者將阻塞，直到有資料；如果channel裡面資料滿了，寫入者將因為無法繼續寫入資料而阻塞。

如果在整個應用程式的生命週期裡，writer和reader都表現為乙個goroutine，始終都在工作，那麼如何在應用程式結束前，通知它們終止呢？在go中，並不推薦像abort執行緒那樣，強行的終止goroutine。因此，抽象的說，必然需要保留乙個入口，能夠跟writer或reader通訊，以告知它們終止。

我們先看reader。我們首先可以想到，利用close函式關閉正在讀取的channel，從而可以喚醒reader，並退出。但是考慮到close並不能很好的處理writer（因為writer試圖寫入乙個已經close的channel，將引發異常）。因此，我們需要設計乙個額外的唯讀channel用於通知：

type routinesignal struct 
}

routinesignal的例項，應當通過外部生成並傳遞給reader，例如：

func (r *reader)init(s *routinesignal)

在reader的迴圈中，就可以這麼寫：

func (r *reader)loop() 
}}

當需要終止goroutine的時候只需要關閉這個額外的channel：

close(signal.done)

看起來很完備了，這可以處理大部分的情況了。這樣做有個弊端，儘管，我們可以期望close喚醒goroutine進而退出，但是並不能知道goroutine什麼時候完成退出，因為goroutine可能在退出前還有一些善後工作，這個時候我們需要sync.waitgroup。改造一下routinesignal：

type routinesignal struct 
wg sync.waitgroup
}

增加乙個sync.waitgroup的例項，在goroutine開始工作時，對wg加1，在goroutine退出前，對wg減1：

func (r *reader)loop() 
}}

外部，只需要等待waitgroup返回即可：

close(signal.done)
signal.wg.wait()

只要wait()返回就能斷定goroutine結束了。

推導一下，不難發現，對於writer也可以採用這種方法。於是，總結一下，我們建立了乙個叫routinesignal的結構，結構裡面包含乙個chan用來通知goroutine結束，包含乙個waitgroup用於goroutine通知外部完成善後。這樣，通過這個結構的例項優雅的終止goroutine，而且還可以確保goroutine終止成功。

Go程式設計技巧 Goroutine的優雅控制

Go語言併發程式設計goroutine

Go語言學習 goroutine

Go語言中Goroutine的設定

Go程式設計技巧 Goroutine的優雅控制

Go語言 併發程式設計goroutine

Go語言學習 goroutine

Go語言中Goroutine的設定

相關推薦

Go語言併發程式設計goroutine