如何使用Golang WaitGroup等待协程完成_同步多个异步任务

sync.WaitGroup 是 Go 中等待多个 goroutine 完成的最常用方式，需主协程初始化、Add() 在 go 前调用、goroutine 内用指针调 Done()，并避免循环变量捕获等问题。

使用 sync.WaitGroup 是 Go 中等待多个 goroutine 完成的最常用、最可靠的方式。它的核心逻辑就三点：计数器初始化、任务启动前加一、任务结束时减一，主协程调用 Wait() 阻塞直到计数器归零。

正确初始化和使用 WaitGroup

WaitGroup 必须在启动 goroutine 前完成初始化，并且 Add() 调用必须在 go 语句之前（或至少在 goroutine 开始执行前），否则可能因竞态导致计数不准确。

• 用 var wg sync.WaitGroup 或 wg := new(sync.WaitGroup) 初始化
• 每次启动 goroutine 前调用 wg.Add(1)
• 在 goroutine 内部最后调用 wg.Done()（等价于 wg.Add(-1)）
• 主 goroutine 调用 wg.Wait() 阻塞等待全部完成

避免常见陷阱

最常见的错误是把 wg.Add(1) 放在 goroutine 内部，或传递指针/值不一致，导致 Wait 不返回或 panic。

• 不要在 goroutine 里调用 Add() —— 主协程负责“注册”任务
• 传入 goroutine 的 WaitGroup 必须是指针（&wg），否则副本操作无效
• 不要重复调用 Wait()，它不是可重入的；也不要在计数为 0 后再调用 Done()
• 如果需要多次复用，每次使用前应确保计数器已归零（通常新建一个更安全）

配合闭包和参数传递的写法

向 goroutine 传参时，注意循环变量捕获问题；WaitGroup 和参数都应显式传入，避免隐式共享。

for i := 0; i < 3; i++ {
    wg.Add(1)
    go func(id int, w *sync.WaitGroup) {
        defer w.Done()
        fmt.Printf("task %d done\n", id)
    }(i, &wg)
}
wg.Wait()

上面写法确保每个 goroutine 拿到独立的 i 值和正确的 WaitGroup 指针。

替代方案对比：WaitGroup vs channel vs errgroup

WaitGroup 最适合“只等完成、不关心结果”的场景；若需收集返回值或错误，channel 更自然；若要统一错误处理和上下文取消，errgroup.Group（来自 golang.org/x/sync/errgroup）是更好选择。

• 纯同步等待 → sync.WaitGroup
• 要取结果或做聚合 → chan + close + range
• 要支持 context 取消或返回 error → errgroup.Group