go CAS操作
  TEZNKK3IfmPf 2023年11月14日 26 0
go

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 := 0
    for {
     
       
        //2.1原子操作
        old := counter
        ok := 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操作,第二次才成功。
【版权声明】本文内容来自摩杜云社区用户原创、第三方投稿、转载,内容版权归原作者所有。本网站的目的在于传递更多信息,不拥有版权,亦不承担相应法律责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@moduyun.com

  1. 分享:
最后一次编辑于 2023年11月14日 0

暂无评论

推荐阅读
  TEZNKK3IfmPf   2023年11月15日   28   0   0 go定时器
  TEZNKK3IfmPf   2023年11月14日   32   0   0 go变量
  TEZNKK3IfmPf   2023年11月14日   36   0   0 字符串go
  TEZNKK3IfmPf   2023年11月14日   30   0   0 go函数
  TEZNKK3IfmPf   2023年11月15日   24   0   0 go
TEZNKK3IfmPf