Go by Example 中文版: 原子计数器

Go 中最主要的状态管理机制是依靠通道间的通信来完成的。 我们已经在工作池的例子中看到过，并且，还有一些其他的方法来管理状态。 这里，我们来看看如何使用 sync/atomic 包在多个协程中进行 _原子计数_。

package main

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

func main() {

我们将使用一个无符号整型（永远是正整数）变量来表示这个计数器。

    var ops uint64

WaitGroup 帮助我们等待所有协程完成它们的工作。

    var wg sync.WaitGroup

我们会启动 50 个协程，并且每个协程会将计数器递增 1000 次。

    for i := 0; i < 50; i++ {
        wg.Add(1)

使用 AddUint64 来让计数器自动增加， 使用 & 语法给定 ops 的内存地址。

        go func() {
            for c := 0; c < 1000; c++ {

                atomic.AddUint64(&ops, 1)
            }
            wg.Done()
        }()
    }

等待，直到所有协程完成。

    wg.Wait()

现在可以安全的访问 ops，因为我们知道，此时没有协程写入 ops， 此外，还可以使用 atomic.LoadUint64 之类的函数，在原子更新的同时安全地读取它们。

    fmt.Println("ops:", ops)
}

预计会进行 50,000 次操作。如果我们使用非原子的 ops++ 来增加计数器， 由于多个协程会互相干扰，运行时值会改变，可能会导致我们得到一个不同的数字。 此外，运行程序时带上 -race 标志，我们可以获取数据竞争失败的详情。

$ go run atomic-counters.go
ops: 50000

接下来，我们看一下管理状态的另一个工具——互斥锁。

下一个例子: 互斥锁

@mmcgrana 和Eli Bendersky编写 | gobyexample-cn 翻译 | 反馈 | 源码 | license