WaitGroup用于等待多个goroutine完成，通过Add增加计数、Done减少计数、Wait阻塞直到计数为零，确保主协程在所有子任务结束后再继续执行。
在 Go 语言中，WaitGroup 是 sync 包提供的一个同步原语，用于等待一组并发的 goroutine 执行完成。它特别适用于主协程需要等待多个子协程结束后再继续执行的场景，比如批量处理任务、并发请求聚合等。
WaitGroup 的核心作用
当启动多个 goroutine 做并行任务时，主协程通常不能立即退出，否则会终止所有子协程。WaitGroup 提供了一种机制，让主协程“等待”所有子任务完成后再继续或退出。
其本质是一个计数器：
- 每启动一个 goroutine，调用 Add(1) 增加计数
- 每个 goroutine 执行完后，调用 Done() 减少计数
- 主协程调用 Wait() 阻塞，直到计数归零
基本使用方法
下面是一个典型的 WaitGroup 使用示例：
立即学习“go语言免费学习笔记（深入）”；
package main
import (
"fmt"
"sync"
"time"
)
func worker(id int, wg *sync.WaitGroup) {
defer wg.Done() // 任务完成，计数减一
fmt.Printf("Worker %d starting\n", id)
time.Sleep(time.Second)
fmt.Printf("Worker %d done\n", id)
}
func main() {
var wg sync.WaitGroup
for i := 1; i <= 3; i++ {
wg.Add(1) // 每次启动前增加计数
go worker(i, &wg)
}
wg.Wait() // 阻塞直到所有 Done 被调用
fmt.Println("All workers finished")
}
输出结果：
Worker 1 starting Worker 2 starting Worker 3 starting Worker 1 done Worker 2 done Worker 3 done All workers finished
使用注意事项
WaitGroup 虽然简单，但使用不当容易出错。以下几点需特别注意：
- Add 的值不能为负数：否则会 panic
- 确保 Done 调用次数与 Add 一致：漏调或多调都会导致死锁或 panic
- WaitGroup 不是可复制类型：不要将其作为参数值传递，应传指针
- Add 应在 Wait 之前调用：如果在 Wait 后 Add，可能引发竞态条件
常见误用与修复
错误写法：在 goroutine 内部才 Add
for i := range tasks {
go func() {
wg.Add(1) // 错误！Add 在 goroutine 内
defer wg.Done()
// ...
}()
}
wg.Wait()
这可能导致主协程已进入 Wait，而 Add 还未执行，从而漏记协程。
正确做法：在 goroutine 启动前调用 Add
for i := range tasks {
wg.Add(1)
go func() {
defer wg.Done()
// ...
}()
}
wg.Wait()
基本上就这些。WaitGroup 是 Go 并发编程中最基础也最实用的同步工具之一，掌握它的使用和边界条件，能有效避免程序提前退出或死锁问题。关键是控制好计数的增减时机，保证一致性。不复杂但容易忽略细节。
