首先初始化WaitGroup,再通过Add增加计数,每个goroutine执行完调用Done,主线程调用Wait阻塞直至所有任务完成。

在Go语言中,sync.WaitGroup 是一种常用的协程同步机制,用于等待一组并发的goroutine执行完成。它特别适用于主线程需要等待多个子任务结束的场景,比如批量发起网络请求、并行处理数据等。使用 WaitGroup 可以避免程序提前退出,确保所有协程都执行完毕。
WaitGroup 核心方法
WaitGroup 提供了三个主要方法:
Add(delta int):增加计数器的值,通常用于添加要等待的goroutine数量。 Done():将计数器减1,一般在goroutine结束时调用,相当于 Add(-1)。 Wait():阻塞当前协程,直到计数器变为0。
基本使用示例
下面是一个简单的例子,展示如何使用 WaitGroup 等待多个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) // 每启动一个goroutine,计数器加1 go worker(i, &wg) } wg.Wait() // 等待所有goroutine完成 fmt.Println("All workers finished")}
输出结果:
立即学习“go语言免费学习笔记(深入)”;
Worker 1 starting
Worker 2 starting
Worker 3 starting
Worker 1 done
Worker 2 done
Worker 3 done
All workers finished
使用注意事项
虽然 WaitGroup 使用简单,但有几个关键点需要注意,避免常见错误:
避免Add负数或超出范围:在调用 Add 时传入负数会引发 panic,必须确保 Add 的值合理。 确保 Done 被调用:每个 Add 对应一次 Done,否则 Wait 会一直阻塞,导致死锁。 传递指针:WaitGroup 应以指针形式传递给goroutine,否则值拷贝会导致各个协程操作的是不同的实例。 Add 应在 go 之前调用:如果在goroutine内部才 Add,可能主线程的 Wait 已经开始,导致遗漏。
实际应用场景
WaitGroup 常用于并行任务处理。例如,从多个API获取数据:
func fetchAllData() { var wg sync.WaitGroup urls := []string{"http://a.com", "http://b.com", "http://c.com"} for _, url := range urls { wg.Add(1) go func(u string) { defer wg.Done() resp, err := http.Get(u) if err != nil { fmt.Printf("Error: %sn", err) return } defer resp.Body.Close() fmt.Printf("Fetched %sn", u) }(url) } wg.Wait() fmt.Println("All requests completed")}
这段代码并发请求多个URL,并等待全部完成后再继续。
基本上就这些。WaitGroup 是Go中最基础也最实用的同步工具之一,掌握它的正确用法对编写可靠的并发程序至关重要。不复杂但容易忽略细节。
以上就是Golang协程同步方法 sync.WaitGroup实践的详细内容,更多请关注创想鸟其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1401344.html
微信扫一扫
支付宝扫一扫