Golang如何实现WaitGroup管理多个goroutine_Golang WaitGroup使用项目实践

答案：sync.WaitGroup用于等待多个goroutine完成，通过Add、Done、Wait方法实现同步。典型应用场景为并行处理批量任务，如并发请求用户数据，需注意避免在goroutine内调用Add、确保Add与Done配对、防止复制使用WaitGroup。结合context可实现超时控制，避免无限阻塞，提升程序健壮性。

在Go语言开发中，经常需要并发执行多个任务并等待它们全部完成。这时 sync.WaitGroup 就成了最常用的同步原语之一。它能有效管理多个goroutine的生命周期，确保主协程在所有子协程执行完毕后再退出。

WaitGroup 基本原理与核心方法

WaitGroup 本质是一个计数器，用来等待一组操作完成。它有三个关键方法：

• Add(delta int)：增加或减少计数器值。通常在启动goroutine前调用 Add(1)。
• Done()：将计数器减1，一般放在goroutine末尾，可配合 defer 使用。
• Wait()：阻塞当前协程，直到计数器归零。

使用时需注意：Add 必须在 Wait 调用前执行，否则可能引发 panic。常见模式是在主协程中 Add，每个子协程结束时调用 Done，主协程通过 Wait 阻塞等待。

实际项目中的典型应用场景

在Web服务中，常需并行处理多个外部API请求或数据库查询。例如批量获取用户信息：

立即学习“go语言免费学习笔记（深入）”；

var wg sync.WaitGroup
results := make([]UserData, len(userIDs))
for i, uid := range userIDs {
wg.Add(1)
go func(index int, userID string) {
defer wg.Done()
data, err := fetchUserData(userID)
if err != nil {
log.Printf("获取用户 %s 失败: %v", userID, err)
return
}
results[index] = data
}(i, uid)
}
wg.Wait()
// 此时 results 已填充完成

这种写法显著提升响应速度。注意闭包中传参要避免变量共享问题，应将循环变量作为参数传入goroutine。

ViiTor实时翻译

AI实时多语言翻译专家！强大的语音识别、AR翻译功能。

116

查看详情

避免常见错误和性能陷阱

WaitGroup 虽简单，但误用容易导致死锁或panic：

• 不要在goroutine内部调用 Add，可能导致Wait提前结束。应在外部先 Add 再启动协程。
• 确保每个 Add 都有对应的 Done，漏掉会导致永久阻塞。
• 不可复制已使用的 WaitGroup，会导致数据竞争。
• 在HTTP处理器等高并发场景中，避免频繁创建WaitGroup带来的开销，可结合 context 控制超时。

若任务数量不确定或动态变化，考虑搭配 channel 或 errgroup 等更高级控制结构。

结合 context 实现超时控制

生产环境中，不能无限等待。可结合 context 实现安全超时：

ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
defer cancel()
done := make(chan struct{})
go func() {
var wg sync.WaitGroup
for _, task := range tasks {
wg.Add(1)
go func(t Task) {
defer wg.Done()
process(t)
}(task)
}
wg.Wait()
close(done)
}()
select {
case <-done:
// 所有任务成功完成
case <-ctx.Done():
// 超时或被取消
log.Println("任务执行超时")
}

这种方式既保证了并发效率，又具备容错能力。

基本上就这些。WaitGroup 是Go并发编程的基石工具，掌握其正确用法对构建稳定服务至关重要。不复杂但容易忽略细节。