在哪里应该调用wg.Add()

哈喽！今天心血来潮给大家带来了《在哪里应该调用wg.Add()》，想必大家应该对Golang都不陌生吧，那么阅读本文就都不会很困难，以下内容主要涉及到，若是你正在学习Golang，千万别错过这篇文章~希望能帮助到你！

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)

这是我从视频中看到的一段代码https://subscription.packtpub.com/video/application_development/9781788994880/97598/97608/goroutines

但是 v 总是小于 100，我认为原因可能是 wg.wait() 会比预期更早结束，因为我们将 wg.add(1) 放在匿名函数内和同一个 goroutine wg 中.done() 将立即被调用，从而主协程从阻塞状态恢复执行。

但是如果我们将 wg.add(1) 放入 for 循环中，v 将始终为 100。

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)

我的问题是为什么我们可以保证 main goroutine 总是阻塞在这里，使得 v 最终等于 100。如果在 for 循环之前向 wg 添加一个任务，并且主 goroutine 在这里恢复执行，因为此时没有任务，是否可以实现这一点。

解决方案

在启动将调用 wg.Done() 的 Goroutine 之前，您应该始终调用 wg.Add()。

在您更正的示例中，main goroutine只能在for循环之后到达wg.Wait()，这保证您调用wg.Add()一百次，因此wg.Wait()将阻塞直到zqbc zqbwg。 Done() 被调用 100 次。

当 wg.Add() 调用在新的 goroutine 中时，不能保证任何 wg.Add() 调用将在 main goroutine 到达 wg.Wait() 之前执行，因为它们同时运行（到目前为止还没有同步）。这种情况下的行为是不确定的（取决于 goroutine 调度程序，它在没有显式同步的情况下是不确定的）。

请注意，如果您知道循环进行 100 迭代，另一种替代方法是在循环之前调用 wg.Add(100)。不过，我建议不要这种做法，因为当循环包含 break 或 continue 操作时需要小心，这可能会导致启动更少的 goroutine，从而最终导致 main goroutine 卡住。是的，在您的情况下，这可能是微不足道的，但如果此代码及时演变，它可能会变得不那么明显，并可能导致未来的错误。当场景中涉及启动 goroutine 时，说它更快是无关紧要的。如果你只在启动 goroutine 之前调用 wg.Add(1) ，那么稍后你有条件地跳过这一部分也没关系，因为你将在启动 goroutine 的同时跳过 wg.Add() ，并且你的代码将保持正确。

使用 sync.WaitGroup 时要遵循的简单“规则”：（引自 this answer）

• 在“原始”goroutine（启动一个新的）中在 go 语句之前调用 WaitGroup.Add()
• 建议延迟调用 WaitGroup.Done()，这样即使 Goroutine 发生恐慌，它也会被调用
• 如果要将 WaitGroup 传递给其他函数（并且不使用包级变量），则必须传递指向它的指针，否则 WaitGroup （这是一个结构体）将被复制，并且 Done() 方法副本上的调用不会在原件上观察到

今天关于《在哪里应该调用wg.Add()》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！