go提供了sync包和channel來解決協程同步和通信。新手對channel通道操做起來更容易產生死鎖,若是時緩衝的channel還要考慮channel放入和取出數據的速率問題。
從字面就能夠理解,sync.WaitGroup是等待一組協程結束。它實現了一個相似任務隊列的結構,你能夠向隊列中加入任務,任務完成後就把任務從隊列中移除,若是隊列中的任務沒有所有完成,隊列就會觸發阻塞以阻止程序繼續運行。
sync.WaitGroup只有3個方法,Add(),Done(),Wait()。 其中Done()是Add(-1)的別名。簡單的來講,使用Add()添加計數,Done()減掉一個計數,計數不爲0, 阻塞Wait()的運行。
簡單示例以下:code
package main import ( "fmt" "sync" ) var waitgroup sync.WaitGroup func test(shownum int) { fmt.Println(shownum) waitgroup.Done() //任務完成,將任務隊列中的任務數量-1,其實.Done就是.Add(-1) } func main() { for i := 0; i < 10; i++ { waitgroup.Add(1) //每建立一個goroutine,就把任務隊列中任務的數量+1 go test(i) } waitgroup.Wait() //.Wait()這裏會發生阻塞,直到隊列中全部的任務結束就會解除阻塞 fmt.Println("done!") }