1. 程式人生 > >golang sync.WaitGroup解決goroutine同步

golang sync.WaitGroup解決goroutine同步

sync包詳細說明參考:https://blog.csdn.net/weixin_42117918/article/details/85053135

go提供了sync包和channel來解決協程同步和通訊。新手對channel通道操作起來更容易產生死鎖,如果時緩衝的channel還要考慮channel放入和取出資料的速率問題。
從字面就可以理解,sync.WaitGroup是等待一組協程結束。它實現了一個類似任務佇列的結構,你可以向佇列中加入任務,任務完成後就把任務從佇列中移除,如果佇列中的任務沒有全部完成,佇列就會觸發阻塞以阻止程式繼續執行。
sync.WaitGroup只有3個方法,Add(),Done(),Wait()

。 其中Done()是Add(-1)的別名。簡單的來說,使用Add()新增計數,Done()減掉一個計數,計數不為0, 阻塞Wait()的執行。 
簡單示例如下:

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

文章出處:https://studygolang.com/articles/9173