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) }
Result:
$ go run example.go
ops: 6312891
执行这个程序,显示我们执行了大约 6312891 次操作