golang sync WaitGroup

這是一個建立於 的文章，其中的資訊可能已經有所發展或是發生改變。

剛才看golang的sync的包，看見一個很有用的功能。就是WaitGroup。

先說說WaitGroup的用途：它能夠一直等到所有的goroutine執行完成，並且阻塞主線程的執行，直到所有的goroutine執行完成。

WaitGroup總共有三個方法：Add(delta int),Done(),Wait()。簡單的說一下這三個方法的作用。

Add:添加或者減少等待goroutine的數量

Done:相當於Add(-1)

Wait:執行阻塞，直到所有的WaitGroup數量變成0

例子：

package mainimport ("fmt""sync""time")func main() {var wg sync.WaitGroupfor i := 0; i < 5; i = i + 1 {wg.Add(1)go func(n int) {// defer wg.Done()defer wg.Add(-1)EchoNumber(n)}(i)}wg.Wait()}func EchoNumber(i int) {time.Sleep(3e9)fmt.Println(i)}

輸出結果：

01234

程式很簡單，只是將每次迴圈的數量過3秒鐘輸出。那麼，這個程式如果不用WaitGroup，那麼將看不見輸出結果。因為goroutine還沒執行完，主線程已經執行完畢。注釋的defer wg.Done()和defer wg.Add(-1)作用一樣。這個很好，原來執行指令碼，都是使用time.Sleep，用一個估計的時間等到子線程執行完。WaitGroup很好。雖然chanel也能實現，但是覺得如果涉及不到子線程與主線程資料同步，這個感覺不錯。

文章來自：http://www.liguosong.com/2014/05/06/golang-sync-waitgroup/