Golang协程等待技巧：sync.WaitGroup用法详解

sync.WaitGroup用于等待多个goroutine完成，通过Add增加计数，Done减少计数，Wait阻塞直至计数为零，确保主协程正确同步子协程执行。

在Golang中，sync.WaitGroup 是一种常用的同步机制，用于等待一组并发的协程（goroutines）执行完成。它特别适合在主协程需要等待多个子协程结束时使用。使用 WaitGroup 可以避免程序提前退出或资源未释放的问题。

WaitGroup 的基本用法

WaitGroup 有三个核心方法：

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

典型使用模式如下：

var wg sync.WaitGroup

for i := 0; i < 3; i++ {

  wg.Add(1) // 每次启动前加1

  go func(id int) {

    defer wg.Done() // 结束时减1

    fmt.Printf("协程 %d 开始工作\n", id)

    time.Sleep(time.Second)

    fmt.Printf("协程 %d 完成\n", id)

  } (i)

}

wg.Wait() // 等待所有协程完成

fmt.Println("所有协程执行完毕")

常见使用场景和注意事项

WaitGroup 适用于批量启动协程并统一等待的场景，比如并发请求、数据处理等。

• 确保每次 Add 都在 go 语句之前调用，防止竞争条件。
• 推荐在协程内部使用 defer wg.Done()，确保即使发生 panic 也能正确计数。
• 不能对已归零的 WaitGroup 执行 Done()，否则会 panic。
• WaitGroup 通常不应被复制，应以指针方式传递给函数。

传递 WaitGroup 到函数中的正确方式

如果需要将 WaitGroup 传入函数，应传递指针：

  defer wg.Done()

  // 执行任务

}

// 调用时

wg.Add(1)

go doWork(&wg)

这样可以避免值拷贝导致的状态不一致问题。

基本上就这些。只要注意 Add 和 Done 的配对，以及调用时机，WaitGroup 就能可靠地帮你管理协程生命周期。