Go by Example 中文:原子计数器

Go 中最主要的状态管理方式是通过通道间的沟通来完成的,我们在工作池的例子中碰到过,但是还是有一些其他的方法来管理状态的。这里我们将看看如何使用 sync/atomic包在多个 Go 协程中进行 原子计数

package main
import "fmt"
import "time"
import "sync/atomic"
import "runtime"
func main() {

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

    var ops uint64 = 0

为了模拟并发更新,我们启动 50 个 Go 协程,对计数器每隔 1ms (译者注:应为非准确时间)进行一次加一操作。

    for i := 0; i < 50; i++ {
        go func() {
            for {

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

                atomic.AddUint64(&ops, 1)

允许其它 Go 协程的执行

                runtime.Gosched()
            }
        }()
    }

等待一秒,让 ops 的自加操作执行一会。

    time.Sleep(time.Second)

为了在计数器还在被其它 Go 协程更新时,安全的使用它,我们通过 LoadUint64 将当前值的拷贝提取到 opsFinal中。和上面一样,我们需要给这个函数所取值的内存地址 &ops

    opsFinal := atomic.LoadUint64(&ops)
    fmt.Println("ops:", opsFinal)
}

执行这个程序,显示我们执行了大约 40,000 次操作

$ go run atomic-counters.go
ops: 40200

下面,我们将看一下互斥锁——管理状态的另一个工具

下一个例子: 互斥锁.