golang中的原子操作在sync/atomic package中。
下文以比较和交换操作函数为例,介绍其使用。
CompareAndSwapInt32
比较和交换操作是原子性的。
// CompareAndSwapInt32 executes the compare-and-swap operation for an int32 value.
func CompareAndSwapInt32(addr *int32, old, new int32) (swapped bool)
判断参数addr指向的值是否与参数old的值相等,
如果相等,用参数new的新值替换掉addr存储的旧值,否则操作就会被忽略。
交换成功,返回true.
example1
package main
import (
"fmt"
"sync/atomic"
)
func main(){
var value int32
fmt.Println("origin value:", value)
swapFlag := atomic.CompareAndSwapInt32(&value, 0, 1)
if swapFlag {
fmt.Println("swap, value:", value)
} else {
fmt.Println("not swap, value:", value)
}
}
上面的代码是简单使用举例。
判断value中的值是否为0,如果是,则将1存储到value的地址中;否则,不做任何操作。
output:
origin value: 0
swap, value: 1
examaple2
下面例子中,有两个goroutine去更新同一地址存储的值,只有一个会操作成功。
package main
import (
"fmt"
"sync/atomic"
"time"
)
func main(){
var value int32
fmt.Println("origin value:", value)
go entry("1", &value)
go entry("2", &value)
time.Sleep(time.Second)
}
func entry(name string, value *int32) {
swapFlag := atomic.CompareAndSwapInt32(value, 0, 1)
if swapFlag {
fmt.Println("goroutine name:",name, ", swap, value:", *value)
} else {
fmt.Println("goroutine name:", name, ", not swap, value:", *value)
}
}
创建两个goroutine,两个goroutine执行相同的流程,同时去更新value。
其中一个会操作成功。
主goroutine等待两个goroutine结束。
output:
origin value: 0
goroutine name: 2 , swap, value: 1
goroutine name: 1 , not swap, value: 1
参考
https://www.kancloud.cn/digest/batu-go/153537
http://ifeve.com/go-concurrency-atomic/
原文地址:https://www.cnblogs.com/lanyangsh/p/10125299.html
时间: 2024-10-07 14:46:29