WaitGroup通过Add、Done、Wait方法实现goroutine同步，主协程调用Add设置任务数，子协程完成时调用Done，主协程Wait阻塞直至所有任务结束，常用于并发请求、批量处理等场景。
在Go语言开发中，WaitGroup 是 sync 包提供的一个非常实用的同步原语，用于等待一组并发的 goroutine 执行完成。它特别适用于主协程需要等待多个子任务结束后再继续执行的场景。掌握 WaitGroup 的正确用法，是编写高效、稳定并发程序的基础。
WaitGroup 核心机制
WaitGroup 内部维护一个计数器，通过三个方法控制：
- Add(n)：增加计数器值，通常用于添加要等待的 goroutine 数量
- Done()：计数器减1，一般在 goroutine 结束时调用
- Wait()：阻塞当前协程，直到计数器归零
典型使用模式是：主协程调用 Add 设置任务数量，每个子协程执行完后调用 Done，主协程通过 Wait 阻塞等待所有任务完成。
基本使用示例
以下是一个简单的并行HTTP请求示例：
立即学习“go语言免费学习笔记（深入）”；
package main
<p>import (
"fmt"
"net/http"
"sync"
)</p><p>func fetchURL(url string, wg *sync.WaitGroup) {
defer wg.Done()</p><pre class="brush:php;toolbar:false;">resp, err := http.Get(url)
if err != nil {
fmt.Printf("Error fetching %s: %v\n", url, err)
return
}
defer resp.Body.Close()
fmt.Printf("Success: %s -> Status %d\n", url, resp.StatusCode)}
func main() { urls := []string{ "https://www.php.cn/link/4d2fe2e8601f7a8018594d98f28706f2", "https://www.php.cn/link/5f69e19efaba426d62faeab93c308f5c", "https://www.php.cn/link/98a733901e53052474f2320d0a3a9473", }
var wg sync.WaitGroup
for _, url := range urls {
wg.Add(1)
go fetchURL(url, &wg)
}
wg.Wait()
fmt.Println("All requests completed.")}
在这个例子中，每次启动一个 goroutine 前调用 Add(1)，并在函数末尾通过 defer 调用 Done() 确保计数器正确减少。
使用 uber-go/dig 在 Golang 中实现依赖注入——基于反射容器，提供 Provide/Invoke、dig.In/dig.Out 参数与结果对象，支持命名依赖等
常见陷阱与最佳实践
使用 WaitGroup 时有几个关键点需要注意：
- 避免在 goroutine 内部调用 Add，否则可能造成竞争或漏加
- 确保每个 Add 都有对应的 Done，否则会永久阻塞
- 传递 WaitGroup 指针而非值，防止拷贝导致状态不一致
- 合理使用 defer Done()，确保即使发生 panic 也能正确计数
错误示例：在 goroutine 中 Add 会导致不可预测行为
// ❌ 错误做法
go func() {
wg.Add(1) // 危险！可能在 Wait 后才执行
defer wg.Done()
// ...
}()
正确方式是在启动 goroutine 前完成 Add 操作。
进阶应用场景
WaitGroup 可与其他并发结构组合使用。例如配合 channel 实现超时控制：
func doWithTimeout() bool {
var wg sync.WaitGroup
done := make(chan bool, 1)
<pre class="brush:php;toolbar:false;">for i := 0; i < 5; i++ {
wg.Add(1)
go func(id int) {
defer wg.Done()
// 模拟耗时任务
// time.Sleep(time.Second)
fmt.Printf("Worker %d done\n", id)
}(i)
}
go func() {
wg.Wait()
done <- true
}()
select {
case <-done:
return true
case <-time.After(3 * time.Second):
return false // 超时
}}
这种模式将 WaitGroup 的等待逻辑放到单独的协程中，使主流程能支持超时退出。
基本上就这些。WaitGroup 虽然简单，但用好并不容易。关键是理解其计数机制，避免竞态，并结合实际场景灵活运用。在真实项目中，它常用于批量任务处理、服务启动关闭、数据预加载等需要协同等待的场合。
