在goroutine开始之前初始化sync.WaitGroup

欢迎各位小伙伴来到golang学习网，相聚于此都是缘哈哈哈！今天我给大家带来《在goroutine开始之前初始化sync.WaitGroup》，这篇文章主要讲到等等知识，如果你对Golang相关的知识非常感兴趣或者正在自学，都可以关注我，我会持续更新相关文章！当然，有什么建议也欢迎在评论留言提出！一起学习！

我有以下代码作为测试的一部分：

expected := 10
    var wg sync.waitgroup
    for i := 0; i < expected; i++ {
        go func(wg *sync.waitgroup) {
            wg.add(1)
            defer wg.done()
            // do something
        }(&wg)
    }
    wg.wait()

令我惊讶的是，在运行“go test”时，我得到了 panic: fail in goroutine after testreadwrite hascompleted。当使用“go test -race”运行时，我没有感到恐慌，但测试后来失败了。在这两种情况下，尽管有 wg.wait()，但 goroutine 并未完成执行。

我进行了以下更改，现在测试按预期工作：

expected := 10
    var wg sync.WaitGroup
    wg.Add(expected)
    for i := 0; i < expected; i++ {
        go func(wg *sync.WaitGroup) {
            defer wg.Done()
            // do something
        }(&wg)
    }
    wg.Wait()

我的疑问是：

1. 到目前为止我看到的很多代码都是在 goroutine 中执行 wg.add(1) 。为什么在这种特定情况下它会表现出意外？这里发生的情况似乎是，一些 goroutine 似乎在其他 goroutine 开始运行之前就完成了运行，并通过了 wg.wait()。在 goroutine 中使用 wg.add(1) 是否危险/需要避免？如果这通常不是问题，那么究竟是什么导致了这里的问题？
2. 添加 wg.add(expected) 是解决此问题的正确方法吗？

正确答案

根据docs -

waitgroup 等待 goroutine 集合完成。主要的 goroutine 调用 add 来设置等待的 goroutine 数量。然后 每个 goroutine 都会运行并在完成后调用 done。同时 time，wait 可以用来阻塞，直到所有 goroutine 都完成。

因此 add() 必须由正在启动其他 goroutine 的 goroutine 调用，在您的情况下是 main goroutine。

在第一个代码片段中，您在其他 goroutine 中调用 add() 而不是导致问题的 main goroutine -

expected := 10
var wg sync.waitgroup
for i := 0; i < expected; i++ {
   go func(wg *sync.waitgroup) {
       wg.add(1) // do not call add() here
       defer wg.done()
       // do something
   }(&wg)
}
wg.wait()

第二个片段正在工作，因为您正在 main goroutine 中调用 add() -

expected := 10
var wg sync.waitgroup
wg.add(expected) // okay
for i := 0; i < expected; i++ {
    go func(wg *sync.waitgroup) {
       defer wg.done()
       // do something
     }(&wg)
}
wg.wait()

添加 wg.add(expected) 是解决此问题的正确方法吗？

您还可以在for循环中调用wg.add(1) -

expected := 10
var wg sync.WaitGroup
for i := 0; i < expected; i++ {
    wg.Add(1) // Okay
    go func(wg *sync.WaitGroup) {
       defer wg.Done()
       // do something
     }(&wg)
}
wg.Wait()

您的第一个方法会出现恐慌，因为 (WaitGroup.Add):

当你的代码末尾调用 wait() 时，没有一个 goroutine 可能还没有开始执行 - 因此 waitgroup 中保存的值是 0。然后当你的 go-routine 执行时，调用 go-routine 已经被执行了。释放。这将导致意外的行为，在你的情况下会出现恐慌。也许您正在使用其中调用 go 例程的值。

你的第二种方法绝对没问题。您还可以在循环内部调用 .add(1) - 但在 go func 块之外

好了，本文到此结束，带大家了解了《在goroutine开始之前初始化sync.WaitGroup》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！