如何在Golang中使用WaitGroup等待任务完成

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

如何在golang中使用waitgroup等待任务完成

在Go语言中,WaitGroupsync 包提供的一个同步原语,用于等待一组并发的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 Otter.ai

一个自动的会议记录和笔记工具,会议内容生成和实时转录

Otter.ai 91 查看详情 Otter.ai 确保每次 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

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
自适应设计中如何解决高度调整导致的界面错位问题?
上一篇 2025年12月2日 13:45:53
无需Photoshop打开PSD文件
下一篇 2025年12月2日 13:45:59

相关推荐

发表回复

登录后才能评论
关注微信