WaitGroup用于等待多个goroutine完成,通过Add增加计数、Done减少计数、Wait阻塞直到计数为零,确保主协程在所有子任务结束后再继续执行。
在 Go 语言中,
WaitGroup
是 sync 包提供的一个同步原语,用于等待一组并发的 goroutine 执行完成。它特别适用于主协程需要等待多个子协程结束后再继续执行的场景,比如批量处理任务、
并发请求
聚合等。
WaitGroup 的核心作用
当启动多个 goroutine 做并行任务时,主协程通常不能立即退出,否则会终止所有子协程。WaitGroup 提供了一种机制,让主协程“等待”所有子任务完成后再继续或退出。
其本质是一个计数器:
每启动一个 goroutine,调用
Add(1)
增加计数
每个 goroutine 执行完后,调用
Done()
减少计数
主协程调用
Wait()
阻塞,直到计数归零
基本使用方法
下面是一个典型的 WaitGroup 使用示例:
立即学习
“
go语言免费学习笔记(深入)
”;
输出结果:
go语言参考手册 中文CHM版
Go 是一个开源的编程语言,它能让构造简单、可靠且高效的软件变得容易。本文给大家带来Go参考手册,需要的可以来下载! Go是从2007年末由Robert Griesemer, Rob Pike, Ken Thompson主持开发,后来还加入了Ian Lance Taylor, Russ Cox等人,并最终于2009年11月开源,在2012年早些时候发布了Go 1稳定版本。现在Go的开发已经是完全开放的,并且拥有一个活跃的社区。 Go 语言特色 简洁、快速、安全 并行、有趣、开源 内存管理、v数组安全、编译
下载
使用注意事项
WaitGroup 虽然简单,但使用不当容易出错。以下几点需特别注意:
Add 的值不能为负数
:否则会 panic
确保 Done 调用次数与 Add 一致
:漏调或多调都会导致死锁或 panic
WaitGroup 不是可复制类型
:不要将其作为参数值传递,应传指针
Add 应在 Wait 之前调用
:如果在 Wait 后 Add,可能引发竞态条件
常见误用与修复
错误写法:在 goroutine 内部才 Add
这可能导致主协程已进入 Wait,而 Add 还未执行,从而漏记协程。
正确做法:在 goroutine 启动前调用 Add
基本上就这些。WaitGroup 是 Go 并发编程中最基础也最实用的同步工具之一,掌握它的使用和边界条件,能有效避免程序提前退出或死锁问题。关键是控制好计数的增减时机,保证一致性。不复杂但容易忽略细节。
package main
import (
"fmt"
"sync"
"time"
)
func worker(id int, wg *sync.WaitGroup) {
defer wg.Done() // 任务完成,计数减一
fmt.Printf("Worker %d starting\n", id)
time.Sleep(time.Second)
fmt.Printf("Worker %d done\n", id)
}
func main() {
var wg sync.WaitGroup
for i := 1; i <= 3; i++ {
wg.Add(1) // 每次启动前增加计数
go worker(i, &wg)
}
wg.Wait() // 阻塞直到所有 Done 被调用
fmt.Println("All workers finished")
}
Worker 1 starting
Worker 2 starting
Worker 3 starting
Worker 1 done
Worker 2 done
Worker 3 done
All workers finished
for i := range tasks {
go func() {
wg.Add(1) // 错误!Add 在 goroutine 内
defer wg.Done()
// ...
}()
}
wg.Wait()
for i := range tasks {
wg.Add(1)
go func() {
defer wg.Done()
// ...
}()
}
wg.Wait()
