快捷導(dǎo)航

Go語言并發(fā)之原子操作詳解

更新時間：2022年12月29日 09:11:56 作者：機(jī)智的程序員小熊

代碼中的加鎖操作因為涉及內(nèi)核態(tài)的上下文切換會比較耗時、代價比較高。針對基本數(shù)據(jù)類型我們還可以使用原子操作來保證并發(fā)安全，本文就來和大家詳細(xì)聊聊，需要的可以參考下

代碼中的加鎖操作因為涉及內(nèi)核態(tài)的上下文切換會比較耗時、代價比較高。針對基本數(shù)據(jù)類型我們還可以使用原子操作來保證并發(fā)安全，因為原子操作是Go語言提供的方法它在用戶態(tài)就可以完成，因此性能比加鎖操作更好。Go語言中原子操作由內(nèi)置的標(biāo)準(zhǔn)庫sync/atomic 提供。

大多數(shù)情況下我們都是針對基本數(shù)據(jù)類型進(jìn)行數(shù)據(jù)操作，能不加鎖就不加鎖。

首先很多人都不相信基本類型并發(fā)修改會出現(xiàn)競態(tài)問題。不妨嘗試一下，并發(fā)加一。

var wg sync.WaitGroup
for i := 0; i < 10000; i++ {
wg.Add(1)
go func () {
defer wg.Done()
xInt32++
}()
}
wg.Wait()
print(xInt32)

無論輸出多少次都無法達(dá)到10000,之所以如此就是因為此處的加1操作并不是原子的，都是先取當(dāng)前值，加1，再賦值，會出現(xiàn)覆蓋的情況。

修改

修改是最常用到的。

func modify(delta int32) {
atomic.AddInt32(&xInt32, delta)
atomic.AddInt64(&xInt64, int64(delta))
atomic.AddUint32(&xuInt32, uint32(delta))
atomic.AddUint64(&xuInt64, uint64(delta))
}

我們忽略了Uintptr的討論，這是內(nèi)存地址的整數(shù)表示，是用來存地址內(nèi)容的，暫時沒有遇到過指針的數(shù)據(jù)計算。

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)

改為原子操作后，發(fā)現(xiàn)每次運(yùn)行都可以得到預(yù)期的結(jié)果10000，

賦值與讀取

在并發(fā)情況下，讀取到某個變量后，在使用時變量內(nèi)容可能會被篡改，所以使用原子讀取。在并發(fā)情況下，為某個變量賦值的時候，必須要防止讀取到寫入一半的錯誤值，所以要用原子寫入。

var xInt32 int32
atomic.StoreInt32(&xInt32, 100)
println(xInt32)
v := atomic.LoadInt32(&xInt32)
println(v)

輸出

100
100

就目前而言，原子讀寫都是為了防止讀寫一半導(dǎo)致數(shù)據(jù)錯誤，但我無法復(fù)現(xiàn)這種錯誤的場景，假如你可以復(fù)現(xiàn)請在本文底部放留言。

var v atomic.Value
v.Store([]int{})
fmt.Println(v.Load().([]int))

也可以存儲其他任意類型，但如果使用到類似append擴(kuò)容原變量的語句，而不是使用直接替換的話，原子操作也是會失效的。

比較并交換

以下是節(jié)選自《Go并發(fā)編程實戰(zhàn)》一書中的例子，比較并交換（Compare And Swap）簡稱CAS，是樂觀鎖的核心思想，所以簡單介紹一下。

var xInt32 int32
for {
    v := atomic.LoadInt32(&xInt32)
    if atomic.CompareAndSwapInt32(&xInt32, v, v+100) {
        break
    }
}
print(xInt32)

這里一種無鎖的結(jié)構(gòu)，是一種思路，在需要改變數(shù)據(jù)的時候，反復(fù)判斷數(shù)據(jù)是否和原數(shù)據(jù)一致
一致時替換，不一致時說明被它處修改，則跳過
在不創(chuàng)建互斥量和不形成臨界區(qū)的情況下，完成并發(fā)安全的值替換操作。