【Java】Java CAS

作者: 黑暗终将过去 | 来源:发表于2018-05-27 17:18 被阅读12次

CAS

首先,先说下悲观锁和乐观锁,悲观锁,每次操作都先上锁,比如独占锁synchronized就是一种悲观锁;乐观锁就是每次不加锁而是假设没有冲突而去完成某项操作,如果因为冲突失败就重试,直到成功为止。乐观锁所用到的机制就是CAS。

CAS:compare and swap

CAS包括三个值:内存地址、期待原值、新值。即当赋值的时候,首先compare,即对比目前的值和期待原值是否一样,如果一样,则更新原值,否则,不更新原值,重新获取。

看一个例子。

例子

public class Counter {
    private AtomicInteger mAtomicInteger = new AtomicInteger(0);
    private int i = 0;
    public static void main(String []args) {
        final Counter cas = new Counter();
        List<Thread> ts = new ArrayList<>(600);
    
        for(int j = 0; j < 100; j++) {
            Thread thread = new Thread(new Runnable() {
                public void run() {
                    for(int i = 0; i < 1000; i++) {
                        cas.count();
                        cas.safeCount();
                    }
                }
            });
            ts.add(thread);
        }
    
        for(Thread thread : ts) {
            thread.start();
        }
    
        //等待子线程都执行完了之后再打印结果
        for(Thread thread : ts) {
            try {
                thread.join();
            } catch(InterruptedException e) {
                e.printStackTrace();
            }
        }
    
        System.out.println(cas.i);
        System.out.println(cas.mAtomicInteger.get());
    }

    private void safeCount() {
        for(;;) {
            int i = mAtomicInteger.get();
            boolean suc = mAtomicInteger.compareAndSet(i, ++i);
            if (suc) {
                break;
            }
        }
    }

    private void count() {
        i++;
    }
}

这个是一个计数器,期待结果是100000,但是,如果直接i++会发现,结果不是100000,原因比较好理解,修改值的时候包括三步:
1、读取i
2、i = i + 1
3、把计算得到的i写入
这个过程明显不是原子操作的,所以导致值不对。
再看使用CAS方式来实现的,即:

for(;;) {
    int i = mAtomicInteger.get();
    boolean suc = mAtomicInteger.compareAndSet(i, ++i);
    if (suc) {
        break;
    }
}

这个非常有意思,compareAndSet传入的一个期待原值一个新值,如果原值和期待原值不一致,就返回false,如果一致,就修改并返回true。如果一致不成功就不断重试,直到成功。

所以其实这里是通过循环CAS来实现的不加锁的原子操作。

ABA问题

CAS操作的是存在着一个经典的ABA问题,这个问题可以从下面的例子引入。

public class ABA {
    private static AtomicInteger atomicInt = new AtomicInteger(100);

    public static void main(String[] args) throws InterruptedException {
        Thread intT1 = new Thread(new Runnable() {
            @Override
            public void run() {
                atomicInt.compareAndSet(100, 101);
                atomicInt.compareAndSet(101, 100);
            }
        });

        Thread intT2 = new Thread(new Runnable() {
            @Override
            public void run() {
                try {
                    TimeUnit.SECONDS.sleep(1);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
                boolean c3 = atomicInt.compareAndSet(100, 101);
                System.out.println(c3); //true
            }
        });

        intT1.start();
        intT2.start();
        intT1.join();
        intT2.join();
    }
}

打印结果是true。

这里初始值是100,线程1把100改成101再改成100,线程2认为100就是期待原值,所以线程2在执行的时候,不知道1已经执行了。所以返回了true。但是实际上2的期待原值100已经不是现在的100了。

ABA问题的解决思路就是加上版本号,即从A->B->A变成1A->2B->3A,这样一来只有版本号和值相等,才认为相等。

从Java1.5开始,java.util.concurrent.atomic包里面引入了带标记的AtomicStampedReference类解决了这个问题。

还是先看这个例子的更新版。

public class ABA {
    private static AtomicStampedReference<Integer> atomicStampedRef = new AtomicStampedReference<Integer>(100, 0);

    public static void main(String[] args) throws InterruptedException {
        Thread refT1 = new Thread(new Runnable() {
            @Override
            public void run() {
                try {
                    TimeUnit.SECONDS.sleep(1);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
                atomicStampedRef.compareAndSet(100, 101, atomicStampedRef.getStamp(), atomicStampedRef.getStamp() + 1);
                atomicStampedRef.compareAndSet(101, 100, atomicStampedRef.getStamp(), atomicStampedRef.getStamp() + 1);
            }
        });

        Thread refT2 = new Thread(new Runnable() {
            @Override
            public void run() {
                int stamp = atomicStampedRef.getStamp();
                try {
                    TimeUnit.SECONDS.sleep(2);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
                boolean c3 = atomicStampedRef.compareAndSet(100, 101, stamp, stamp + 1);
                System.out.println(c3); //false
            }
        });

        refT1.start();
        refT2.start();
    }
}

打印结果是false。
这里compareAndSet方法做了两件事,一个是先检查当前引用是否等于预期引用,并且检查当前版本号是否等于预期版本号,如果相等则进行更新。

相关文章

网友评论

    本文标题:【Java】Java CAS

    本文链接:https://www.haomeiwen.com/subject/ueobjftx.html