Go CAS操作
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操作,第二次才成功。
總結
- 上一篇: C#-数组截取的方法
- 下一篇: 使用 MyBatis 实体类里的 Dou