并发 Go 程序中应该将 `wg.Add()` 放在哪里？

理解 wg.Add() 的放置

在并发 Go 程序中，sync.WaitGroup 类型充当同步原语来协调任务的完成。使用 WaitGroup 时，放置 wg.Add() 调用对于确保正确的等待行为至关重要。

考虑以下代码：

<code class="go">var wg sync.WaitGroup
var v int32 = 0

for i := 0; i < 100; i++ {
    go func() {
        wg.Add(1) // wrong place
        atomic.AddInt32(&v, 1)
        wg.Done()
    }()
}
wg.Wait()
fmt.Println(v)</code>

此代码旨在并行化一系列任务并同时增加共享变量 v。然而，输出结果通常会小于 100。问题在于匿名 goroutine 中 wg.Add(1) 的放置。

通过将 wg.Add(1) 放置在 goroutine 中，主函数Goroutine 可以在调用 wg.Done() 之前恢复执行。发生这种情况的原因是并发模型允许 wg.Add(1) 调用和 wg.Done() 调用同时执行，可能导致主 Goroutine 中的等待提前终止。

解决此问题问题，wg.Add(1)调用应该移到goroutine之外，如下所示：

<code class="go">var wg sync.WaitGroup
var v int32 = 0

for i := 0; i < 100; i++ {
    wg.Add(1)
    go func() {
        atomic.AddInt32(&v, 1)
        wg.Done()
    }()
}
wg.Wait()
fmt.Println(v)</code>

通过将wg.Add(1)移到goroutine之外，我们确保主goroutine会阻塞在 wg.Wait() 调用上，直到所有 goroutine 完成并调用 wg.Done()。这保证了 v 的最终值是准确的。

作为一般实践，建议在启动将调用 wg.Done() 的 goroutine 之前始终调用 wg.Add()。这可确保正确的等待行为并消除可能出现的任何竞争条件。

以上是并发 Go 程序中应该将 `wg.Add()` 放在哪里？的详细内容。更多信息请关注PHP中文网其他相关文章！