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

在Go语言中，WaitGroup是sync包提供的核心同步工具，专为精准管理多个goroutine的生命周期而设计：通过Add预设任务数量、Done标记单个协程完成、Wait阻塞主协程直至所有任务归零，从而确保主程序不会提前退出、避免竞态与资源泄漏；文中不仅剖析了其内部计数器机制和三大方法的协作逻辑，还通过可运行示例揭示了关键实践要点——如Add必须在goroutine启动前调用、需传递指针而非值类型、Add与Done严格匹配等，让开发者轻松掌握批量并发任务（如数据抓取、日志处理、微服务调用）中的可靠同步方案。

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

WaitGroup 基本原理

WaitGroup 内部维护一个计数器：

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

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

基本使用示例

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

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(2  time.Second)
fmt.Printf("Worker %d done\n", id)
}
func main() {
var wg sync.WaitGroup
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 使用简单，但对协程生命周期管理非常有效。只要注意调用时机和同步方式，就能写出稳定高效的并发代码。

终于介绍完啦！小伙伴们，这篇关于《Golang如何用WaitGroup管理协程生命周期》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！