首页  >  文章  >  后端开发  >  为了在 Go 中正确同步 Goroutine,应该将 `wg.Add()` 和 `wg.Done()` 放在哪里?

为了在 Go 中正确同步 Goroutine,应该将 `wg.Add()` 和 `wg.Done()` 放在哪里?

Susan Sarandon
Susan Sarandon原创
2024-11-04 05:28:29787浏览

 Where Should You Place `wg.Add()` and `wg.Done()` for Correct Goroutine Synchronization in Go?

用于 Goroutine 同步的 WaitGroup 插入位置

在 Go 中,sync.WaitGroup 类型通过跟踪活动的数量来促进 Goroutine 之间的同步goroutine 并等待所有 goroutine 完成执行后再继续。正确放置 wg.Add() 和 wg.Done() 调用对于确保正确的行为至关重要。

考虑以下代码:

<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>

在此示例中,wg.Add( 1) 被错误地放置在匿名函数中。这种不正确的放置可能会导致 v 值不一致,通常小于 100。

这种差异的原因在于 goroutine 的并发性质。当放置在 goroutine 中时,wg.Add(1) 可能会与 wg.Done() 同时运行,导致 wg.Wait() 过早恢复并低估已完成的 goroutine 数量。

要解决此问题,在启动每个 goroutine 之前,应将 wg.Add(1) 放置在 for 循环中:

<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.Wait() 将阻塞,直到所有 wg.Done() 调用完成,从而使 v 能够准确反映已完成的 Goroutine 的数量。

总之,始终在启动 Goroutine 之前放置 wg.Add()一旦 goroutine 完成执行,就会执行 wg.Done() 以确保正确的同步和可靠的结果计算。

以上是为了在 Go 中正确同步 Goroutine,应该将 `wg.Add()` 和 `wg.Done()` 放在哪里?的详细内容。更多信息请关注PHP中文网其他相关文章!

声明:
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn