Go语言教程：原子计数器

返回Go语言教程首页

概念简介

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

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

课程导航

学习上一篇：Go语言教程:速率限制    学习下一篇：Go语言教程：互斥锁

相关资料

本例程github源代码：https://github.com/xg-wang/gobyexample/tree/master/examples/atomic-counters