sync.WaitGroup用于等待多个goroutine完成,通过Add增加计数,Done减少计数,Wait阻塞直至计数为零,确保主协程正确同步子协程执行。

在Golang中,sync.WaitGroup 是一种常用的同步机制,用于等待一组并发的协程(goroutines)执行完成。它特别适合在主协程需要等待多个子协程结束时使用。使用 WaitGroup 可以避免程序提前退出或资源未释放的问题。
WaitGroup 的基本用法
WaitGroup 有三个核心方法:
Add(delta int):增加计数器,通常传入要启动的协程数量。 Done():在协程结束时调用,将计数器减1。 Wait():阻塞主协程,直到计数器归零。
典型使用模式如下:
// 示例:等待3个协程完成
var wg sync.WaitGroup
立即学习“go语言免费学习笔记(深入)”;
for i := 0; i
wg.Add(1) // 每次启动前加1
go func(id int) {
defer wg.Done() // 结束时减1
fmt.Printf(“协程 %d 开始工作n”, id)
time.Sleep(time.Second)
fmt.Printf(“协程 %d 完成n”, id)
} (i)
}
wg.Wait() // 等待所有协程完成
fmt.Println(“所有协程执行完毕”)
常见使用场景和注意事项
WaitGroup 适用于批量启动协程并统一等待的场景,比如并发请求、数据处理等。
确保每次 Add 都在 go 语句之前调用,防止竞争条件。 推荐在协程内部使用 defer wg.Done(),确保即使发生 panic 也能正确计数。 不能对已归零的 WaitGroup 执行 Done(),否则会 panic。 WaitGroup 通常不应被复制,应以指针方式传递给函数。
传递 WaitGroup 到函数中的正确方式
如果需要将 WaitGroup 传入函数,应传递指针:
func doWork(wg *sync.WaitGroup) {
defer wg.Done()
// 执行任务
}
// 调用时
wg.Add(1)
go doWork(&wg)
这样可以避免值拷贝导致的状态不一致问题。
基本上就这些。只要注意 Add 和 Done 的配对,以及调用时机,WaitGroup 就能可靠地帮你管理协程生命周期。
以上就是如何在Golang中使用sync.WaitGroup等待协程完成的详细内容,更多请关注创想鸟其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1417950.html
微信扫一扫
支付宝扫一扫