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

【问题标题】：How do you make sure goroutines finish in a for-loop using WaitGroup?你如何使用 WaitGroup 确保 goroutine 在 for 循环中完成？
【发布时间】：2017-12-30 19:34:53
【问题描述】：

每次 for 循环迭代时，我都会在 goroutine 中运行一个函数，并且我使用 sync.WaitGroup 来确保 goroutines 全部完成。但是，我在用计数器测试并发性时出现了奇怪的行为。在下面的示例中，我尝试使用 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++
}

【问题讨论】：

等待组按预期工作。问题是 w、y 和 z 存在数据竞争。
@CeriseLimón 我需要使用互斥锁来增加吗？实现这一目标的最有效方法是什么？
使用互斥锁或sync/atomic。如果没有看到您正在解决的实际问题，就很难对有效的解决方案发表评论。

标签： go concurrency synchronization goroutine

【解决方案1】：

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

你有两个选择。

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

【讨论】：