Go语言sync.WaitGroup：避免并发等待问题的正确用法
在Go语言并发编程中，sync.WaitGroup是用于等待一组goroutine完成的同步机制。本文将分析一个常见的sync.WaitGroup使用方法错误，并提供正确的解决方案。
问题：非原子操作导致等待失败
以下代码片段演示了一个错误的sync.WaitGroup使用案例，其结果与预期不符：
立即学习“go语言免费学习笔记（深入）”；
func wg() {
wg := &sync.WaitGroup{}
go wgcore(1, wg)
go wgcore(2, wg)
go wgcore(3, wg)
wg.Wait()
}
func wgcore(i int64, wg *sync.WaitGroup) {
wg.Add(1)
defer wg.Done()
fmt.Println(i)
}
错误原因在于wg.Add(1)调用并非与goroutine启动原子操作。wg.Wait()可能在所有wg.Add(1)调用完成之前就执行，导致程序过早结束。
解决方案：预先设置计数器
正确的使用方法是在启动goroutine之前，预先设置WaitGroup的计数器：
func Wg() {
wg := &sync.WaitGroup{}
wg.Add(3)
go wgCore(1, wg)
go wgCore(2, wg)
go wgCore(3, wg)
wg.Wait()
}
func wgCore(i int64, wg *sync.WaitGroup) {
defer wg.Done()
fmt.Println(i)
}
在这个修正后的版本中，wg.Add(3)在启动任何goroutine之前设置计数器为3。这样，wg.Wait()就能准确等待所有三个goroutine完成。
关键：原子性与计数器初始化
sync.WaitGroup计数器的正确初始化至关重要。 由于goroutine启动顺序的不确定性，在每个goroutine内部增加计数会导致wg.Wait()无法准确判断所有goroutine是否完成。因此，务必在启动所有goroutine 之前，根据goroutine的数量预先设置WaitGroup的计数器，确保wg.Wait()能够准确地等待所有goroutine执行完毕。
