带有sync.waitGroup的Goroutine每次输出不同的值

Question

下面的代码每次执行后都会打印不同的值，但我希望值相同，如何在不使用time.Sleep的情况下更改下面的代码

package main

import (
    "fmt"
    "sync"
)

var total int
var wg sync.WaitGroup

// Inc increments the counter for the given key.
func inc(num int) {
    total += num
    wg.Done()
}

// Value returns the current value of the counter for the given key.
func getValue() int {
    return total
}

func main() {
    for i := 1; i <= 1000; i++ {
        wg.Add(1)
        go inc(i)
    }
    wg.Wait()
    fmt.Println(getValue())
}

Answer 1

你有一个数据竞赛，结果是不确定的。您必须同步对共享变量的访问：

var total int
var lock sync.Mutex
var wg sync.WaitGroup

// Inc increments the counter for the given key.
func inc(num int) {
    lock.Lock()
    defer lock.Unlock()
    total += num
    wg.Done()
}

// Value returns the current value of the counter for the given key.
func getValue() int {
    lock.Lock()
    defer lock.Unlock()
    return total
}

或者，使用sync/atomic 访问/修改变量。

Answer 2

你每次得到不同值的原因是total += num中的竞争条件。

一个简单的解决方法是添加一个互斥锁： var mu sync.Mutex

并在inc 中使用它：

func inc(num int) {
    mu.Lock()
    defer mu.Unlock()
    total += num
    wg.Done()
}

Answer 3

已经提到您有“数据竞争”，使用Mutex 是一种解决方案。或者你可以使用atomic 包，这样更快。

package main

import (
    "fmt"
    "sync"
    "sync/atomic"
)

var total uint64
var wg sync.WaitGroup

func inc(num uint64) {
    atomic.AddUint64(&total, 1)
    wg.Done()
}

// Value returns the current value of the counter for the given key.
func getValue() uint64 {
    return atomic.LoadUint64(&total)
}

func main() {
    for i := uint64(1); i <= 1000; i++ {
        wg.Add(1)
        go inc(i)
    }
    wg.Wait()
    fmt.Println(getValue())
}