Go语言sync.WaitGroup：避免并发等待问题的正确用法

在Go语言并发编程中，sync.WaitGroup是用于等待一组goroutine完成的同步机制。本文将分析一个常见的sync.WaitGroup使用方法错误，并提供正确的解决方案。

问题：非原子操作导致等待失败

以下代码片段演示了一个错误的sync.WaitGroup使用案例，其结果与预期不符：

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执行完毕。