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 原子操作函數