go语言从例子开始之Example35.原子计数器

OStack程序员社区-中国程序员成长平台 › 门户 › 编程› Go›Go教程

原作者: [db:作者] 来自: [db:来源] 收藏邀请

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

Example:

package main

import "fmt"
import "time"
import "sync/atomic"
import "runtime"

func main() {
    //我们将使用一个无符号整型数来表示（永远是正整数）这个计数器。
    var ops uint64 = 0
    
    for i:=1; 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)
}