
在Go语言中,WaitGroup 是 sync 包提供的一个同步原语,用于等待一组并发的 goroutine 执行完成。它特别适合在主协程需要等待多个子协程结束时使用,而不需要关闭通道或处理复杂的信号机制。
基本用法和核心方法
WaitGroup 有三个主要方法:
Add(delta int):增加计数器,通常传入要启动的 goroutine 数量。 Done():在每个 goroutine 结束时调用,相当于计数器减1。 Wait():阻塞当前协程,直到计数器归零。
典型结构是:主协程调用 Add(n) 设置任务数量,每个子协程执行完后调用 Done(),主协程通过 Wait() 阻塞等待全部完成。
简单示例:等待多个goroutine完成
package mainimport ( "fmt" "sync" "time")func worker(id int, wg *sync.WaitGroup) { defer wg.Done() // 任务完成,计数器减1 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完成 fmt.Println("All workers finished")}
输出会显示三个 worker 依次开始和结束,最后打印“All workers finished”。
立即学习“go语言免费学习笔记(深入)”;
注意事项和常见错误
使用 WaitGroup 时需注意以下几点:
避免Add负值:调用 Add(-n) 会导致 panic,除非你明确知道计数器足够大。 确保Done被调用:每个 Add(1) 必须对应一次 Done(),否则程序会一直阻塞。 传递指针:应将 WaitGroup 的指针传给 goroutine,避免值拷贝导致状态不一致。 Add应在Wait前调用:最好在 go 语句前调用 Add,防止竞争条件。
适用场景
WaitGroup 适用于以下情况:
批量启动多个 goroutine 并等待它们全部完成(如并行请求、数据抓取)。 无需从 goroutine 获取返回值,只关心执行完成。 作为轻量级同步工具,比通道更简洁。
基本上就这些。只要记住 Add、Done、Wait 的配合使用,就能安全地控制并发流程。
以上就是Golang如何使用WaitGroup等待goroutine完成的详细内容,更多请关注创想鸟其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1419916.html
微信扫一扫
支付宝扫一扫