Golang协程等待技巧：WaitGroup使用详解

## Golang使用WaitGroup等待协程完成：原理、用法与最佳实践 在Golang并发编程中，`sync.WaitGroup`是实现协程同步的关键工具。它允许主协程等待一组子协程完成任务，适用于批量任务处理、并发请求等场景。本文深入解析`WaitGroup`的原理与用法，包括`Add`、`Done`和`Wait`三个核心方法的作用，并通过代码示例展示其典型使用结构和实际应用场景，例如并发获取API接口数据。同时，本文还探讨了使用`WaitGroup`时常见的错误与注意事项，以及与channel的对比选择，助你掌握这一Go语言并发编程中基础且实用的同步工具，提升程序性能和稳定性。了解如何正确使用`WaitGroup`，避免竞态条件、重复`Done`或忘记`Done`等问题，确保你的Golang程序高效可靠地运行。

sync.WaitGroup用于等待一组goroutine完成，通过Add增加计数、Done减少计数、Wait阻塞至计数归零，适用于批量任务同步，如并发请求处理。

在Go语言中，sync.WaitGroup 是一种常用的同步机制，用于等待一组并发的goroutine执行完成。它特别适合在主协程中启动多个子协程并等待它们全部结束的场景，比如批量任务处理、并发请求等。

WaitGroup基本用法

WaitGroup 通过计数器控制等待逻辑：调用 Add(n) 增加等待数量，每个goroutine执行完后调用 Done() 减一，主协程通过 Wait() 阻塞直到计数器归零。

典型使用结构如下：

var wg sync.WaitGroup

for i := 0; i < 5; i++ {
    wg.Add(1)
    go func(id int) {
        defer wg.Done()
        // 模拟任务
        fmt.Printf("Goroutine %d 正在执行\n", id)
    }(i)
}

wg.Wait() // 等待所有goroutine完成
fmt.Println("全部完成")

关键点：

• Add 必须在 go语句之前 调用，否则可能产生竞态条件
• Done 通常用 defer 确保一定会执行
• Wait 放在主协程最后调用，阻塞直到所有任务结束

常见错误与注意事项

使用 WaitGroup 时容易出错的地方包括：

• Add 调用时机错误：如果在 goroutine 内部才 Add，可能导致主协程 Wait 提前返回
• 重复 Done：一个 goroutine 多次 Done 会导致计数器负数，panic
• 忘记 Done：某个 goroutine 没调用 Done，导致主协程永远阻塞

正确做法是：在启动 goroutine 前就确定任务数量，并提前 Add。

实际应用场景示例

假设我们要并发获取多个API接口数据：

urls := []string{"http://a.com", "http://b.com", "http://c.com"}
results := make([]string, len(urls))
var wg sync.WaitGroup

for i, url := range urls {
    wg.Add(1)
    go func(i int, u string) {
        defer wg.Done()
        resp, err := http.Get(u)
        if err != nil {
            results[i] = "error"
            return
        }
        results[i] = resp.Status
    }(i, url)
}

wg.Wait()
fmt.Println("结果:", results)

这里通过 WaitGroup 确保所有HTTP请求完成后再输出结果，避免了主协程提前退出的问题。

与channel的对比选择

WaitGroup 适用于只需要“等待完成”而不需要返回值的场景。如果需要收集每个goroutine的结果，结合 channel 使用更合适：

• 只关心完成：用 WaitGroup
• 需要返回数据：WaitGroup + channel 或直接用 channel 控制
• 有超时需求：建议用 context 配合 channel 实现

WaitGroup 设计简洁，开销小，是Go并发编程中最基础也最实用的同步工具之一。

理论要掌握，实操不能落！以上关于《Golang协程等待技巧：WaitGroup使用详解》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！