-2

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

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

令我惊讶的是，我panic: Fail in goroutine after TestReadWrite has completed在运行“go test”时得到了。当使用“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. 到目前为止，我看到的很多代码都wg.Add(1)在 goroutine 中。为什么在这种特定情况下会出现意外行为？这里似乎发生的是，一些 goroutine 似乎已经完成运行，并且在其他 goroutine 开始运行之前就通过了 wg.Wait()。在 goroutine 中使用 wg.Add(1) 是否危险/要避免？如果这通常不是问题，那么究竟是什么导致了这里的问题？
  2. 是否添加wg.Add(expected)了解决此问题的正确方法？
4

2 回答 2

2

您的第一种方法会导致恐慌（WaitGroup.Add）：

将可能为负的增量添加到 WaitGroup 计数器。如果计数器变为零，则所有在 Wait 上阻塞的 goroutine 都会被释放。如果计数器变为负数，请添加恐慌。

...

...

通常这意味着对 Add 的调用应该在创建 goroutine 的语句或其他要等待的事件之前执行

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

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

于 2021-07-22T05:28:21.623 回答
1

根据文档-

WaitGroup 等待一组 goroutine 完成。主 goroutine 调用 Add 来设置要等待的 goroutine 的数量。然后每个 goroutine 运行并在完成时调用 Done。同时，Wait 可以用来阻塞，直到所有的 goroutine 都完成。

所以Add()必须由一个启动其他goroutine的goroutine调用，在你的情况下是maingoroutine。

在第一个代码片段中，您Add()在其他 goroutine 内部调用而不是导致问题的主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()

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

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) 是解决此问题的正确方法吗？

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

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()
于 2021-07-22T05:03:27.890 回答