經常看到有人會問如何等待主協程中創建的協程執行完畢之后再結束主協程，例如如下代碼：

package main import ("fmt" ) func main() {go func() {fmt.Println("Goroutine 1")}()go func() {fmt.Println("Goroutine 2")}() }

執行以上代碼很可能看不到輸出，因為有可能這兩個協程還沒得到執行主協程已經結束了，而主協程結束時會結束所有其他協程。解決辦法是可以在main函數結尾加上等待：

package main import ("fmt""time" ) func main() {go func() {fmt.Println("Goroutine 1")}()go func() {fmt.Println("Goroutine 2")}()time.Sleep(time.Second * 1) // 睡眠1秒，等待上面兩個協程結束 }

這并不是完美的解決方法，如果這兩個協程中包含復雜的操作，可能很耗時間，就無法確定需要睡眠多久，當然可以用管道實現同步：

package main import ("fmt" ) func main() {ch := make(chan struct{})count := 2 // count 表示活動的協程個數go func() {fmt.Println("Goroutine 1")ch <- struct{}{} // 協程結束，發出信號}()go func() {fmt.Println("Goroutine 2")ch <- struct{}{} // 協程結束，發出信號}()for range ch {// 每次從ch中接收數據，表明一個活動的協程結束count--// 當所有活動的協程都結束時，關閉管道if count == 0 {close(ch)}} }

上面的解決方案是比較完美的方案，但是Go提供了更簡單的方法——使用sync.WaitGroup。WaitGroup顧名思義，就是用來等待一組操作完成的。WaitGroup內部實現了一個計數器，用來記錄未完成的操作個數，它提供了三個方法，Add()用來添加計數。Done()用來在操作結束時調用，使計數減一。Wait()用來等待所有的操作結束，即計數變為0，該函數會在計數不為0時等待，在計數為0時立即返回。

package main import ("fmt""sync" ) func main() {var wg sync.WaitGroupwg.Add(2) // 因為有兩個動作，所以增加2個計數go func() {fmt.Println("Goroutine 1")wg.Done() // 操作完成，減少一個計數}()go func() {fmt.Println("Goroutine 2")wg.Done() // 操作完成，減少一個計數}()wg.Wait() // 等待，直到計數為0 }

可見用sync.WaitGroup是最簡單的方式。

總結

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

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