前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >CAS 原理深入剖析,深入内核源码的那种

CAS 原理深入剖析,深入内核源码的那种

作者头像
kk大数据
发布2021-03-13 22:22:28
1K0
发布2021-03-13 22:22:28
举报
文章被收录于专栏:kk大数据

一、CAS 简介

CAS 的意思是 compare and swap,比较并交换。

CAS 的示意图如下:

比如一个很简单的操作,把变量 A = 2 加 1,结果为 3.

则先读取 A 的当前值 E 为 2,在内存计算结果 V 为 3,比较之前读出来的 A 的当前值 2 和 最新值,如果最新值为 2 ,表示这个值没有被别人改过,则放心的把最终的值更新为 3.

有一种情况是,在你更新结果之前,其他有个线程在中途把 A 更新成了 5 ,又更新回了 2。但是在当前线程看起来,没有被改过。这就是 ABA 问题。

二、CAS 的实现

在 java 中,原子类都是用 cas 来实现的,我们可以看一看源码。

AtomicInteger 类的 getAndIncrement() 方法

代码语言:javascript
复制
public final int getAndIncrement() {
    return unsafe.getAndAddInt(this, valueOffset, 1);
}

发现是调用了 unsafe 类的 getAndAddInt,继续看这个方法:

代码语言:javascript
复制
public final int getAndAddInt(Object var1, long var2, int var4) {
    int var5;
    do {
        var5 = this.getIntVolatile(var1, var2);
    } while(!this.compareAndSwapInt(var1, var2, var5, var5 + var4));

    return var5;
}

这里是一个 while 循环,直到比较成功,来看一下这个 compareAndSwapInt 方法

代码语言:javascript
复制
public final native boolean compareAndSwapInt(Object var1, long var2, int var4, int var5);

发现这是一个 native 方法,意味着是 c 或者 c++ 写的虚拟机的实现。

在网上找到了 HotSpot 虚拟机的源码,找了一些资料发现了这个 compareAndSwapInt 的 c++ 代码:

在 unsafe.cpp 这个文件的 Unsafe_CompareAndSwapInt 这个方法里

发现最终调用的是 Atomic:: cmpxchg 方法,我们再找到 atomic_linux_x86.inline.hpp 这个文件

其中有一句 LOCK_IF_MP(%4) "cmpxchgl %1,(%3)"

LOCK_IF_MP 是一个宏,这个宏的定义是:

mp 的意思是 multi processor,意思是在多核 cpu 上,要锁一下这个指令。

到这里,结论是,最终调用了一条汇编指令:lock cmpxchg 指令,来实现底层 cas 的。

也就是 cpu 中有一条 cmpxchg 指令。

但是这条指令不是原子的,也就是拿出来和比较是两个操作,中间有可能被别人打断。

所以需要在这个过程加上 lock,意思是,我在对这个内存操作的过程中,不允许被别人打断。

可以简单理解为把内存总线锁住,别人不允许修改这块内存。

三、ABA 问题的解决

很简单,可以在数据上加上版本号即可,改了一次就新增一个版本号。

在 Java 中,可以使用 AtomicStampedReference 来解决这个问题

代码语言:javascript
复制
import java.util.concurrent.atomic.AtomicInteger;
import java.util.concurrent.atomic.AtomicStampedReference;

public class ABASingle {

    public static void main(String[] args) {
        AtomicInteger atomicInt = new AtomicInteger(100);
        atomicInt.compareAndSet(100, 101);
        atomicInt.compareAndSet(101, 100);
        System.out.println("new value = " + atomicInt.get());
        boolean result1 = atomicInt.compareAndSet(100, 101);
        System.out.println(result1); // result:true

        AtomicInteger v1 = new AtomicInteger(100);
        AtomicInteger v2 = new AtomicInteger(101);
        AtomicStampedReference<AtomicInteger> stampedRef = new AtomicStampedReference<AtomicInteger>(
                v1, 0);

        int stamp = stampedRef.getStamp();
        stampedRef.compareAndSet(v1, v2, stampedRef.getStamp(),
                stampedRef.getStamp() + 1);
        stampedRef.compareAndSet(v2, v1, stampedRef.getStamp(),
                stampedRef.getStamp() + 1);
        System.out.println("new value = " + stampedRef.getReference());
        boolean result2 = stampedRef.compareAndSet(v1, v2, stamp, stamp + 1);
        System.out.println(result2); // result:false
    }
}
本文参与 腾讯云自媒体同步曝光计划,分享自微信公众号。
原始发表:2021-03-06,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 KK架构 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 一、CAS 简介
  • 二、CAS 的实现
  • 三、ABA 问题的解决
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档