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.WaitGroup    for 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 也能實現，但是覺得如果涉及不到子線程與主線程資料同步，這個感覺不錯。