本文深入探讨了在 go 语言中，当启动多个 goroutine 并行处理任务时，如何优雅且高效地等待所有 goroutine 完成其工作。我们将重点介绍并演示 `sync.waitgroup` 这一 标准库 提供的机制，它是实现此类 并发 同步的惯用且推荐方式，相比于手动管理通道，`waitgroup` 提供了更简洁、健壮的解决方案。
在 Go 语言 的并发编程 中，我们经常会遇到需要并行处理大量数据或执行耗时操作的场景。通过启动多个 goroutine 来并发执行这些任务，可以显著提高程序的性能。然而，一个常见的挑战是，主 goroutine 需要在所有子 goroutine 完成其工作后才能继续执行后续逻辑，例如汇总结果或释放资源。如果主 goroutine 不等候，它可能会提前退出，导致部分并发任务未能完成，或者后续操作依赖于未就绪的数据。
一种直观但并非最惯用的同步方式是使用通道（channel）。例如，为每个 goroutine 分配一个写入通道的权限，并在它们完成时向通道发送一个信号，主 goroutine 则通过 循环 接收这些信号来判断所有任务是否完成。虽然这种方法可行，但对于简单的“等待所有任务完成”的场景，它显得有些冗余和复杂。Go标准库 提供了更为简洁和高效的 sync.WaitGroup 类型来专门处理这类同步需求。
解决方案：使用 sync.WaitGroup
sync.WaitGroup 是 Go 语言中专门用于等待一组 goroutine 完成的同步原语。它维护一个内部计数器，通过 Add 方法增加计数，Done 方法减少计数，以及 Wait 方法阻塞直到计数器归零。
sync.WaitGroup 工作原理详解
sync.WaitGroup 提供了三个核心方法：
立即学习“go 语言免费学习笔记（深入）”；
- Add(delta int): 用于增加 WaitGroup 的内部计数器。通常在启动新的 goroutine 之前调用，delta 参数表示要增加的计数。例如，wg.Add(1) 表示增加一个需要等待的 goroutine。
- Done(): 用于减少 WaitGroup 的内部计数器。通常在每个 goroutine 完成其工作时调用，表示该 goroutine 已完成。
- Wait(): 阻塞当前 goroutine，直到 WaitGroup 的内部计数器归零。这意味着所有通过 Add 方法增加的 goroutine 都已通过 Done 方法完成。
实践示例
假设我们有一个 Huge 函数，需要对一个 切片 lst 中的每个 foo 元素执行一个耗时操作 performSlow。我们希望并发地执行 performSlow，并在所有操作完成后，再调用 someValue 函数。
package main import ("fmt" "sync" "time") // 假设这是一个耗时操作 func performSlow(item string) {fmt.Printf(" 开始处理： %sn", item) time.Sleep(time.Second) // 模拟耗时操作 fmt.Printf(" 完成处理： %sn", item) } // 定义一个示例类型 type foo string func Huge(lst []foo) string {var wg sync.WaitGroup // 声明一个 WaitGroup 变量 for _, item := range lst {wg.Add(1) // 每启动一个 goroutine，计数器加 1 // 使用匿名函数包裹 performSlow，以便在 goroutine 内部调用 wg.Done() go func(data foo) {defer wg.Done() // 确保无论如何，goroutine 结束时都会调用 Done() performSlow(string(data)) }(item) // 将 item 作为参数传递给匿名函数，避免 闭包 陷阱 } wg.Wait() // 阻塞，直到所有 goroutine 都调用了 Done()，计数器归零 fmt.Println(" 所有并发任务已完成。") return someValue(lst) // 所有任务完成后，执行后续逻辑 } // 假设这是需要所有并发任务完成后才能调用的函数 func someValue(lst []foo) string {return fmt.Sprintf(" 基于 %v 的最终结果。", lst) } func main() { items := []foo{"A", "B", "C", "D"} result := Huge(items) fmt.Println(result) }代码解释：
- var wg sync.WaitGroup: 在 Huge 函数内部声明一个 WaitGroup 实例。
- wg.Add(1): 在 for 循环中，每次启动一个新的 goroutine 之前，我们都会调用 wg.Add(1)。这会增加 WaitGroup 的内部计数器，表示有一个新的任务需要等待完成。
- go func(data foo) {…}(item): 我们为每个切片元素启动一个独立的 goroutine。注意这里使用了匿名函数并传入 item 作为参数 data，这是为了避免 Go 闭包在循环中捕获循环变量的常见陷阱。
- defer wg.Done(): 这是 WaitGroup 的最佳实践。在每个 goroutine 内部，我们使用 defer 关键字确保 wg.Done() 在 performSlow 函数执行完毕（无论成功还是发生 panic）后被调用。这会将 WaitGroup 的计数器减 1。
- wg.Wait(): 在 for 循环结束后，主 goroutine 调用 wg.Wait()。这个调用会阻塞主 goroutine，直到 WaitGroup 的内部计数器变为零。一旦计数器归零，Wait() 方法返回，主 goroutine 才能继续执行，确保了 someValue(lst) 在所有并发操作完成后才被调用。
为何 sync.WaitGroup 是更优选择
- 简洁性与可读性: 相比于手动创建和管理通道来同步，WaitGroup 的 API 更加简洁明了，直接表达了“等待一组任务完成”的意图。
- 效率: WaitGroup 专为这种场景设计，其内部实现通常比基于通道的通用通信机制更高效，因为它避免了不必要的通道操作和上下文切换。
- 惯用: sync.WaitGroup 是 Go 语言社区公认的、用于此类同步问题的标准和惯用方式。
使用注意事项与最佳实践
- Add 必须在 go func() 之前调用: 确保 WaitGroup 的计数器在 goroutine 开始执行之前就已经增加。如果在 goroutine 内部调用 Add，可能会出现竞态条件，导致 Wait() 在所有 Add 调用完成之前就返回。
- 使用 defer wg.Done(): 强烈建议在 goroutine 的开头使用 defer wg.Done()。这确保了即使 goroutine 因错误或 panic 而提前退出，Done() 也会被调用，避免 WaitGroup 永远无法归零，导致 Wait() 永久阻塞。
- 避免闭包陷阱: 在循环中启动 goroutine 时，如果 goroutine 内部需要使用循环变量，应将其作为参数传递给匿名函数，以避免所有 goroutine 最终都引用同一个（循环结束时的）变量值。如示例中的 go func(data foo) {…}(item)。
- WaitGroup 的生命周期 : WaitGroup 实例通常在需要同步的函数内部创建和使用。如果需要在多个函数之间共享 WaitGroup，应通过 指针 传递 *sync.WaitGroup。
总结
sync.WaitGroup 是 Go 语言中实现并发同步的强大 工具 ，特别适用于“等待所有并发任务完成”的场景。通过合理地使用 Add、Done 和 Wait 方法，开发者可以构建出高效、健壮且易于理解的并发程序。掌握 sync.WaitGroup 是 Go 语言 并发编程 中不可或缺的一项技能，它能帮助我们编写出更符合 Go 语言哲学的高质量代码。
