Golang协程等待技巧：WaitGroup使用详解

想要高效管理 Golang 协程？`sync.WaitGroup` 是你的得力助手！本文深入解析 `sync.WaitGroup` 的用法，它通过 `Add`、`Done` 和 `Wait` 三个核心方法，实现主协程等待多个子协程完成的功能。`Add` 用于增加计数，`Done` 用于减少计数，而 `Wait` 则会阻塞主协程，直到计数归零。了解 `sync.WaitGroup` 的正确使用方式，包括 `Add` 的调用时机、`defer wg.Done()` 的重要性，以及避免复制 `WaitGroup` 等注意事项，能有效避免程序提前退出和资源未释放，提升 Golang 并发程序的稳定性和可靠性。掌握这些技巧，让你的 Golang 协程管理更上一层楼！

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

  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 就能可靠地帮你管理协程生命周期。

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《Golang协程等待技巧：WaitGroup使用详解》文章吧，也可关注golang学习网公众号了解相关技术文章。