WaitGroup用于等待多个goroutine完成，通过Add增加计数、Done减少计数、Wait阻塞直至计数为零。
在Go语言中，WaitGroup 是 sync 包提供的一个同步原语，用于等待一组并发的 goroutine 执行完成。它特别适合在主协程需要等待多个子协程结束时使用，而不需要关闭通道或处理复杂的信号机制。
基本用法和核心方法
WaitGroup 有三个主要方法：
-
Add(delta int)：增加计数器，通常传入要启动的 goroutine 数量。 -
Done()：在每个 goroutine 结束时调用，相当于计数器减1。 -
Wait()：阻塞当前协程，直到计数器归零。
典型结构是：主协程调用 Add(n) 设置任务数量，每个子协程执行完后调用 Done()，主协程通过 Wait() 阻塞等待全部完成。
简单示例：等待多个goroutine完成
package main
import (
"fmt"
"sync"
"time"
)
func worker(id int, wg *sync.WaitGroup) {
defer wg.Done() // 任务完成，计数器减1
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) // 启动goroutine
}
wg.Wait() // 等待所有worker完成
fmt.Println("All workers finished")
}
输出会显示三个 worker 依次开始和结束，最后打印“All workers finished”。
立即学习“go语言免费学习笔记（深入）”；
注意事项和常见错误
使用 WaitGroup 时需注意以下几点：
-
避免Add负值：调用
Add(-n)会导致 panic，除非你明确知道计数器足够大。 -
确保Done被调用：每个
Add(1)必须对应一次Done()，否则程序会一直阻塞。 -
传递指针：应将
WaitGroup的指针传给 goroutine，避免值拷贝导致状态不一致。 -
Add应在Wait前调用：最好在
go语句前调用Add，防止竞争条件。
适用场景
WaitGroup 适用于以下情况：
- 批量启动多个 goroutine 并等待它们全部完成（如并行请求、数据抓取）。
- 无需从 goroutine 获取返回值，只关心执行完成。
- 作为轻量级同步工具，比通道更简洁。
基本上就这些。只要记住 Add、Done、Wait 的配合使用，就能安全地控制并发流程。
