sync.WaitGroup 是 Go 中用于等待一组 goroutine 完成的轻量同步工具,核心为 Add、Done、Wait 三步,需注意计数匹配、避免复制和误调用。
Go 语言中,sync.WaitGroup 是最常用、最轻量的协程(goroutine)同步工具之一,用于等待一组协程执行完毕。它不关心协程内部逻辑,只负责“计数”和“阻塞等待”,适合主协程等待多个子协程完成后再继续执行的场景。
WaitGroup 的使用遵循固定模式:声明 → 增加计数 → 启动协程 → 等待完成。漏掉任意一步都可能导致 panic 或死锁。
var wg sync.WaitGroup
wg.Add(n):在启动协程前,把要等待的协程数量加进去(n > 0)。注意不能在协程里调用 Add,否则可能因竞态导致计数错误wg.Done():每个协程结束前必须调用一次,等价于 wg.Add(-1)。推荐用 defer wg.Done() 确保执行wg.Wait():会阻塞直到内部计数归零。该调用应放在所有 go 语句之后,且只能被调用一次WaitGroup 表面简单,但几个典型错误会让程序崩溃或卡死:
func worker(wg *sync.WaitGroup, job int)
模拟向多个 URL 发起 HTTP 请求,并等待全部完成:
func fetchAll(urls []string) { var wg sync.WaitGroup results := make(chan string, len(urls))
for _, url := range urls { wg.Add(1) go func(u string) { defer wg.Done() resp, err := http.Get(u) if err != nil { results <- "error: " + err.Error() return } results <- fmt.Sprintf("success: %s", resp.Status) }(url) // 注意:需传参避免闭包引用同一变量 } go func() { wg.Wait() close(results) }() for res := range results { fmt.Println(res) }}
这个例子体现了 WaitGroup 和 channel 的配合:WaitGroup 确保所有请求发起并结束,channel 收集结果,避免额外锁或共享变量。
WaitGroup vs 其他同步机制
不是所有并发等待都要用 WaitGroup:
sync.Mutex + sync.Cond 或 sync.Once
context.Context 配合 channel select,WaitGroup 本身不支持超时sync.Once 更合适;若需多次通知,考虑 chan struct{} 或 sync.Cond
WaitGroup 的定位很明确:**仅做“数量同步”,不做“通信”或“协调”。用对了,简洁高效;用错了,调试困难。
基本上就这些。WaitGroup 不复杂但容易忽略细节,掌握 Add/Wait/Done 的时机和生命周期,就能稳稳控住并发节奏。