Go语言WaitGroup详解：轻松避免并发编程死锁与等待难题

Go语言并发编程中，`sync.WaitGroup`常用于等待一组goroutine完成，但错误的使用方法可能导致等待失败。本文分析了`sync.WaitGroup`的常见错误：非原子操作导致`wg.Add(1)`与goroutine启动并非原子操作，从而使`wg.Wait()`过早结束。正确的使用方法是在启动goroutine之前，预先使用`wg.Add()`设置计数器，确保`wg.Wait()`准确等待所有goroutine完成，避免并发等待问题。文章将详细讲解错误案例及解决方案，帮助开发者正确使用`sync.WaitGroup`，提升Go并发编程效率。

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

终于介绍完啦！小伙伴们，这篇关于《Go语言WaitGroup详解：轻松避免并发编程死锁与等待难题》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！