Golang如何使用sync.WaitGroup等待goroutine完成_Golang WaitGroup使用实践-Golang-PHP中文网

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

P粉602998670

发布： 2025-12-22 18:40:56

原创

818人浏览过

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("全部完成")

登录后复制

关键点：

立即学习“go语言免费学习笔记（深入）”；

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

常见错误与注意事项

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

VALL-E

VALL-E是一种用于文本到语音生成 (TTS) 的语言建模方法

142

查看详情

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