亲宝软件园·资讯

展开

Java CAS

温文艾尔 人气:0

开端

在学习源码之前我们先从一个需求开始

需求

我们开发一个网站,需要对访问量进行统计,用户每发送一次请求,访问量+1.如何实现?我们模拟有100个人同时访问,并且每个人对咱们的网站发起10次请求,最后总访问次数应该是1000次

1.代码

package day03;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;
/**
 * Description
 * User:
 * Date:
 * Time:
 */
public class Demo {
    //总访问量
    static int count = 0;
    //模拟访问的方法
    public static void request() throws InterruptedException {
        //模拟耗时5毫秒
        TimeUnit.MILLISECONDS.sleep(5);
        count++;
    }
    public static void main(String[] args) throws InterruptedException {
        long startTime = System.currentTimeMillis();
        int threadSize=100;
        CountDownLatch countDownLatch = new CountDownLatch(threadSize);
        for (int i=0;i<threadSize;i++){
            Thread thread = new Thread(new Runnable() {
                @Override
                public void run() {
                    //每个用户访问10次网站
                    try {
                        for (int j=0;j<10;j++) {
                            request();
                        }
                    }catch (InterruptedException e) {
                        e.printStackTrace();
                    }finally {
                        countDownLatch.countDown();
                    }
                }
            });
            thread.start();
        }
        //怎么保证100个线程执行之后,执行后面的代码
        countDownLatch.await();
        long endTime = System.currentTimeMillis();
        System.out.println(Thread.currentThread().getName()+"耗时:"+(endTime-startTime)+",count:"+count);
    }
}

我们多输出几次结果

main耗时:66,count:950

main耗时:67,count:928

发现每一次count都不相同,和我们期待的1000相差一点,这里就牵扯到了并发问题,我们的count++在底层实际上由3步操作组成

这并不是一个线程安全的过程,如果有A、B两个线程同时执行count++,同时执行到第一步,得到的count是一样的,三步操作完成后,count只加1,导致count结果不正确

那么怎么解决这个问题呢?

我们可以考虑使用synchronized关键字和ReentrantLock对资源加锁,保证并发的正确性,多线程的情况下,可以保证被锁住的资源被串行访问

1.1修改后的代码

package day03;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;
/**
 * Description
 * User:
 * Date:
 * Time:
 */
public class Demo02 {
    //总访问量
    static int count = 0;
    //模拟访问的方法
    public static synchronized void request() throws InterruptedException {
        //模拟耗时5毫秒
        TimeUnit.MILLISECONDS.sleep(5);
        count++;
    }
    public static void main(String[] args) throws InterruptedException {
        long startTime = System.currentTimeMillis();
        int threadSize=100;
        CountDownLatch countDownLatch = new CountDownLatch(threadSize);
        for (int i=0;i<threadSize;i++){
            Thread thread = new Thread(new Runnable() {
                @Override
                public void run() {
                    //每个用户访问10次网站
                    try {
                        for (int j=0;j<10;j++) {
                            request();
                        }
                    }catch (InterruptedException e) {
                        e.printStackTrace();
                    }finally {
                        countDownLatch.countDown();
                    }
                }
            });
            thread.start();
        }
        //怎么保证100个线程执行之后,执行后面的代码
        countDownLatch.await();
        long endTime = System.currentTimeMillis();
        System.out.println(Thread.currentThread().getName()+"耗时:"+(endTime-startTime)+",count:"+count);
    }
}

执行结果

main耗时:5630,count:1000

可以看到,由于sychronized锁住了整个方法,虽然结果正确,但因为线程执行方法均为串行执行,导致运行效率大大下降

那么我们如何才能使程序执行无误时,效率还不会降低呢?

缩小锁的范围,升级上述3步中第三步的实现

1.2代码改进:CAS模仿

package day03;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;
/**
 * Description
 * User:
 * Date:
 * Time:
 */
