簡介

不要把增加其計數器值的操作和調用其Wait方法的代碼，放在不同的 goroutine 中執行。換句話說，要杜絕對同一個WaitGroup值的兩種操作的并發執行

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

sync.WaitGroup只有3個方法，Add()，Done()，Wait()。

其中Done()是Add(-1)的別名。簡單的來說，使用Add()添加計數，Done()減掉一個計數，計數不為0, 阻塞Wait()的運行。

也可以通過這種方式創建

wg := new(sync.WaitGroup)

• Add 用來添加 goroutine 的個數

• Done 執行一次數量減 1

• Wait 用來等待結束

例子

同時開三個協程去請求網頁， 等三個請求都完成后才繼續 Wait 之后的工作

var wg sync.WaitGroup var urls = []string{"http://www.golang.org/","http://www.google.com/","http://www.somestupidname.com/", } for _, url := range urls {// Increment the WaitGroup counter. wg.Add(1)// Launch a goroutine to fetch the URL. go func(url string) {// Decrement the counter when the goroutine completes. defer wg.Done()// Fetch the URL. http.Get(url)}(url) } // Wait for all HTTP fetches to complete. wg.Wait()

或者下面的代碼
用于測試 給chan發送 1千萬次，并接受1千萬次的性能

import ("fmt""sync""time" )const (num = 10000000 )func main() {TestFunc("testchan", TestChan) }func TestFunc(name string, f func()) {st := time.Now().UnixNano()f()fmt.Printf("task %s cost %d \r\n", name, (time.Now().UnixNano()-st)/int64(time.Millisecond)) }func TestChan() {var wg sync.WaitGroupc := make(chan string)wg.Add(1)go func() {for _ = range c {}wg.Done()}()for i := 0; i < num; i++ {c <- "123"}close(c)wg.Wait() }

總結

以上是生活随笔為你收集整理的golang中的WaitGroup的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。