您的位置：首页 > 编程语言 > Go语言

Go原子计数

2016-09-17 08:25 204 查看

发表回复

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

先看代码：

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

本条目发布于2015年10月6日。属于go语言分类。作者是baiyuxiong。