跳转到主内容
极星编程网:以代码为星,赴技术山海!

Golang 中的 WaitGroup 有什么用_Golang 并发协程等待机制详解

WaitGroup用于等待多个goroutine完成,通过Add增加计数、Done减少计数、Wait阻塞直到计数为零,确保主协程在所有子任务结束后再继续执行。 在 Go 语言中, WaitGroup 是 sync 包提供的一个同步原语,用于等待一组并发的 goroutine 执行完成。它特别适用于主协程需要等待多个子协程结束后再继续执行的场景,比如批量处理任务、 并发请求 聚合等。 WaitGroup 的核心作用 当启动多个 goroutine 做并行任务时,主协程通常不能立即退出,否则会终止所有子协程。WaitGroup 提供了一种机制,让主协程“等待”所有子任务完成后再继续或退出。 其本质是一个计数器: 每启动一个 goroutine,调用 Add(1) 增加计数 每个 goroutine 执行完后,调用 Done() 减少计数 主协程调用 Wait() 阻塞,直到计数归零 基本使用方法 下面是一个典型的 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") }
输出结果: 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数组安全、编译 下载
Worker 1 starting Worker 2 starting Worker 3 starting Worker 1 done Worker 2 done Worker 3 done All 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 并发编程中最基础也最实用的同步工具之一,掌握它的使用和边界条件,能有效避免程序提前退出或死锁问题。关键是控制好计数的增减时机,保证一致性。不复杂但容易忽略细节。

相关文章