WaitGroup用于协调多个goroutine的完成,通过Add增加计数、Done减少计数、Wait阻塞等待归零。示例中三个worker并发执行,主协程等待它们完成后再退出。需注意Add在goroutine外调用、传递指针、Add与Done匹配,避免重复Wait。适用于批量任务同步场景。

在Go语言中,WaitGroup 是 sync 包提供的同步原语之一,用于等待一组并发的协程(goroutine)完成任务。它特别适用于主协程需要等待其他多个子协程执行完毕后再继续的场景。
WaitGroup 基本原理
WaitGroup 内部维护一个计数器:
- Add(n):将计数器加上 n,通常用于增加待处理的协程数量。
- Done():将计数器减1,一般在协程结束前调用,表示该协程已完成。
- Wait():阻塞当前协程,直到计数器归零。
通过这三个方法,可以协调主协程和其他协程的生命周期,避免程序提前退出或资源竞争。
基本使用示例
以下是一个简单的例子,展示如何使用 WaitGroup 等待多个协程完成:
立即学习“go语言免费学习笔记(深入)”;
JTBC CMS(5.0) 是一款基于PHP和MySQL的内容管理系统原生全栈开发框架,开源协议为AGPLv3,没有任何附加条款。系统可以通过命令行一键安装,源码方面不基于任何第三方框架,不使用任何脚手架,仅依赖一些常见的第三方类库如图表组件等,您只需要了解最基本的前端知识就能很敏捷的进行二次开发,同时我们对于常见的前端功能做了Web Component方式的封装,即便是您仅了解HTML/CSS也
package mainimport ( "fmt" "sync" "time" )
func worker(id int, wg sync.WaitGroup) { defer wg.Done() // 任务完成,计数器减1 fmt.Printf("Worker %d starting\n", id) time.Sleep(2 time.Second) fmt.Printf("Worker %d done\n", id) }
func main() { var wg sync.WaitGroup
for i := 1; i <= 3; i++ { wg.Add(1) // 每启动一个协程,计数器加1 go worker(i, &wg) // 启动协程 } wg.Wait() // 主协程等待所有协程完成 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注意事项和最佳实践
使用 WaitGroup 时需注意以下几点,避免常见错误:
- 确保 Add 在协程启动前调用:如果在协程内部才调用 Add,可能因调度问题导致计数器未及时更新,引发 panic。
- 每个 Add 对应一次 Done:Add 和 Done 的调用次数必须匹配,否则可能死锁或 panic。
- 传递 WaitGroup 指针:多个协程应共享同一个 WaitGroup 实例,因此需传指针,避免值拷贝。
- 避免重复 Wait:Wait 只能安全调用一次。若需多次等待不同组任务,应使用多个 WaitGroup 或重置控制逻辑。
适用场景
WaitGroup 常用于以下情况:
- 批量发起网络请求,并行处理后统一返回。
- 初始化多个服务组件,等待全部准备就绪。
- 数据分片处理,如并行计算、文件读取等。
基本上就这些。WaitGroup 使用简单,但对协程生命周期管理非常有效。只要注意调用时机和同步方式,就能写出稳定高效的并发代码。









