当前位置：首页 > 文章列表 > Golang > Go教程 > Golang协程等待技巧：WaitGroup使用详解

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

2025-12-01 18:41:29 0浏览收藏

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

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

Golang如何使用sync.WaitGroup等待goroutine完成_Golang WaitGroup使用实践

在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请求完成后再输出结果，避免了主协程提前退出的问题。