WaitGroup用于等待多个goroutine完成,通过Add增加计数、Done减少计数、Wait阻塞直到计数为零,确保主协程正确同步子任务。

在Go语言中,WaitGroup 是
sync
包提供的一个同步原语,用于等待一组并发的 goroutine 完成任务。它非常适合用于主协程需要等待多个子协程执行完毕后再继续的场景。下面通过实际例子说明如何正确使用 WaitGroup 实现多个 goroutine 的同步。
WaitGroup 基本机制
WaitGroup 内部维护一个计数器:
Add(n):增加计数器的值,表示要等待 n 个 goroutineDone():在每个 goroutine 结束时调用,相当于计数器减1Wait():阻塞主协程,直到计数器归零
典型使用模式是:主协程启动多个 goroutine 前调用 Add,每个子协程结束前调用 Done,主协程调用 Wait 等待所有完成。
基本使用示例
以下是一个简单的并发下载模拟,使用 WaitGroup 等待所有任务完成:
立即学习“go语言免费学习笔记(深入)”;
package mainimport ("fmt""sync""time")
func downloadFile(id int, wg sync.WaitGroup) {defer wg.Done() // 任务完成,计数器减1fmt.Printf("开始下载文件 %dn", id)time.Sleep(2 time.Second) // 模拟耗时操作fmt.Printf("完成下载文件 %dn", id)}
func main() {var wg sync.WaitGroup
for i := 1; i <= 3; i++ { wg.Add(1) // 增加等待计数 go downloadFile(i, &wg)}wg.Wait() // 阻塞,直到所有 goroutine 调用 Donefmt.Println("所有文件下载完成")
}
输出结果会显示三个任务并发执行,最后统一输出“所有文件下载完成”。
常见使用注意事项
使用 WaitGroup 时有几个关键点需要注意,避免出现死锁或 panic:
确保 Done 调用次数与 Add 一致:少调用会导致 Wait 永不返回,多调用会 panic通常使用 defer wg.Done():保证即使发生 panic 也能正确释放计数WaitGroup 不应被复制:传递给 goroutine 时应传指针Add 应在 Wait 之前调用:否则可能引发竞态条件
结合通道的扩展用法
有时除了等待完成,还需要收集结果。可以结合 channel 和 WaitGroup 使用:
func processTasks(tasks []int) []string { var wg sync.WaitGroup results := make(chan string, len(tasks))for _, task := range tasks { wg.Add(1) go func(t int) { defer wg.Done() result := fmt.Sprintf("处理任务 %d 完成", t) results <- result }(task)}go func() { wg.Wait() close(results)}()var output []stringfor res := range results { output = append(output, res)}return output
}
这种方式既能并发执行,又能安全地收集所有结果,避免主协程提前退出导致 goroutine 被中断。
基本上就这些。WaitGroup 是 Go 并发编程中最基础也最常用的同步工具之一,掌握其正确用法对编写可靠的并发程序至关重要。
以上就是GolangWaitGroup同步多个goroutine实践的详细内容,更多请关注创想鸟其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1406293.html
微信扫一扫
支付宝扫一扫