亲宝软件园·资讯

展开

多线程之美6一CAS与自旋锁

夕阳下飞奔的猪 人气:1

1、什么是CAS

CAS 即 compare and swap 比较并交换, 涉及到三个参数,内存值V, 预期值A, 要更新为的值B, 拿着预期值A与内存值V比较,相等则符合预期,将内存值V更新为B, 不相等,则不能更新V。

为什么预期值A与内存值V不一样了呢?

在多线程环境下,对于临界区的共享资源,所有线程都可以访问修改,这时为了保证数据不会发生错误,通常会对访问临界区资源加锁,同一时刻最多只能让一个线程访问(独占模式下),这样会让线程到临界区时串行执行,加锁操作可能会导致并发性能降低,而循环CAS可以实现让多个线程不加锁去访问共享资源,却也可以保证数据正确性。 如 int share = 1,线程A获取到share的值1,想要将其修改为2,这时线程B抢先修改share = 3了,线程A这时拿着share =1 预期值与实际内存中已经变为3的值比较, 不相等,cas失败,这时就重新获取最新的share再次更新,需要不断循环,直到更新成功;这里可能会存在线程一直在进行循环cas,消耗cpu资源。

cas缺点:

1、存在ABA问题

2、循环cas, 可能会花费大量时间在循环,浪费cpu资源

