WaitGroup用于同步等待多个Goroutine完成,通过Add增加计数、Done减少计数、Wait阻塞直至计数为零,适用于并发任务控制。

在Go语言中,WaitGroup 是一种常用的同步机制,用于等待一组并发执行的 Goroutine 完成任务。它属于 sync 包,适用于主线程需要等待多个子任务结束后再继续执行的场景。这种模式常用于批量处理、并发请求、数据采集等。
sync.WaitGroup 内部维护一个计数器,用来记录正在运行的 Goroutine 数量。主要提供三个方法:
Add(-1)。以下是一个使用 WaitGroup 等待多个 Goroutine 完成的示例:
package main
import (
"fmt"
"sync"
"time"
)
func worker(id int, wg *sync.WaitGroup) {
defer wg.Done() // 任务完成,计数器减1
fmt.Printf("Worker %d starting\n", id)
time.Sleep(time.Second)
fmt.Printf("Worker %d done\n", id)
}
func main() {
var wg sync.WaitGroup
for i := 1; i <= 3; i++ {
wg.Add(1) // 增加计数器
go worker(i, &wg)
}
wg.Wait() // 阻塞,直到所有 worker 调用 Done()
fmt.Println("All workers completed")
}
输出结果为:
立即学习“go语言免费学习笔记(深入)”;
Worker 1 starting Worker 2 starting Worker 3 starting Worker 1 done Worker 2 done Worker 3 done All workers completed
在这个例子中,主线程通过 wg.Wait() 等待所有 worker 执行完毕,确保最后的提示信息在所有任务结束后才打印。
使用 WaitGroup 时需注意以下几点以避免死锁或 panic:
go 语句之后调用 Add,可能因竞态导致计数器未及时更新,造成漏计。Wait 永不返回或 panic。*sync.WaitGroup 的指针传入函数,避免值拷贝导致无法正确同步。虽然也可以使用 channel 实现类似功能(如通过关闭 channel 广播完成信号),但 WaitGroup 更简洁直观,特别适合“只关心完成、不传递数据”的场景。而 channel 更适合需要传递状态或结果的复杂同步逻辑。
基本上就这些。WaitGroup 是 Go 中轻量且高效的同步工具,掌握它的正确用法对编写可靠的并发程序至关重要。
以上就是Golang如何使用WaitGroup等待多个Goroutine完成_Golang WaitGroup同步模式解析的详细内容,更多请关注php中文网其它相关文章!
每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号