如何使用Golang通过WaitGroup同步协程_Golang协程同步机制讲解

sync.WaitGroup 是 Go 中用于等待一组 goroutine 完成的轻量同步工具，核心为 Add、Done、Wait 三步，需注意计数匹配、避免复制和误调用。

Go 语言中，sync.WaitGroup 是最常用、最轻量的协程（goroutine）同步工具之一，用于等待一组协程执行完毕。它不关心协程内部逻辑，只负责“计数”和“阻塞等待”，适合主协程等待多个子协程完成后再继续执行的场景。

WaitGroup 的核心三步用法

WaitGroup 的使用遵循固定模式：声明 → 增加计数 → 启动协程 → 等待完成。漏掉任意一步都可能导致 panic 或死锁。

声明 WaitGroup 变量：通常在主 goroutine 中定义，如 var wg sync.WaitGroup
调用 wg.Add(n)：在启动协程前，把要等待的协程数量加进去（n > 0）。注意不能在协程里调用 Add，否则可能因竞态导致计数错误
协程内调用 wg.Done()：每个协程结束前必须调用一次，等价于 wg.Add(-1)。推荐用 defer wg.Done() 确保执行
主协程调用 wg.Wait()：会阻塞直到内部计数归零。该调用应放在所有 go 语句之后，且只能被调用一次

常见误用与避坑指南

WaitGroup 表面简单，但几个典型错误会让程序崩溃或卡死：

Add 在 goroutine 内调用：因为 Add 和 Wait 之间没有同步保障，可能 Wait 先执行，计数仍为 0，直接返回，后续 Done 导致 panic（计数负值）
Done 调用次数不匹配：比如某个分支没执行 defer Done，或重复调用 Done，都会破坏计数平衡
Wait 被多次调用：Wait 不是重入安全的，第二次调用会 panic
WaitGroup 被复制传递：WaitGroup 不可拷贝（含结构体字段赋值、函数传值、切片 append 等），应始终传指针：func worker(wg *sync.WaitGroup, job int)

一个实用示例：并发请求并汇总结果

模拟向多个 URL 发起 HTTP 请求，并等待全部完成：

func fetchAll(urls []string) { var wg sync.WaitGroup results := make(chan string, len(urls))

for _, url := range urls {
    wg.Add(1)
    go func(u string) {
        defer wg.Done()
        resp, err := http.Get(u)
        if err != nil {
            results <- "error: " + err.Error()
            return
        }
        results <- fmt.Sprintf("success: %s", resp.Status)
    }(url) // 注意：需传参避免闭包引用同一变量
}

go func() {
    wg.Wait()
    close(results)
}()

for res := range results {
    fmt.Println(res)
}