在Golang中,可以使用WaitGroup來管理并發的goroutine,以確保在所有goroutine完成之前,主程序不會退出。
WaitGroup是一個計數信號量,它的值初始為0。每個goroutine在執行之前,通過調用WaitGroup的Add方法將計數加1。在goroutine的末尾,通過調用WaitGroup的Done方法將計數減1。主程序調用WaitGroup的Wait方法,將會被阻塞,直到所有goroutine的計數變為0。
下面是一個使用WaitGroup的示例:
package main
import (
"fmt"
"sync"
)
func main() {
var wg sync.WaitGroup
for i := 0; i < 5; i++ {
wg.Add(1)
go func(id int) {
defer wg.Done()
fmt.Printf("goroutine %d\n", id)
}(i)
}
wg.Wait()
fmt.Println("All goroutines have finished")
}
在上面的例子中,主程序啟動了5個goroutine,并通過調用wg.Add(1)將計數加1。每個goroutine在執行完任務后,通過調用wg.Done()將計數減1。最后,主程序調用wg.Wait(),將會被阻塞,直到所有goroutine的計數變為0。當所有goroutine完成后,主程序會打印"All goroutines have finished"。
使用WaitGroup可以很方便地管理并發的goroutine,確保它們的執行順序和主程序的退出順序一致。