通过原子计数可以在多线程情况下,对同一个数值进行加减操作,一般用于状态同步。
先看代码:
1
2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
package main
import "fmt" import "time" import "sync/atomic" import "runtime" func main() { // 定义一个整数 var ops uint64 = 0 // 使用50个线程给ops累加数值 for i := 0; i < 50; i++ { go func() { for { // 每次加1 atomic.AddUint64(&ops, 1) // 这个函数用于时间片切换 //可以理解为高级版的time.Sleep() //避免前面的for循环将CPU时间片都卡在一个线程里,使得其它线程没有执行机会 runtime.Gosched() } }() } //停一秒,上面50个线程有1秒的执行时间 time.Sleep(time.Second) // 获取结果 opsFinal := atomic.LoadUint64(&ops) fmt.Println("ops:", opsFinal) } |
打印结果类似:
ops: 40200
如果不使用原子计数,直接使用ops =ops+1会导致多线程时计数不准确。
打开Go源码中的atomic包,可以看到相关算法都是用汇编语言写的。所以原子计数执行效率非常高。