【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