Go原子计数

通过原子计数可以在多线程情况下,对同一个数值进行加减操作,一般用于状态同步。

先看代码:

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包,可以看到相关算法都是用汇编语言写的。所以原子计数执行效率非常高。

posted @ 2015-03-04 10:56  klain  阅读(4857)  评论(0编辑  收藏  举报