public class Demo03 {
    //总访问量
    volatile static int count = 0;
    //模拟访问的方法
    public static void request() throws InterruptedException {
        //模拟耗时5毫秒
        TimeUnit.MILLISECONDS.sleep(5);
//        count++;
        int expectCount;
        while (!compareAndSwap(expectCount=getCount(),expectCount+1)){}
    }
    /**
     * @param expectCount 期待的值,比如最刚开始count=3
     * @param newCount 新值 count+1之后的值,4
     * @return
     */
    public static synchronized  boolean compareAndSwap(int expectCount,int newCount){
        if (getCount()==expectCount){
            count = newCount;
            return true;
        }
        return false;
    }
    public static int getCount(){return count;}
    public static void main(String[] args) throws InterruptedException {
        long startTime = System.currentTimeMillis();
        int threadSize=100;
        CountDownLatch countDownLatch = new CountDownLatch(threadSize);
        for (int i=0;i<threadSize;i++){
            Thread thread = new Thread(new Runnable() {
                @Override
                public void run() {
                    //每个用户访问10次网站
                    try {
                        for (int j=0;j<10;j++) {
                            request();
                        }
                    }catch (InterruptedException e) {
                        e.printStackTrace();
                    }finally {
                        countDownLatch.countDown();
                    }
                }
            });
            thread.start();
        }
        //怎么保证100个线程执行之后,执行后面的代码
        countDownLatch.await();
        long endTime = System.currentTimeMillis();
        System.out.println(Thread.currentThread().getName()+"耗时:"+(endTime-startTime)+",count:"+count);
    }
}

main耗时:67,count:1000

2.CAS分析

CAS全称“CompareAndSwap”,中文翻译过来为“比较并替换”

定义:

2.1Java对CAS的支持

java中提供了对CAS操作的支持,具体在sun.misc.unsafe类中,声明如下

public final native boolean compareAndSwapObject(Object var1, long var2, Object var4, Object var5);
public final native boolean compareAndSwapInt(Object var1, long var2, int var4, int var5);
public final native boolean compareAndSwapLong(Object var1, long var2, long var4, long var6);

2.2CAS实现原理是什么?

CAS通过调用JNI的代码实现,JNI:java native interface,允许java调用其他语言。而compareAndSwapxxx系列的方法就是借助C语言来调用cpu底层指令实现的

以常用的Intel x86平台为例,最终映射到cpu的指令为"cmpxchg",这是一个原子指令,cpu执行此命令时,实现比较并替换的操作

现代计算机动不动就上百核心,cmpxchg怎么保证多核心下的线程安全?

系统底层在进行CAS操作的时候,会判断当前系统是否为多核心系统,如果是就给“总线”加锁,只有一个线程会对总线加锁成功,加锁之后执行CAS操作,也就是说CAS的原子性是平台级别的

2.3CAS存在的问题

2.3.1什么是ABA问题?

CAS需要在操作值的时候检查下值有没有发生变化,如果没有发生变化则更新,但是如果一个值原来是A,在CAS方法执行之前,被其他线程修改为B,然后又修改回了A,那么CAS方法执行检查的时候会发现它的值没有发生变化,但是实际却不是原来的A了,这就是CAS的ABA问题

在这里插入图片描述

可以看到上图中线程A在真正更改A之前,A已经被其他线程修改为B然后又修改为A了

程序模拟ABA问题

package day04;
import java.util.concurrent.atomic.AtomicInteger;
/**
 * Description
 * User:
 * Date:
 * Time:
 */
