4.8 原子操作 #
本节源码位置 https://github.com/golang-minibear2333/golang/tree/master/4.concurrent/4.8-atomic/
代码中的加锁操作因为涉及内核态的上下文切换会比较耗时、代价比较高。针对基本数据类型我们还可以使用原子操作来保证并发安全,因为原子操作是Go语言提供的方法它在用户态就可以完成,因此性能比加锁操作更好。Go语言中原子操作由内置的标准库sync/atomic
提供。
大多数情况下我们都是针对基本数据类型进行数据操作,能不加锁就不加锁。
首先很多人都不相信基本类型并发修改会出现竞态问题。不妨尝试一下,并发加一。
var wg sync.WaitGroup
for i := 0; i < 10000; i++ {
wg.Add(1)
go func () {
defer wg.Done()
xInt32++
}()
}
wg.Wait()
print(xInt32)
无论输出多少次都无法达到10000
,之所以如此就是因为此处的加1操作并不是原子的,都是先取当前值,加1,再赋值,会出现覆盖的情况。
4.8.1 修改 #
修改是最常用到的。
func modify(delta int32) {
atomic.AddInt32(&xInt32, delta)
atomic.AddInt64(&xInt64, int64(delta))
atomic.AddUint32(&xuInt32, uint32(delta))
atomic.AddUint64(&xuInt64, uint64(delta))
}
我们忽略了Uintptr
的讨论,这是内存地址的整数表示,是用来存地址内容的,暂时没有遇到过指针的数据计算。
var wg sync.WaitGroup
for i := 0; i < 10000; i++ {
wg.Add(1)
go func () {
defer wg.Done()
//xInt32++
modify(1)
}()
}
wg.Wait()
print(xInt32)
改为原子操作后,发现每次运行都可以得到预期的结果10000
,
4.8.2 赋值与读取 #
在并发情况下,读取到某个变量后,在使用时变量内容可能会被篡改,所以使用原子读取。 在并发情况下,为某个变量赋值的时候,必须要防止读取到写入一半的错误值,所以要用原子写入。
var xInt32 int32
atomic.StoreInt32(&xInt32, 100)
println(xInt32)
v := atomic.LoadInt32(&xInt32)
println(v)
输出
100
100
就目前而言,原子读写都是为了防止读写一半导致数据错误,但我无法复现这种错误的场景,假如你可以复现请在本文底部放留言。
var v atomic.Value
v.Store([]int{})
fmt.Println(v.Load().([]int))
也可以存储其他任意类型,但如果使用到类似append
扩容原变量的语句,而不是使用直接替换的话,原子操作也是会失效的。
4.8.3 比较并交换 #
以下是节选自《Go并发编程实战》一书中的例子,比较并交换(Compare And Swap)简称CAS,是乐观锁的核心思想,所以简单介绍一下。
var xInt32 int32
for {
v := atomic.LoadInt32(&xInt32)
if atomic.CompareAndSwapInt32(&xInt32, v, v+100) {
break
}
}
print(xInt32)
- 这里一种无锁的结构,是一种思路,在需要改变数据的时候,反复判断数据是否和原数据一致
- 一致时替换,不一致时说明被它处修改,则跳过
- 在不创建互斥量和不形成临界区的情况下,完成并发安全的值替换操作。
4.8.4 小结 #
- 最常用原子操作中的修改、基本类型的值赋值,其他不常用
- 在其他类型出现并发的时候尽可能使用
sync
包提供的并发安全的类型,下一节讲。 - 通过通信共享内存;不要通过共享内存进行通信。尽量使用通道。