Question

每次for循环迭代时我都在goroutine中运行一个函数，我正在使用sync.WaitGroup来确保goroutine都完成了。但是，我正在用计数器测试并发性的奇怪行为。在下面的示例中，我尝试使用4种不同的技术（w，x，y，z）跟踪线程数，并获得4种不同的结果。我理解的唯一结果是x，因为它在for循环中递增。我在这里缺少什么？

package main

import "fmt"
import "sync"

var w = 0

func main() {
  x := 0
  y := 0
  z := 0
  var wg sync.WaitGroup
  for i := 0; i < 10000; i++ {
    wg.Add(1)
    x++
    go func() {
      z++
      test(&y)
      wg.Done()
    }()
  }
  wg.Wait()
  fmt.Println(w, x, y, z) // 8947 10000 8831 8816
}

func test(y *int) {
  w++
  *y++
}

Answer 1

sync.Waitgroup按预期工作。 w，y和z不会达到10000，因为多个goroutine正在同时递增它们，并且Go的增量不是并发安全的：它是作为普通的fetch实现的 - 增量重新分配操作。

您有两种选择。

选项1：互斥

type incrementer struct {
    sync.Mutex
    i int
}

func (i *incrementer) Add(n int) {
    i.Lock()
    defer i.Unlock()
    i.i += n
}

并将此类型用于w，y和z。

完整示例：https://play.golang.org/p/6wWUK2xnOCW

选项2：sync.atomic

var w int32 = 0

go func(){
    // in the loop
    atomic.AddInt32(&w, 1)

}()

完整示例：https://play.golang.org/p/oUCGgKYC1-Y

你如何使用WaitGroup确保goroutine在for循环中完成？

1 个答案:

选项1：互斥

选项2：sync.atomic