public class Test01 {
    public static AtomicInteger a = new AtomicInteger();
    public static void main(String[] args) {
        Thread main = new Thread(new Runnable() {
            @Override
            public void run() {
                System.out.println(Thread.currentThread().getName()+"执行,a的值为:"+a.get());
                try {
                    int expect = a.get();
                    int update = expect+1;
                    //让出cpu
                    Thread.sleep(1000);
                    boolean b = a.compareAndSet(expect, update);
                    System.out.println(Thread.currentThread().getName()+"CAS执行:"+b+",a的值为:"+a.get());
                }
                 catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        },"主线程");
//        main.start();
        Thread thread1 = new Thread(new Runnable() {
            @Override
            public void run() {
                try {
                    Thread.sleep(20);
                    a.incrementAndGet();
                    System.out.println(Thread.currentThread().getName()+"更改a的值为:"+a.get());
                    a.decrementAndGet();
                    System.out.println(Thread.currentThread().getName()+"更改a的值为:"+a.get());
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        },"其他线程");
        main.start();
        thread1.start();
    }
}

主线程执行,a的值为:0
其他线程更改a的值为:1
其他线程更改a的值为:0
主线程CAS执行:true,a的值为:1

可以看到,在执行CAS之前,a被其他线程修改为1又修改为0,但是对执行CAS并没有影响,因为它根本没有察觉到其他线程对a的修改

2.3.2如何解决ABA问题

解决ABA问题最简单的方案就是给值加一个修改版本号,每次值变化,都会修改它的版本号,CAS操作时都去对比此版本号

在java中的ABA解决方案(AtomicStampedReference

AtomicStampedReference主要包含一个对象引用及一个可以自动更新的整数stamp的pair对象来解决ABA问题

AtomicStampedReference源码

    /**
     * Atomically sets the value of both the reference and stamp
     * to the given update values if the
     * current reference is {@code ==} to the expected reference
     * and the current stamp is equal to the expected stamp.
     *
     * @param expectedReference the expected value of the reference 期待引用
     * @param newReference the new value for the reference          新值引用
     * @param expectedStamp the expected value of the stamp         期望引用的版本号
     * @param newStamp the new value for the stamp                  新值的版本号
     * @return {@code true} if successful
     */
    public boolean compareAndSet(V   expectedReference,
                                 V   newReference,
                                 int expectedStamp,
                                 int newStamp) {
        Pair<V> current = pair;
        return
            expectedReference == current.reference &&//期望引用与当前引用保持一致
            expectedStamp == current.stamp &&//期望引用版本号与当前版本号保持一致
            ((newReference == current.reference &&//新值引用与当前引用一致并且新值版本号与当前版本号保持一致
              newStamp == current.stamp)
                    ||//如果上述版本号不一致,则通过casPair方法新建一个Pair对象,更新值和版本号,进行再次比较
             casPair(current, Pair.of(newReference, newStamp)));
    }
    private boolean casPair(Pair<V> cmp, Pair<V> val) {
        return UNSAFE.compareAndSwapObject(this, pairOffset, cmp, val);
    }

使用AtomicStampedReference解决ABA问题代码

package day04;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.concurrent.atomic.AtomicReference;
import java.util.concurrent.atomic.AtomicStampedReference;
/**
 * Description
 * User:
 * Date:
 * Time:
 */
public class Test02 {
    public static AtomicStampedReference<Integer> a = new AtomicStampedReference(new Integer(1),1);
    public static void main(String[] args) {
        Thread main = new Thread(new Runnable() {
            @Override
            public void run() {
                System.out.println(Thread.currentThread().getName()+"执行,a的值为:"+a.getReference());
                try {
                    Integer expectReference = a.getReference();
                    Integer newReference = expectReference+1;
                    Integer expectStamp = a.getStamp();
                    Integer newStamp = expectStamp+1;
                    //让出cpu
                    Thread.sleep(1000);
                    boolean b = a.compareAndSet(expectReference, newReference,expectStamp,newStamp);
                    System.out.println(Thread.currentThread().getName()+"CAS执行:"+b);
                }
                catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        },"主线程");
//        main.start();
        Thread thread1 = new Thread(new Runnable() {
            @Override
            public void run() {
                try {
                    Thread.sleep(20);
                    a.compareAndSet(a.getReference(),a.getReference()+1,a.getStamp(),a.getStamp()+1);
                    System.out.println(Thread.currentThread().getName()+"更改a的值为:"+a.getReference());
                    a.compareAndSet(a.getReference(),a.getReference()-1,a.getStamp(),a.getStamp()-1);
                    System.out.println(Thread.currentThread().getName()+"更改a的值为:"+a.getReference());
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        },"其他线程");
        main.start();
        thread1.start();
    }
}

主线程执行,a的值为:1
其他线程更改a的值为:2
其他线程更改a的值为:1
主线程CAS执行:false

因为AtomicStampedReference执行CAS会去检查版本号,版本号不一致则不会进行CAS,所以ABA问题成功解决

总结

本篇文章就到这里了,希望能够给你带来帮助,也希望您能够多多关注的更多内容!   

加载全部内容

相关教程
猜你喜欢
用户评论