您的位置:首页 >如何在Golang中使用WaitGroup等待任务完成
发布于2025-11-18 阅读(0)
扫一扫,手机访问
WaitGroup用于等待多个goroutine完成,通过Add增加计数、Done减少计数、Wait阻塞直至计数为零,需确保Add与Done配对使用,避免并发错误。

在Go语言中,WaitGroup 是 sync 包提供的一个同步原语,用于等待一组并发的goroutine执行完成。它特别适用于主协程需要等待多个子任务结束后再继续的场景。
WaitGroup 有三个核心方法:
Add(delta int):增加计数器,通常传入要启动的goroutine数量。Done():在每个goroutine结束时调用,表示该任务已完成(相当于 Add(-1))。Wait():阻塞当前协程,直到计数器归零。典型使用模式如下:
var wg sync.WaitGroup
for 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 时有几个关键点需要注意:
Add 都有对应的 Done,否则程序可能永久阻塞或 panic。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.WaitGroup
urls := []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,就能正确等待所有任务结束。不复杂但容易忽略细节。
售后无忧
立即购买>office旗舰店
售后无忧
立即购买>office旗舰店
售后无忧
立即购买>office旗舰店
售后无忧
立即购买>office旗舰店
正版软件
正版软件
正版软件
正版软件
正版软件
1
2
3
7
9