深度揭秘Go语言sync.WaitGroup：并发控制的得力工具

在Go语言中，`sync.WaitGroup`是并发控制的一个工具，用于等待一组goroutine的结束。 `sync.WaitGroup`类型在`sync`包中定义，并且它主要有两个方法：`Add`和`Done`。 * `Add`方法用于向`WaitGroup`添加一个或多个等待的goroutine。如果传递的参数是负数，那么会引发运行时错误。通常，我们使用`Add`方法在启动goroutine之前向`WaitGroup`添加一个计数器。 * `Done`方法用于在`WaitGroup`中标记一个goroutine的结束。这个方法将减少`WaitGroup`的计数器。当计数器达到零时，表示所有的goroutine都已完成，此时可以安全地退出等待。 使用`sync.WaitGroup`的一个典型示例如下： ```go package main import ( "fmt" "sync" "time" ) func main() { var wg sync.WaitGroup numGoroutines := 5 wg.Add(numGoroutines) for i := 0; i < numGoroutines; i++ { go func(id int) { defer wg.Done() fmt.Printf("Goroutine %d started\n", id) time.Sleep(time.Second) fmt.Printf("Goroutine %d finished\n", id) }(i) } wg.Wait() fmt.Println("All goroutines have finished.") } ``` 在这个示例中，我们创建了一个`sync.WaitGroup`实例，并在启动每个goroutine之前使用`Add`方法将其计数器增加1。每个goroutine在结束时使用`Done`方法将计数器减少1。我们使用`Wait`方法等待所有的goroutine完成。当计数器达到零时，我们输出一条消息表示所有的goroutine都已经完成。 ######[AI写代码神器 | 446点数解答 | 2023-11-09 18:13:53]