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

**Golang并发控制：使用WaitGroup优雅等待协程完成** 在Golang中，`sync.WaitGroup`是实现协程同步的利器，尤其适用于批量任务处理和并发请求等场景。本文深入解析`WaitGroup`的用法，包括`Add`、`Done`和`Wait`方法，通过增加计数、减少计数和阻塞等待，确保所有协程执行完毕。通过示例代码，详细讲解了如何正确使用`WaitGroup`，避免常见的`Add`调用时机错误、重复`Done`以及忘记`Done`等问题。此外，本文还对比了`WaitGroup`与channel在不同场景下的选择，并提供了并发获取API接口数据的实际应用示例。掌握`WaitGroup`，让你的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学习网公众号，带你了解更多关于的知识点！