原子操作即是进行过程中不能被中断的操作。针对某个值的原子操作在被进行的过程中,CPU绝不会再去进行其他的针对该值的操作。
为了实现这样的严谨性,原子操作仅会由一个独立的CPU指令代表和完成。
- GO语言提供的原子操作都是非入侵式的,由标准库sync/atomic中的众多函数代表
- 类型包括int32,int64,uint32,uint64,uintptr,unsafe.Pointer,共六个。
- 这些函数提供的原子操作共有五种:增或减,比较并交换,载入,存储和交换
int各种类型取值范围
类型 | 长度(字节) | 值范围 |
---|---|---|
int8 | 1 | -128 ~ 127 |
uint8(byte) | 1 | 0 ~ 255 |
int16 | 2 | 32768~32767 |
uint16 | 2 | 0~65535 |
int32 | 4 | 2147483648~2147483647 |
uint32 | 4 | 0~4294967295 |
int64 | 8 | -9223372036854775808~9223372036854775807 |
uint64 | 8 | 0~18446744073709551615 |
int | 平台相关 | 平台相关 |
uint | 平台相关 | 平台相关 |
uintptr | 同指针 | 在32位平 下为4字节,64位平 下为8字节 |
增或减Add
函数名称都以Add为前缀,并后跟针对的具体类型的名称。
- 被操作的类型只能是数值类型
- int32,int64,uint32,uint64,uintptr类型可以使用原子增或减操作
- 第一个参数值必须是一个指针类型的值,以便施加特殊的CPU指令
- 第二个参数值的类型和第一个被操作值的类型总是相同的。
示例
package main
import (
"fmt"
"sync/atomic"
)
func main(){
var i32 int32
fmt.Println("=====old i32 value=====")
fmt.Println(i32)
//第一个参数值必须是一个指针类型的值,因为该函数需要获得被操作值在内存中的存放位置,以便施加特殊的CPU指令
//结束时会返回原子操作后的新值
newI32 := atomic.AddInt32(&i32,3)
fmt.Println("=====new i32 value=====")
fmt.Println(i32)
fmt.Println(newI32)
var i64 int64
fmt.Println("=====old i64 value=====")
fmt.Println(i64)
newI64 := atomic.AddInt64(&i64,-3)
fmt.Println("=====new i64 value=====")
fmt.Println(i64)
fmt.Println(newI64)
}
结果:
/usr/local/go/bin/go run /Users/liuxinming/go/src/free/learngo/atomic/add/add.go
=====old i32 value=====
0
=====new i32 value=====
-3
-3
=====old i64 value=====
0
=====new i64 value=====
-3
-3
比较并交换CAS
Compare And Swap 简称CAS,在sync/atomic包种,这类原子操作由名称以‘CompareAndSwap’为前缀的若干个函数代表。
- 声明如下
func CompareAndSwapInt32(addr *int32, old, new int32) (swapped bool)
- 调用函数后,会先判断参数addr指向的被操作值与参数old的值是否相等
- 仅当此判断得到肯定的结果之后,才会用参数new代表的新值替换掉原先的旧值,否则操作就会被忽略。
so, 需要用for循环不断进行尝试,直到成功为止
使用锁的做法趋于悲观
- 我们总假设会有并发的操作要修改被操作的值,并使用锁将相关操作放入临界区中加以保护
- 使用CAS操作的做法趋于乐观
- 总是假设被操作值未曾被改变(即与旧值相等),并一旦确认这个假设的真实性就立即进行值替换。
示例
package main
import (
"fmt"
"sync/atomic"
)
var value int32
func main() {
fmt.Println("======old value=======")
fmt.Println(value)
fmt.Println("======CAS value=======")
addValue(3)
fmt.Println(value)
}
//不断地尝试原子地更新value的值,直到操作成功为止
func addValue(delta int32){
//在被操作值被频繁变更的情况下,CAS操作并不那么容易成功
//so 不得不利用for循环以进行多次尝试
for {
v := value
if atomic.CompareAndSwapInt32(&value, v, (v + delta)){
//在函数的结果值为true时,退出循环
break
}
//操作失败的缘由总会是value的旧值已不与v的值相等了.
//CAS操作虽然不会让某个Goroutine阻塞在某条语句上,但是仍可能会使流产的执行暂时停一下,不过时间大都极其短暂.
}
}
结果:
======old value=======
0
======CAS value=======
3
载入Load
上面的比较并交换案例总 v:= value为变量v赋值,但… 要注意,在进行读取value的操作的过程中,其他对此值的读写操作是可以被同时进行的,那么这个读操作很可能会读取到一个只被修改了一半的数据.
- so so so , 我们要使用sync/atomic代码包同样为我们提供了一系列的函数,以Load为前缀(载入),来确保这样的糟糕事情发生。
示例
package main
import (
"fmt"
"sync/atomic"
)
var value int32
func main() {
fmt.Println("======old value=======")
fmt.Println(value)
fmt.Println("======CAS value=======")
addValue(3)
fmt.Println(value)
}
//不断地尝试原子地更新value的值,直到操作成功为止
func addValue(delta int32){
//在被操作值被频繁变更的情况下,CAS操作并不那么容易成功
//so 不得不利用for循环以进行多次尝试
for {
//v := value
//在进行读取value的操作的过程中,其他对此值的读写操作是可以被同时进行的,那么这个读操作很可能会读取到一个只被修改了一半的数据.
//因此我们要使用载入
v := atomic.LoadInt32(&value)
if atomic.CompareAndSwapInt32(&value, v, (v + delta)){
//在函数的结果值为true时,退出循环
break
}
//操作失败的缘由总会是value的旧值已不与v的值相等了.
//CAS操作虽然不会让某个Goroutine阻塞在某条语句上,但是仍可能会使流产的执行暂时停一下,不过时间大都极其短暂.
}
}
- atomic.LoadInt32接受一个*int32类型的指针值
- 返回该指针指向的那个值
存储Store
与读取操作相对应的是写入操作。 而sync/atomic包也提供了与原子的载入函数相对应的原子的值存储函数。 以Store为前缀
- 在原子地存储某个值的过程中,任何CPU都不会进行针对同一个值的读或写操作。
- 原子的值存储操作总会成功,因为它并不会关心被操作值的旧值是什么
- 和CAS操作有着明显的区别
fmt.Println("======Store value=======")
atomic.StoreInt32(&value, 10)
fmt.Println(value)
交换Swap
- 与CAS操作不同,原子交换操作不会关心被操作的旧值。
- 它会直接设置新值
- 它会返回被操作值的旧值
- 此类操作比CAS操作的约束更少,同时又比原子载入操作的功能更强
实际案例(继续改造上一版代码)
//数据文件的实现类型
type myDataFile struct {
f *os.File //文件
fmutex sync.RWMutex //被用于文件的读写锁
rcond *sync.Cond //读操作需要用到的条件变量
woffset int64 // 写操作需要用到的偏移量
roffset int64 // 读操作需要用到的偏移量
dataLen uint32 //数据块长度
}
//此处省略...
func (df *myDataFile) Read() (rsn int64, d Data, err error){
// 读取并更新读偏移量
var offset int64
for {
offset = atomic.LoadInt64(&df.roffset)
if atomic.CompareAndSwapInt64(&df.roffset, offset, (offset + int64(df.dataLen))){
break
}
}
//读取一个数据块,最后读取的数据块序列号
rsn = offset / int64(df.dataLen)
bytes := make([]byte, df.dataLen)
//读写锁:读锁定
df.fmutex.RLock()
defer df.fmutex.RUnlock()
for {
_, err = df.f.ReadAt(bytes, offset)
if err != nil {
if err == io.EOF {
//暂时放弃fmutex的 读锁,并等待通知的到来
df.rcond.Wait()
continue
}
}
break
}
d = bytes
return
}
func (df *myDataFile) Write(d Data) (wsn int64, err error){
//读取并更新写的偏移量
var offset int64
for {
offset = atomic.LoadInt64(&df.woffset)
if atomic.CompareAndSwapInt64(&df.woffset, offset, (offset + int64(df.dataLen))){
break
}
}
//写入一个数据块,最后写入数据块的序号
wsn = offset / int64(df.dataLen)
var bytes []byte
if len(d) > int(df.dataLen){
bytes = d[0:df.dataLen]
}else{
bytes = d
}
df.fmutex.Lock()
defer df.fmutex.Unlock()
_, err = df.f.Write(bytes)
//发送通知
df.rcond.Signal()
return
}
func (df *myDataFile) Rsn() int64{
offset := atomic.LoadInt64(&df.roffset)
return offset / int64(df.dataLen)
}
func (df *myDataFile) Wsn() int64{
offset := atomic.LoadInt64(&df.woffset)
return offset / int64(df.dataLen)
}
完整代码放在GITHUB:
https://github.com/lxmgo/learngo
一些学习过程整理,希望对大家学习Go语言有所帮助。