3、只能更新一个值(也可解决,AtomicReference

2、原子类 Atomic

原子类在JUC的atomic包下提供了 AtomicInteger,AtomicBoolean, AtomicLong等基本数据类型原子类,还有可传泛型的AtomicReference

类图结构如下:

分别用AtomicInteger和 Integer 演示多个线程执行自增操作,是否能够保证原子性,执行结果是否正确

代码如下:

/**
 * @author zdd
 * 2019/12/22 10:47 上午
 * Description: 演示AtomicInteger原子类原子操作
 */
public class CasAtomicIntegerTest {
    static  final Integer THREAD_NUMBER = 10;
    static  AtomicInteger atomicInteger = new AtomicInteger(0);
    static  volatile Integer integer = 0;

    public static void main(String[] args) throws InterruptedException {
        ThreadTask task = new ThreadTask();
        Thread[] threads = new Thread[THREAD_NUMBER];
        //1,开启10个线程
        for (int j = 0; j < THREAD_NUMBER; j++) {
            Thread thread  = new Thread(task);
            threads[j]= thread;
        }
        for (Thread thread:threads) {
            //开启线程
            thread.start();
            //注: join 为了保证主线程在所有子线程执行完毕后再打印结果,否则主线程就阻塞等待
           // thread.join();
        }

        // 主线程休眠5s, 等待所有子线程执行完毕再打印
        TimeUnit.SECONDS.sleep(5);

        System.out.println("执行完毕,atomicInteger的值为: "+ atomicInteger.get());
        System.out.println("执行完毕,integer的值为 : "+ integer);
    }

    public static void  safeIncr() {
        atomicInteger.incrementAndGet();
    }
    public static void  unSafeIncr() {
        integer ++;
    }

    static class ThreadTask implements  Runnable{
        @Override
        public void run() {
            // 任务体,分别安全和非安全方式自增1000次
            for (int i = 0; i < 1000; i++) {
                safeIncr();
            }
            for (int i = 0; i < 1000; i++) {
                unSafeIncr();
            }
        }
    }
}

执行结果如下:

疑问:上文代码中注,我本想让主线程调用每个子线程 join方法,保证主线程在所有子线程执行完毕之后再执行打印结果,然而这样执行导致非安全的Integer自增结果也正确,猜想是在执行join方法,导致这10个子线程排队有序在执行了? 因此注释了该行代码 ,改为让主线程休眠几秒来保证在子线程执行后再打印。

AtomicInteger如何保证原子性,AtomicInteger持有Unsafe对象,其大部分方法是本地方法,底层实现可保证原子操作。

public class AtomicInteger extends Number implements java.io.Serializable {
    // setup to use Unsafe.compareAndSwapInt for updates
    private static final Unsafe unsafe = Unsafe.getUnsafe();

来看一下 AtomicInteger 的自增方法 incrementAndGet(),先自增,再返回增加后的值。

代码如下:

  public final int incrementAndGet() {
       //调用unsafe的方法
        return unsafe.getAndAddInt(this, valueOffset, 1) + 1;
    }

继续看unsafe如何实现

  public final int getAndAddInt(Object var1, long var2, int var4) {
        int var5;
        do {
          //1.获取当前对象的内存中的值A
            var5 = this.getIntVolatile(var1, var2);
          //2. var1,var2联合获取内存中的值V,var5是期望中的值A, var5+var4 是将要更新为的新值
        } while(!this.compareAndSwapInt(var1, var2, var5, var5 + var4));
       //3. 更新成功,跳出while循环,返回更新成功时内存中的值(可能下一刻就被其他线程修改)
        return var5;
    }

执行流程图如下:

Unsafe 的compareAndSwapInt是本地方法,可原子地执行更新操作,更新成功返回true,否则false

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

3、CAS的ABA问题

什么是ABA问题?

例如 线程A获取变量atomicInteger =100, 想要将其修改为2019 (此时还未修改), 这时线程B抢先进来将atomicInteger先修改为101,再修改回atomicInteger =100,这时线程A开始去更新atomicInteger的值了,此时预期值和内存值相等,更新成功atomicInteger =2019;但是线程A 并不知道这个值其实已经被人修改过了。

代码演示如下:

/**
 * zdd
 * Description: cas的ABA问题
 */
public class CasTest1 {

   // static AtomicInteger atomicInteger = new AtomicInteger(100);
   /* 这里使用原子引用类,传入Integer类型,
    * 和AtomicInteger一样,AtomicReference使用更灵活,泛型可指定任何引用类型。
    * 也可用上面注释代码
    */
    static AtomicReference<Integer>  reference = new AtomicReference<>(100);

    public static void main(String[] args) {
  
      //1.开启线程A
        new Thread(()-> {
            Integer expect =  reference.get();
            try {
                //模拟执行任务,让线程B抢先修改
                TimeUnit.SECONDS.sleep(3);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            System.out.println( "执行3s任务后, 修改值是否成功 "+ reference.compareAndSet(expect,2019)+ "  当前值为: "+ reference.get());
        },"A").start();     
    //2.开启线程B
        new Thread(()-> {
            // expect1 =100
            Integer expect1 =  reference.get();
            //1,先修改为101,再修改回100,产生ABA问题
            reference.compareAndSet(expect1,101);
            //expect2 =101
            Integer expect2 =  reference.get();
            reference.compareAndSet(expect2, 100);
        },"B").start();

    }
}      

执行结果如下:可见线程A修改成功

A 执行3s任务后, 修改值是否成功:true  当前值为: 2019

4、ABA问题的解决方式

解决CAS的ABA问题,是参照数据库乐观锁,添加一个版本号,每更新一次,次数+1,就可解决ABA问题了。

AtomicStampedReference

/**
 * zdd
 * 2019/11/4 6:30 下午
 * Description:
 */
public class CasTest1 {
  //设置初始值和版本号
    static  AtomicStampedReference<Integer> stampedReference = new AtomicStampedReference<>(100,1);

    public static void main(String[] args) {
        //2,采用带有版本号的 
        new Thread(()-> {
            Integer  expect = stampedReference.getReference();
            int     stamp = stampedReference.getStamp();
            try {
                //休眠3s,让线程B执行完ABA操作
                TimeUnit.SECONDS.sleep(3);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            //此时 stamp=1,与实际版本号3不等,这里更新失败就是stamp没有获取到最新的
            System.out.println("是否修改成功: "+stampedReference.compareAndSet(expect, 101, stamp, stamp +1));
            System.out.println("当前 stamp 值: " + stampedReference.getStamp()+ "当前 reference: " +stampedReference.getReference());

        },"A").start();

        new Thread(()-> {
            Integer expect = stampedReference.getReference();
            int stamp = stampedReference.getStamp();
            try {
                //休眠1s,让线程A获取都旧的值和版本号
                TimeUnit.SECONDS.sleep(1);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            // 1,100 -> 101, 版本号 1-> 2
            stampedReference.compareAndSet(expect, 101 , stamp, stamp+1);
            //2, 101 ->100, 版本号 2->3
            Integer expect2 = stampedReference.getReference();
            stampedReference.compareAndSet(expect2, 100, stampedReference.getStamp(), stampedReference.getStamp() + 1);

        },"B").start();
    }
}

执行结果如下:

是否修改成功: false
当前 stamp 值: 3  当前 reference: 100

5、利用cas实现自旋锁

package cas;

import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicReference;

/**
 * @author zdd
 * 2019/12/22 9:12 下午
 * Description: 利用cas手动实现自旋锁
 */
public class SpinLockTest {

    static   AtomicReference<Thread>  atomicReference = new AtomicReference<>();

    public static void main(String[] args) {
        SpinLockTest spinLockTest = new SpinLockTest();
        //测试使用自旋锁,达到同步锁一样的效果 ,开启2个子线程
        new Thread(()-> {
            spinLockTest.lock();
            System.out.println(Thread.currentThread().getName()+" 开始执行,startTime: "+System.currentTimeMillis());
            try {
                //休眠3s
                TimeUnit.SECONDS.sleep(3); } catch (InterruptedException e) { e.printStackTrace();
            }
            System.out.println(Thread.currentThread().getName()+" 结束执行,endTime: "+System.currentTimeMillis());
            spinLockTest.unLock();
        },"线程A").start();

        new Thread(()-> {
            spinLockTest.lock();
            System.out.println(Thread.currentThread().getName()+" 开始执行,startTime: "+System.currentTimeMillis());
            try {
                //休眠3s
                TimeUnit.SECONDS.sleep(3); } catch (InterruptedException e) { e.printStackTrace();
            }
            System.out.println(Thread.currentThread().getName()+" 结束执行,endTime: "+System.currentTimeMillis());
            spinLockTest.unLock();
        },"线程B").start();
    }
    public static void lock() {
      Thread currentThread =  Thread.currentThread();
      for (;;) {
          boolean flag =atomicReference.compareAndSet(null,currentThread);
         //cas更新成功,则跳出循环,否则一直轮询
          if(flag) {
              break;
          }
      }
    }
    public static void unLock() {
        Thread currentThread = Thread.currentThread();
        Thread momeryThread  = atomicReference.get();
        //比较内存中线程对象与当前对象,不等抛出异常,防止未获取到锁的线程调用unlock
        if(currentThread != momeryThread) {
            throw new IllegalMonitorStateException();
        }
        //释放锁
        atomicReference.compareAndSet(currentThread,null);
    }
}

执行结果如下图:

6、总结

通过全文,我们可以知道cas的概念,它的优缺点;原子类的使用,内部借助Unsafe类循环cas更新操作实现无锁情况下保证原子更新操作,进一步我们能够自己利用循环cas实现自旋锁SpinLock,它与同步锁如ReentrantLock等区别在于自旋锁是在未获取到锁情况,一直在轮询,线程时非阻塞的,对cpu资源占用大,适合查询多修改少场景,并发性能高;同步锁是未获取到锁,阻塞等待,两者各有适用场景。

加载全部内容

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