WaitGroup用于等待多个goroutine完成,通过Add增加计数、Done减少计数、Wait阻塞直至计数为零,需确保Add与Done配对使用,避免并发错误。

在Go语言中,WaitGroup 是 sync 包提供的一个同步原语,用于等待一组并发的goroutine执行完成。它特别适用于主协程需要等待多个子任务结束后再继续的场景。
WaitGroup 基本用法
WaitGroup 有三个核心方法:
Add(delta int):增加计数器,通常传入要启动的goroutine数量。 Done():在每个goroutine结束时调用,表示该任务已完成(相当于 Add(-1))。 Wait():阻塞当前协程,直到计数器归零。
典型使用模式如下:
var wg sync.WaitGroupfor i := 0; i < 5; i++ { wg.Add(1) go func(id int) { defer wg.Done() // 模拟任务执行 fmt.Printf("任务 %d 完成\n", id) }(i)}wg.Wait() // 等待所有任务完成fmt.Println("所有任务已结束")
避免常见错误
使用 WaitGroup 时有几个关键点需要注意:
立即学习“go语言免费学习笔记(深入)”;
Otter.ai
一个自动的会议记录和笔记工具,会议内容生成和实时转录
91 查看详情
确保每次 Add 都有对应的 Done,否则程序可能永久阻塞或 panic。 不要在 goroutine 外部调用 Done,应放在 goroutine 内部并配合 defer 使用。 Add 调用可以在启动 goroutine 前完成,但不能晚于 Wait。
例如,下面这种写法是安全的:
wg.Add(3)go task(&wg)go task(&wg)go task(&wg)wg.Wait()
实际应用场景
WaitGroup 常用于批量请求、数据抓取、并行处理等场景。比如同时发起多个HTTP请求:
var wg sync.WaitGroupurls := []string{"http://example.com", "http://httpbin.org", "http://httpstat.us"}for _, url := range urls { wg.Add(1) go func(u string) { defer wg.Done() resp, err := http.Get(u) if err != nil { fmt.Printf("请求失败 %s: %v\n", u, err) return } defer resp.Body.Close() fmt.Printf("成功获取 %s, 状态码: %s\n", u, resp.Status) }(url)}wg.Wait()fmt.Println("所有请求完成")
基本上就这些。只要记得配对使用 Add 和 Done,并在主线程调用 Wait,就能正确等待所有任务结束。不复杂但容易忽略细节。
以上就是如何在Golang中使用WaitGroup等待任务完成的详细内容,更多请关注创想鸟其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1114211.html
微信扫一扫
支付宝扫一扫