在Go语言并发编程中,sync.WaitGroup是控制多个goroutine协同执行的常用工具,很多开发者在编写复用逻辑时会考虑它是否支持安全重用,这需要从它的设计逻辑和使用规则入手分析。

sync.WaitGroup基础使用回顾
sync.WaitGroup的核心作用是等待一组goroutine全部执行完成,它提供了三个核心方法:
- Add(delta int):设置需要等待的goroutine数量,delta可以为正数也可以为负数
- Done():等价于Add(-1),通常在goroutine退出时调用,表示当前goroutine执行完成
- Wait():阻塞当前调用方,直到所有需要等待的goroutine都执行完成
下面是一个基础的使用示例,演示等待两个goroutine执行完成:
package main
import (
"fmt"
"sync"
"time"
)
func main() {
var wg sync.WaitGroup
// 设置需要等待2个goroutine
wg.Add(2)
go func() {
defer wg.Done()
time.Sleep(time.Second)
fmt.Println("goroutine 1 执行完成")
}()
go func() {
defer wg.Done()
time.Sleep(time.Second)
fmt.Println("goroutine 2 执行完成")
}()
// 阻塞等待所有goroutine完成
wg.Wait()
fmt.Println("所有goroutine执行完成,主程序退出")
}
sync.WaitGroup的重用前提
sync.WaitGroup本身支持重用,但必须满足一个核心前提:上一次的Wait()调用必须已经返回,才能再次使用这个WaitGroup实例。也就是说,必须确保前一轮的所有goroutine都已经执行完毕,Wait()不再阻塞,才能开启下一轮的Add、Done、Wait流程。
如果违反这个前提,在Wait()还未返回时就再次调用Add方法,会触发panic,错误信息为sync: WaitGroup is reused before previous Wait has returned。这是因为WaitGroup内部维护了一个计数器,在Wait()阻塞期间,计数器处于活跃等待状态,此时修改计数器会破坏内部状态。
正确的重用示例
下面的代码演示了安全的重用方式,每一轮都先等待上一轮的所有goroutine完成,再启动下一轮:
package main
import (
"fmt"
"sync"
"time"
)
func main() {
var wg sync.WaitGroup
// 第一轮并发任务
taskCount1 := 3
wg.Add(taskCount1)
for i := 0; i < taskCount1; i++ {
go func(index int) {
defer wg.Done()
time.Sleep(time.Millisecond * 500)
fmt.Printf("第一轮任务 %d 执行完成n", index)
}(i)
}
wg.Wait()
fmt.Println("第一轮所有任务完成")
// 第二轮并发任务,此时上一轮Wait已返回,可以安全重用
taskCount2 := 2
wg.Add(taskCount2)
for i := 0; i < taskCount2; i++ {
go func(index int) {
defer wg.Done()
time.Sleep(time.Millisecond * 300)
fmt.Printf("第二轮任务 %d 执行完成n", index)
}(i)
}
wg.Wait()
fmt.Println("第二轮所有任务完成,程序退出")
}
错误的重用场景与风险
最常见的错误是在Wait()还在阻塞时就启动下一轮的Add调用,比如下面的错误示例:
package main
import (
"fmt"
"sync"
"time"
)
func main() {
var wg sync.WaitGroup
wg.Add(2)
go func() {
defer wg.Done()
time.Sleep(time.Second)
fmt.Println("第一个goroutine完成")
}()
go func() {
defer wg.Done()
time.Sleep(time.Second)
fmt.Println("第二个goroutine完成")
}()
// 错误:在Wait阻塞前就再次调用Add,会触发panic
go func() {
time.Sleep(time.Millisecond * 200)
wg.Add(1) // 此时上一轮的Wait还未返回,重用非法
go func() {
defer wg.Done()
fmt.Println("错误的goroutine")
}()
}()
wg.Wait()
fmt.Println("程序退出")
}
运行上述代码会直接触发panic,导致程序崩溃。此外,还有一种隐性的错误场景:在Done()调用次数超过Add()设置的初始值时,计数器会变成负数,同样会触发panic,错误信息为sync: negative WaitGroup counter,这种问题在重用时如果计数逻辑混乱也很容易出现。
重用的注意事项
为了在重用sync.WaitGroup时避免问题,需要注意以下几点:
- 每一轮的Add()调用必须放在启动goroutine之前,避免goroutine先执行Done()导致计数器提前归零
- 确保Add()传入的delta和Done()的调用次数完全匹配,不要多也不要少
- 如果需要多轮并发任务,建议每一轮都重新声明一个新的WaitGroup实例,虽然会有少量内存开销,但可以避免逻辑混乱导致的问题,代码可读性也更高
- 不要在多个goroutine中同时调用同一个WaitGroup的Add()方法,避免计数竞争问题
总结
sync.WaitGroup是支持安全重用的,但必须严格遵守上一轮Wait()返回后才能再次使用的规则。实际开发中,要么严格保证重用的时序正确,要么每一轮都创建新的WaitGroup实例,两种方式都可以避免并发场景下的异常。理解它的底层计数逻辑,能帮助开发者更好地规避使用风险,写出更稳定的Go并发代码。
sync.WaitGroupGo并发编程goroutineWaitGroup重用修改时间:2026-07-09 03:51:28