Golang如何使用WaitGroup管理协程生命周期

WaitGroup用于协调多个goroutine的完成，通过Add增加计数、Done减少计数、Wait阻塞等待归零。示例中三个worker并发执行，主协程等待它们完成后再退出。需注意Add在goroutine外调用、传递指针、Add与Done匹配，避免重复Wait。适用于批量任务同步场景。

在Go语言中，WaitGroup 是 sync 包提供的同步原语之一，用于等待一组并发的协程（goroutine）完成任务。它特别适用于主协程需要等待其他多个子协程执行完毕后再继续的场景。

WaitGroup 基本原理

WaitGroup 内部维护一个计数器：

• Add(n)：将计数器加上 n，通常用于增加待处理的协程数量。
• Done()：将计数器减1，一般在协程结束前调用，表示该协程已完成。
• Wait()：阻塞当前协程，直到计数器归零。

通过这三个方法，可以协调主协程和其他协程的生命周期，避免程序提前退出或资源竞争。

基本使用示例

以下是一个简单的例子，展示如何使用 WaitGroup 等待多个协程完成：

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

package main
<p>import (
"fmt"
"sync"
"time"
)</p><p>func worker(id int, wg <em>sync.WaitGroup) {
defer wg.Done() // 任务完成，计数器减1
fmt.Printf("Worker %d starting\n", id)
time.Sleep(2 </em> time.Second)
fmt.Printf("Worker %d done\n", id)
}</p><p>func main() {
var wg sync.WaitGroup</p><pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">for i := 1; i <= 3; i++ {
wg.Add(1)           // 每启动一个协程，计数器加1
go worker(i, &wg)   // 启动协程
}
wg.Wait() // 主协程等待所有协程完成
fmt.Println("All workers finished")

}

输出结果类似：

Worker 1 starting
Worker 2 starting
Worker 3 starting
Worker 1 done
Worker 2 done
Worker 3 done
All workers finished

注意事项和最佳实践

使用 WaitGroup 时需注意以下几点，避免常见错误：

• 确保 Add 在协程启动前调用：如果在协程内部才调用 Add，可能因调度问题导致计数器未及时更新，引发 panic。
• 每个 Add 对应一次 Done：Add 和 Done 的调用次数必须匹配，否则可能死锁或 panic。
• 传递 WaitGroup 指针：多个协程应共享同一个 WaitGroup 实例，因此需传指针，避免值拷贝。
• 避免重复 Wait：Wait 只能安全调用一次。若需多次等待不同组任务，应使用多个 WaitGroup 或重置控制逻辑。

适用场景

WaitGroup 常用于以下情况：

• 批量发起网络请求，并行处理后统一返回。
• 初始化多个服务组件，等待全部准备就绪。
• 数据分片处理，如并行计算、文件读取等。

基本上就这些。WaitGroup 使用简单，但对协程生命周期管理非常有效。只要注意调用时机和同步方式，就能写出稳定高效的并发代码。