golang 原子操作函数

Posted lanyangsh

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了golang 原子操作函数相关的知识,希望对你有一定的参考价值。

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/








以上是关于golang 原子操作函数的主要内容,如果未能解决你的问题,请参考以下文章

Golang✔️走进 Go 语言✔️ 第二十课 原子操作 & 互斥体

Golang✔️走进 Go 语言✔️ 第二十课 原子操作 & 互斥体

[Go] golang原子函数实现goroutine同步

原子操作--sync/atomic的用法

带有 mgo 的 Go (golang) 中的 MongoDB:如何更新记录、确定更新是不是成功并在单个原子操作中获取数据?

golang goroutine例子[golang并发代码片段]