在Golang中,可以使用WaitGroup来管理并发的goroutine,以确保在所有goroutine完成之前,主程序不会退出。
WaitGroup是一个计数信号量,它的值初始为0。每个goroutine在执行之前,通过调用WaitGroup的Add方法将计数加1。在goroutine的末尾,通过调用WaitGroup的Done方法将计数减1。主程序调用WaitGroup的Wait方法,将会被阻塞,直到所有goroutine的计数变为0。
下面是一个使用WaitGroup的示例:
package mainimport ("fmt""sync")func main() {var wg sync.WaitGroupfor 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,确保它们的执行顺序和主程序的退出顺序一致。

