Golang用WaitGroup等待goroutine完成方法

WaitGroup用于等待多个goroutine完成，通过Add增加计数、Done减少计数、Wait阻塞直至计数为零。

在Go语言中，WaitGroup 是 sync 包提供的一个同步原语，用于等待一组并发的 goroutine 执行完成。它特别适合在主协程需要等待多个子协程结束时使用，而不需要关闭通道或处理复杂的信号机制。

基本用法和核心方法

WaitGroup 有三个主要方法：

• Add(delta int)：增加计数器，通常传入要启动的 goroutine 数量。
• Done()：在每个 goroutine 结束时调用，相当于计数器减1。
• Wait()：阻塞当前协程，直到计数器归零。

典型结构是：主协程调用 Add(n) 设置任务数量，每个子协程执行完后调用 Done()，主协程通过 Wait() 阻塞等待全部完成。

简单示例：等待多个goroutine完成

package main

import (
    "fmt"
    "sync"
    "time"
)

func worker(id int, wg *sync.WaitGroup) {
    defer wg.Done() // 任务完成，计数器减1
    fmt.Printf("Worker %d starting\n", id)
    time.Sleep(time.Second) // 模拟工作
    fmt.Printf("Worker %d done\n", id)
}

func main() {
    var wg sync.WaitGroup

    for i := 1; i <= 3; i++ {
        wg.Add(1)           // 增加计数
        go worker(i, &wg)   // 启动goroutine
    }

    wg.Wait() // 等待所有worker完成
    fmt.Println("All workers finished")
}

输出会显示三个 worker 依次开始和结束，最后打印“All workers finished”。

注意事项和常见错误

使用 WaitGroup 时需注意以下几点：

• 避免Add负值：调用 Add(-n) 会导致 panic，除非你明确知道计数器足够大。
• 确保Done被调用：每个 Add(1) 必须对应一次 Done()，否则程序会一直阻塞。
• 传递指针：应将 WaitGroup 的指针传给 goroutine，避免值拷贝导致状态不一致。
• Add应在Wait前调用：最好在 go 语句前调用 Add，防止竞争条件。

适用场景

WaitGroup 适用于以下情况：

• 批量启动多个 goroutine 并等待它们全部完成（如并行请求、数据抓取）。
• 无需从 goroutine 获取返回值，只关心执行完成。
• 作为轻量级同步工具，比通道更简洁。

基本上就这些。只要记住 Add、Done、Wait 的配合使用，就能安全地控制并发流程。