在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,确保它们的执行顺序和主程序的退出顺序一致。