Go CAS操作

go中的Cas操作與java中類似，都是借用了CPU提供的原子性指令來實現。CAS操作修改共享變量時候不需要對共享變量加鎖，而是通過類似樂觀鎖的方式進行檢查，本質還是不斷的占用CPU 資源換取加鎖帶來的開銷（比如上下文切換開銷）。下面一個例子使用CAS來實現計數器

package main import ("fmt""sync""sync/atomic" )var (counter int32 //計數器wg sync.WaitGroup //信號量 )func main() {threadNum := 5//1. 五個信號量wg.Add(threadNum)//2.開啟5個線程for i := 0; i < threadNum; i++ {go incCounter(i)}//3.等待子線程結束wg.Wait()fmt.Println(counter) }func incCounter(index int) {defer wg.Done()spinNum := 0for {//2.1原子操作old := counterok := atomic.CompareAndSwapInt32(&counter, old, old+1)if ok {break} else {spinNum++}}fmt.Printf("thread,%d,spinnum,%d\n",index,spinNum)} [root@localhost cas]# go run test1.go thread,4,spinnum,0 thread,0,spinnum,0 thread,1,spinnum,0 thread,2,spinnum,0 thread,3,spinnum,0 5

• 如上代碼main線程首先創建了5個信號量，然后開啟五個線程執行incCounter方法 incCounter內部執行代碼2.1
  使用cas操作遞增counter的值，
• atomic.CompareAndSwapInt32具有三個參數，第一個是變量的地址，第二個是變量當前值，第三個是要修改變量為多少，該函數如果發現傳遞的old值等于當前變量的值，則使用第三個變量替換變量的值并返回true，否則返回false。
• 這里之所以使用無限循環是因為在高并發下每個線程執行CAS并不是每次都成功，失敗了的線程需要重寫獲取變量當前的值，然后重新執行CAS操作。讀者可以把線程數改為10000或者更多會發現輸出thread,5329,spinnum,1其中1說明該線程嘗試了兩個CAS操作，第二次才成功。

總結

以上是生活随笔為你收集整理的Go CAS操作的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。