WaitGroup用于等待协程完成,通过Add、Done、Wait方法实现同步。示例中启动3个协程,主协程等待其全部完成。

在Go语言中,协程(goroutine)是实现并发的核心机制。但多个协程并行执行时,主程序可能在协程完成前就退出,导致任务丢失。为解决这个问题,sync.WaitGroup 是最常用的协程同步工具。
WaitGroup 是什么?
WaitGroup 用于等待一组协程完成任务。它内部维护一个计数器,通过三个方法控制流程:
Add(n):增加计数器,表示要等待 n 个协程 Done():计数器减1,通常在协程末尾调用 Wait():阻塞主协程,直到计数器归零
基本使用示例
下面是一个简单的例子,启动3个协程并发执行任务,主协程等待它们全部完成:
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) // 每启动一个协程,计数+1 go worker(i, &wg) } wg.Wait() // 阻塞,直到所有协程调用 Done() fmt.Println("All workers finished")}
输出结果会是三个 worker 依次打印开始和完成,最后输出 “All workers finished”。
立即学习“go语言免费学习笔记(深入)”;
常见使用技巧和注意事项
使用 WaitGroup 时,有几个关键点要注意,避免死锁或 panic:
确保每个 Add 对应一个 Done,否则 Wait 会一直阻塞 避免在协程中调用 Add,可能导致主协程还没 Wait 就结束 通常将 WaitGroup 指针传入协程函数,避免值拷贝 使用 defer wg.Done() 可确保即使发生 panic 也能正确计数
错误示例(可能导致死锁):
go func() { wg.Add(1) // ❌ 不应在协程内 Add defer wg.Done() // ...}()
正确做法是在 go 之前调用 Add:
wg.Add(1)go func() { defer wg.Done() // 任务逻辑}()
实际应用场景
WaitGroup 常用于批量处理任务,比如并发请求多个API、并行处理文件、爬虫抓取等。
例如,同时请求多个外部服务:
urls := []string{"http://example1.com", "http://example2.com"}var wg sync.WaitGroupfor _, 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 } fmt.Printf("Success: %s, Status: %sn", u, resp.Status) resp.Body.Close() }(url)}wg.Wait()fmt.Println("All requests completed")
基本上就这些。WaitGroup 简单高效,是Go并发编程中最基础也最重要的同步手段之一。用好它,就能安全地管理多个协程的生命周期。
以上就是Golang协程同步怎么做 sync.WaitGroup实践的详细内容,更多请关注创想鸟其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1400378.html
微信扫一扫
支付宝扫一扫