WaitGroup用于等待多个goroutine完成,通过Add增加计数、Done减少计数、Wait阻塞直至归零,示例中并发下载网页并等待全部完成,需注意Add与Done配对、避免提前Wait及在goroutine内调用Done。

在Go语言中,WaitGroup 是 sync 包提供的一个同步原语,用于等待一组并发的 goroutine 完成任务。它非常适合用于主协程需要等待多个子协程执行完毕后再继续的场景。
WaitGroup 基本用法
WaitGroup 有三个核心方法:
- Add(n int):增加计数器,表示要等待的 goroutine 数量
- Done():减少计数器,通常在 goroutine 结束时调用
- Wait():阻塞主协程,直到计数器归零
下面是一个使用 WaitGroup 实现并发任务同步的简单示例:
并发下载多个网页
假设我们要并发地从多个 URL 下载内容,并等待所有下载完成:
Modoer 是一款以本地分享,多功能的点评网站管理系统。采用 PHP+MYSQL 开发设计,开放全部源代码。因具有非凡的访问速度和卓越的负载能力而深受国内外朋友的喜爱,不局限于商铺类点评,真正实现了多类型的点评,可以让您的网站点评任何事与物,同时增加产品模块,也更好的网站产品在网站上展示。Modoer点评系统 2.5 Build 20110710更新列表1.同步 旗舰版系统框架2.增加 限制图片
立即学习“go语言免费学习笔记(深入)”;
package main
import (
"fmt"
"io/ioutil"
"net/http"
"sync"
)
func fetch(url string, wg *sync.WaitGroup) {
defer wg.Done() // 任务完成,计数器减1
fmt.Printf("开始获取: %s\n", url)
resp, err := http.Get(url)
if err != nil {
fmt.Printf("请求失败 %s: %v\n", url, err)
return
}
defer resp.Body.Close()
body, _ := ioutil.ReadAll(resp.Body)
fmt.Printf("成功获取 %s,响应长度: %d\n", url, len(body))
}
func main() {
urls := []string{
"https://httpbin.org/delay/1",
"https://httpbin.org/status/200",
"https://httpbin.org/headers",
}
var wg sync.WaitGroup
for _, url := range urls {
wg.Add(1) // 每启动一个 goroutine,计数加1
go fetch(url, &wg) // 并发执行
}
wg.Wait() // 等待所有任务完成
fmt.Println("所有任务已完成")
}
常见使用注意事项
使用 WaitGroup 时需要注意以下几点,避免出现死锁或 panic:
- 确保每个 Add 都有对应的 Done,否则可能造成永久阻塞
- 不要在 goroutine 外部调用 Done,应放在 goroutine 内部并通过指针传递 WaitGroup
- 避免在 Add 调用之前就执行 Wait,否则可能漏掉某些任务
- 建议使用 defer wg.Done() 确保即使发生 panic 也能正确计数
基本上就这些。WaitGroup 是实现简单并发控制非常实用的工具,适用于批量任务处理、并行计算等场景,掌握它的正确用法能有效提升 Go 程序的并发效率和稳定性。









