Go 语言中 sync.WaitGroup 的正确计数姿势

Golang不知道大家是否熟悉？今天我将给大家介绍《Go 语言中 sync.WaitGroup 的正确计数姿势》，这篇文章主要会讲到等等知识点，如果你在看完本篇文章后，有更好的建议或者发现哪里有问题，希望大家都能积极评论指出，谢谢！希望我们能一起加油进步！

wg.Add()必须在go语句前调用，否则Wait可能提前返回或永久阻塞；Done()须由子goroutine配对调用且推荐defer；WaitGroup零值可用但禁止复制；它仅计数归零，不处理错误、顺序与超时。

wg.Add() 必须在 go 语句之前调用

这是最常踩的坑：把 wg.Add(1) 放在 go 启动协程内部，或和 go 并列但顺序不保，会导致 wg.Wait() 永远阻塞或 panic。

原因很简单：wg.Add() 是主 goroutine 主动“登记任务数”，而 wg.Wait() 只认当前计数器值。如果 Add 还没执行完，Wait 就已开始检查（甚至可能在 go 启动前就执行了），那它看到的就是 0，直接返回；或者更糟——部分 Add 晚到，造成计数器始终 >0。

• ✅ 正确写法：循环中先 wg.Add(1)，再 go func() {...}()
• ❌ 错误写法：go func() { wg.Add(1); ...; wg.Done() }() —— 这等于动态追加，且无法保证 Add 在 Wait 前完成
• ⚠️ 危险写法：go func() { defer wg.Done(); ... }(); wg.Add(1) —— 主 goroutine 可能已执行到 Wait()，此时计数仍是 0

计数器不能为负，Done() 要配对且只由子 goroutine 调用

sync: negative WaitGroup counter 这个 panic 几乎都源于 Done() 被多调、漏调 Add()，或在主 goroutine 中手动调了 Done()。

Done() 的语义是“我这个任务完成了”，它必须由对应子 goroutine 自己声明，不能由外部代劳，也不能靠猜路径手动调——一旦有 error return、panic 或提前退出，就容易漏。

• ✅ 推荐写法：defer wg.Done() —— panic 时也触发，覆盖所有退出路径
• ❌ 禁止写法：wg.Add(-1) 或 defer wg.Add(-1) —— Add 不是线程安全的减法，且违反契约
• ⚠️ 高危模式：多个 defer wg.Done() 在同一函数里（比如嵌套 defer 或重复注册）

零值可用，但绝不能复制 WaitGroup 实例

var wg sync.WaitGroup 的零值本身完全合法，可直接用。但一旦你把它赋值给另一个变量、传进函数时不传指针、或作为 struct 字段被拷贝，就出问题了。

因为 sync.WaitGroup 内部含 noCopy 字段，编译期 vet 会报错；更隐蔽的是，运行时拷贝会导致两个实例维护各自独立的计数器，Done() 在副本上调用，对原实例毫无影响 —— 表现就是 Wait() 死锁，且无任何错误提示。

• ✅ 安全用法：始终传递指针，如 func worker(url string, wg *sync.WaitGroup)
• ❌ 危险用法：w2 := wg、doSomething(wg)（参数是值类型）、struct{ wg sync.WaitGroup } 直接嵌入
• ⚠️ 注意点：Go 1.21+ 对 noCopy 检查更严格，某些隐式拷贝（如切片 append 含 WaitGroup 的 struct）也可能触发 vet 报警

WaitGroup 不处理错误和顺序，只管“数字归零”

很多人误以为 wg.Wait() 返回就代表“所有任务成功完成”。其实它只看计数器是否为 0 —— 即使某个 goroutine panic 了、某个 HTTP 请求 404 了、某个任务 1ms 就 return 了，只要它调了 Done()，WaitGroup 就认为“完成”。

这意味着：错误收集、结果聚合、超时控制、执行顺序保障，全部得你自己来。

• ✅ 实践建议：用外部 slice 或 channel 收集返回值/错误，如 errs := make([]error, len(urls))
• ❌ 不能依赖：用 wg.Wait() 判断某次请求是否成功，或假设 goroutine 按启动顺序完成
• ⚠️ 易忽略点：没有内置超时机制 —— 若某个 goroutine 卡住没调 Done()，Wait() 会永远阻塞，需配合 context 或定时器自行实现

理论要掌握，实操不能落！以上关于《Go 语言中 sync.WaitGroup 的正确计数姿势》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！