WaitGroup是Go語言中的一個同步原語,用于等待一組goroutine的完成。
下面是一個簡單的應用實例,演示了如何使用WaitGroup等待一組goroutine的完成:
package main
import (
"fmt"
"sync"
"time"
)
// 模擬耗時操作,每個goroutine等待一秒鐘
func doSomething(i int, wg *sync.WaitGroup) {
defer wg.Done()
fmt.Printf("Goroutine %d start\n", i)
time.Sleep(time.Second)
fmt.Printf("Goroutine %d done\n", i)
}
func main() {
var wg sync.WaitGroup
// 啟動5個goroutine
for i := 0; i < 5; i++ {
wg.Add(1)
go doSomething(i, &wg)
}
// 等待所有goroutine完成
wg.Wait()
fmt.Println("All goroutines done")
}
在上面的例子中,我們定義了一個doSomething函數來模擬耗時的操作。在main函數中,我們創建了一個WaitGroup,并在啟動每個goroutine之前調用Add方法來增加計數器。
然后,每個goroutine執行doSomething函數,并在完成后調用Done方法來遞減計數器。
最后,我們調用Wait方法來等待所有goroutine完成。一旦計數器歸零,Wait方法將返回,程序繼續執行。
運行上面的代碼,你將看到類似以下的輸出:
Goroutine 0 start
Goroutine 1 start
Goroutine 2 start
Goroutine 3 start
Goroutine 4 start
Goroutine 0 done
Goroutine 2 done
Goroutine 3 done
Goroutine 4 done
Goroutine 1 done
All goroutines done
從輸出中可以看出,所有的goroutine都已經完成,并且在最后打印了"All goroutines done"。
通過使用WaitGroup,我們可以方便地等待一組goroutine的完成,從而控制程序的執行順序和并發度。這在并發編程中是非常常見和實用的技術。