Go语言之城 - 程序示例: 原子计数器

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

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

下一个例子: 互斥锁

Go语言之城 | 闽ICP备20003902号-3