WaitGroup用于等待多个goroutine完成,通过Add增加计数、Done减少计数、Wait阻塞直到计数为零,确保主协程在所有子任务结束后再继续执行。

在 Go 语言中,WaitGroup 是 sync 包提供的一个同步原语,用于等待一组并发的 goroutine 执行完成。它特别适用于主协程需要等待多个子协程结束后再继续执行的场景,比如批量处理任务、并发请求聚合等。
WaitGroup 的核心作用
当启动多个 goroutine 做并行任务时,主协程通常不能立即退出,否则会终止所有子协程。WaitGroup 提供了一种机制,让主协程“等待”所有子任务完成后再继续或退出。
其本质是一个计数器:
每启动一个 goroutine,调用 Add(1) 增加计数 每个 goroutine 执行完后,调用 Done() 减少计数 主协程调用 Wait() 阻塞,直到计数归零
基本使用方法
下面是一个典型的 WaitGroup 使用示例:
立即学习“go语言免费学习笔记(深入)”;
package mainimport ( "fmt" "sync" "time")func worker(id int, wg *sync.WaitGroup) { defer wg.Done() // 任务完成,计数减一 fmt.Printf("Worker %d startingn", id) time.Sleep(time.Second) fmt.Printf("Worker %d donen", 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 startingWorker 2 startingWorker 3 startingWorker 1 doneWorker 2 doneWorker 3 doneAll workers finished
使用注意事项
WaitGroup 虽然简单,但使用不当容易出错。以下几点需特别注意:
Add 的值不能为负数:否则会 panic 确保 Done 调用次数与 Add 一致:漏调或多调都会导致死锁或 panic WaitGroup 不是可复制类型:不要将其作为参数值传递,应传指针 Add 应在 Wait 之前调用:如果在 Wait 后 Add,可能引发竞态条件
常见误用与修复
错误写法:在 goroutine 内部才 Add
for i := range tasks { go func() { wg.Add(1) // 错误!Add 在 goroutine 内 defer wg.Done() // ... }()}wg.Wait()
这可能导致主协程已进入 Wait,而 Add 还未执行,从而漏记协程。
正确做法:在 goroutine 启动前调用 Add
for i := range tasks { wg.Add(1) go func() { defer wg.Done() // ... }()}wg.Wait()
基本上就这些。WaitGroup 是 Go 并发编程中最基础也最实用的同步工具之一,掌握它的使用和边界条件,能有效避免程序提前退出或死锁问题。关键是控制好计数的增减时机,保证一致性。不复杂但容易忽略细节。
以上就是Golang 中的 WaitGroup 有什么用_Golang 并发协程等待机制详解的详细内容,更多请关注创想鸟其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1423365.html
微信扫一扫
支付宝扫一扫