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!") }