答案:sync.WaitGroup用于等待一组goroutine完成,通过Add增加计数,Done减少计数,Wait阻塞直至计数归零。示例中启动5个worker,主协程等待全部完成。

在Golang中,sync.WaitGroup 是一种常用的同步机制,用于等待一组并发的goroutine执行完成。它特别适合在主函数或主线程中启动多个子任务,并确保所有任务都结束后再继续执行后续逻辑。
基本原理
WaitGroup通过计数器来跟踪正在运行的goroutine数量。调用 Add(n) 增加计数器,每个goroutine执行完成后调用 Done() 减少计数器,而 Wait() 会阻塞直到计数器归零。
使用步骤和示例
以下是使用 WaitGroup 的典型流程:创建 WaitGroup 变量:通常是一个指针或值类型变量。 在启动goroutine前调用 Add:每次启动一个goroutine就调用 Add(1)。 在每个goroutine末尾调用 Done:确保无论正常结束还是出错都能触发 Done。 在需要等待的地方调用 Wait:主协程等待所有任务完成。
示例代码:
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) // 启动goroutine } wg.Wait() // 阻塞,直到所有worker调用Done fmt.Println("All workers finished")}
注意事项
正确使用 WaitGroup 要注意以下几点:避免重复调用 Done:每个 goroutine 只应调用一次 Done,否则会导致 panic。 Add 的值不能为负:除了初始设置外,Add传入负数会引发错误。 确保 Wait 在所有 Add 之后调用:如果 Wait 先于 Add 执行,可能无法正确捕获所有任务。 传递 WaitGroup 应使用指针:在函数间传递时,应传指针以避免副本问题。
常见误用与修复
一个常见错误是将 wg 以值方式传入函数,导致每个 goroutine 操作的是副本:
// 错误示例go func(wg sync.WaitGroup) { // 值传递,复制了wg defer wg.Done()}(wg)
应改为指针传递:
go func(wg *sync.WaitGroup) { defer wg.Done()}(&wg)
基本上就这些。只要掌握 Add、Done、Wait 的配合,就能安全地协调并发任务的完成。
以上就是如何在Golang中使用sync.WaitGroup等待并发完成的详细内容,更多请关注创想鸟其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1414582.html
微信扫一扫
支付宝扫一扫