通过原子计数可以在多线程情况下,对同一个数值进行加减操作,一般用于状态同步。
先看代码:
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包,可以看到相关算法都是用汇编语言写的。所以原子计数执行效率非常高。