原子变量类Atomic*

最新推荐文章于 2024-07-10 17:46:51 发布

codedot

最新推荐文章于 2024-07-10 17:46:51 发布

阅读量1.2k

点赞数 1

分类专栏： Java并发

本文链接：https://blog.csdn.net/m0_53474063/article/details/112390020

版权

Java并发专栏收录该内容

10 篇文章 4 订阅

订阅专栏

原子变量最主要的一个特点就是所有的操作都是原子的，synchronized关键字也可以做到对变量的原子操作。只是synchronized的成本相对较高，需要获取锁对象，释放锁对象，如果不能获取到锁，还需要阻塞在阻塞队列上进行等待。而如果单单只是为了解决对变量的原子操作，建议使用原子变量。

一、原子变量的基本概念

原子变量保证了该变量的所有操作都是原子的，不会因为多线程的同时访问而导致脏数据的读取问题。我们先看一段synchronized关键字保证变量原子性的代码：

public class Counter {
    private int count;

    public synchronized void addCount(){
        this.count++;
    }
}

简单的count++操作，线程对象首先需要获取到Counter 类实例的对象锁，然后完成自增操作，最后释放对象锁。整个过程中，无论是获取锁还是释放锁都是相当消耗成本的，一旦不能获取到锁，还需要阻塞当前线程等等。

对于这种情况，我们可以将count变量声明成原子变量，那么对于count的自增操作都可以以原子的方式进行，就不存在脏数据的读取了。原原子变量类可以分为 4 组：

① 基本类型

AtomicBoolean - 布尔类型原子类
AtomicInteger - 整型原子类
AtomicLong - 长整型原子类

② 引用类型

AtomicReference - 引用类型原子类
AtomicMarkableReference - 带有标记位的引用类型原子类
AtomicStampedReference - 带有版本号的引用类型原子类，彻底解决了ABA问题

③ 数组类型

AtomicIntegerArray - 整形数组原子类
AtomicLongArray - 长整型数组原子类
AtomicReferenceArray - 引用类型数组原子类

④ 属性更新器类型

AtomicIntegerFieldUpdater - 整型字段的原子更新器。
AtomicLongFieldUpdater - 长整型字段的原子更新器。
AtomicReferenceFieldUpdater - 原子更新引用类型里的字段。

二、AtomicInteger的基本使用及实现原理

AtomicInteger的基本使用

首先看它的两个构造函数：

private volatile int value;

public AtomicInteger(int initialValue) {
    value = initialValue;
}
public AtomicInteger() {

}

可以看到，我们在通过构造函数构造AtomicInteger原子变量的时候，如果指定一个int的参数，那么该原子变量的值就会被赋值，否则就是默认的数值0。

也有获取和设置这个value值的方法：

public final int get()
public final void set(int newValue)

当然，这两个方法并不是原子的，所以一般也很少使用，而以下的这些基于原子操作的方法则相对使用的频繁：

//基于原子操作，获取当前原子变量中的值并为其设置新值
public final int getAndSet(int newValue)
//基于原子操作，比较当前的value是否等于expect，如果是设置为update并返回true，否则返回false
public final boolean compareAndSet(int expect, int update)
//基于原子操作，获取当前的value值并自增一
public final int getAndIncrement()
//基于原子操作，获取当前的value值并自减一
public final int getAndDecrement()
//基于原子操作，获取当前的value值并为value加上delta
public final int getAndAdd(int delta)
//还有一些反向的方法，比如：先自增在获取值的等等

下面我们实现一个计数器的例子，之前我们使用synchronized实现过，现在我们使用原子变量再次实现该问题。

//自定义一个线程类
public class MyThread extends Thread {

    public static AtomicInteger value = new AtomicInteger();

    @Override
    public void run(){
        try {
            Thread.sleep((long) ((Math.random())*100));
            //原子自增
            value.incrementAndGet();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }
}

//main函数中启动100条线程并让他们启动
public static void main(String[] args) throws InterruptedException {
    Thread[] threads = new Thread[100];
    for (int i=0;i<100;i++){
        threads[i] = new MyThread();
        threads[i].start();
    }

    for (int j=0;j<100;j++){
        threads[j].join();
    }

    System.out.println("value:"+MyThread.value);
}

AtomicInteger的内部实现原理

AtomicInteger的实现原理有点像我们的包装类，内部主要操作的是value字段，这个字段保存就是原子变量的数值。value字段定义如下：

private volatile int value;

首先value字段被volatile修饰，即不存在内存可见性问题。由于其内部实现原子操作的代码几乎类似，我们主要学习下incrementAndGet方法的实现。

在揭露该方法的实现原理之前，我们先看另一个方法：

public final boolean compareAndSet(int expect, int update{
     return unsafe.compareAndSwapInt(this, valueOffset, expect, update);
}

compareAndSet方法又被称为CAS，该方法调用unsafe的一个compareAndSwapInt方法，这个方法是native，我们看不到源码，但是我们需要知道该方法完成的一个目标：比较当前原子变量的值是否等于expect，如果是则将其修改为update并返回true，否则直接返回false。当然，这个操作本身就是原子的，较为底层的实现。

在jdk1.7之前，我们的incrementAndGet方法是这样实现的：

public final int incrementAndGet() {
    for (;;) {
        int current = get();
        int next = current + 1;
        if (compareAndSet(current, next))
            return next;
    }
}

方法体是一个死循环，current获取到当前原子变量中的值，由于value被修饰volatile，所以不存在内存可见性问题，数据一定是最新的。然后current加一后赋值给next，调用我们的CAS原子操作判断value是否被别的线程修改过，如果还是原来的值，那么将next的值赋值给value并返回next，否则重新获取当前value的值，再次进行判断，直到操作完成。

incrementAndGet方法的一个很核心的思想是，在加一之前先去看看value的值是多少，真正加的时候再去看一下，如果发现变了，不操作数据，否则为value加一。

但是在jdk1.8以后，做了一些优化，但是最后还是调用的compareAndSwapInt方法。但基本思想还是没变。

三、AtomicReference的基本使用

示例：基于 AtomicReference 实现一个简单的自旋锁

public class AtomicReferenceDemo2 {

    private static int ticket = 10;

    public static void main(String[] args) {
        threadSafeDemo();
    }

    private static void threadSafeDemo() {
        SpinLock lock = new SpinLock();
        ExecutorService executorService = Executors.newFixedThreadPool(3);
        for (int i = 0; i < 5; i++) {
            executorService.execute(new MyThread(lock));
        }
        executorService.shutdown();
    }

    /**
     * 基于 {@link AtomicReference} 实现的简单自旋锁
     */
    static class SpinLock {

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

        public void lock() {
            Thread current = Thread.currentThread();
            while (!atomicReference.compareAndSet(null, current)) {}
        }

        public void unlock() {
            Thread current = Thread.currentThread();
            atomicReference.compareAndSet(current, null);
        }

    }

    /**
     * 利用自旋锁 {@link SpinLock} 并发处理数据
     */
    static class MyThread implements Runnable {

        private SpinLock lock;

        public MyThread(SpinLock lock) {
            this.lock = lock;
        }

        @Override
        public void run() {
            while (ticket > 0) {
                lock.lock();
                if (ticket > 0) {
                    System.out.println(Thread.currentThread().getName() + " 卖出了第 " + ticket + " 张票");
                    ticket--;
                }
                lock.unlock();
            }
        }

    }

}

原子类的实现基于 CAS 机制，而 CAS 存在 ABA 问题。正是为了解决 ABA 问题，才有了 AtomicMarkableReference 和 AtomicStampedReference。

AtomicMarkableReference 使用一个布尔值作为标记，修改时在 true / false 之间切换。这种策略不能根本上解决 ABA 问题，但是可以降低 ABA 发生的几率。常用于缓存或者状态描述这样的场景。

public class AtomicMarkableReferenceDemo {

    private final static String INIT_TEXT = "abc";

    public static void main(String[] args) throws InterruptedException {

        final AtomicMarkableReference<String> amr = new AtomicMarkableReference<>(INIT_TEXT, false);

        ExecutorService executorService = Executors.newFixedThreadPool(3);
        for (int i = 0; i < 10; i++) {
            executorService.submit(new Runnable() {
                @Override
                public void run() {
                    try {
                        Thread.sleep(Math.abs((int) (Math.random() * 100)));
                    } catch (InterruptedException e) {
                        e.printStackTrace();
                    }

                    String name = Thread.currentThread().getName();
                    if (amr.compareAndSet(INIT_TEXT, name, amr.isMarked(), !amr.isMarked())) {
                        System.out.println(Thread.currentThread().getName() + " 修改了对象！");
                        System.out.println("新的对象为：" + amr.getReference());
                    }
                }
            });
        }

        executorService.shutdown();
        executorService.awaitTermination(3, TimeUnit.SECONDS);
    }

}

AtomicStampedReference 使用一个整型值做为版本号，每次更新前先比较版本号，如果一致，才进行修改。通过这种策略，可以根本上解决 ABA 问题。

public class AtomicStampedReferenceDemo {

    private final static String INIT_REF = "pool-1-thread-3";

    private final static AtomicStampedReference<String> asr = new AtomicStampedReference<>(INIT_REF, 0);

    public static void main(String[] args) throws InterruptedException {

        System.out.println("初始对象为：" + asr.getReference());

        ExecutorService executorService = Executors.newFixedThreadPool(3);
        for (int i = 0; i < 3; i++) {
            executorService.execute(new MyThread());
        }

        executorService.shutdown();
        executorService.awaitTermination(3, TimeUnit.SECONDS);
    }

    static class MyThread implements Runnable {

        @Override
        public void run() {
            try {
                Thread.sleep(Math.abs((int) (Math.random() * 100)));
            } catch (InterruptedException e) {
                e.printStackTrace();
            }

            final int stamp = asr.getStamp();
            if (asr.compareAndSet(INIT_REF, Thread.currentThread().getName(), stamp, stamp + 1)) {
                System.out.println(Thread.currentThread().getName() + " 修改了对象！");
                System.out.println("新的对象为：" + asr.getReference());
            }
        }

    }

}

四、数组类型的原子类

Java 提供了以下针对数组的原子类：

AtomicIntegerArray - 整形数组原子类
AtomicLongArray - 长整型数组原子类
AtomicReferenceArray - 引用类型数组原子类

已经有了针对基本类型和引用类型的原子类，为什么还要提供针对数组的原子类呢？

数组类型的原子类为 数组元素 提供了 volatile 类型的访问语义，这是普通数组所不具备的特性——volatile 类型的数组仅在数组引用上具有 volatile 语义。

示例：AtomicIntegerArray 使用示例（AtomicLongArray 、AtomicReferenceArray 使用方式也类似）

public class AtomicIntegerArrayDemo {

    private static AtomicIntegerArray atomicIntegerArray = new AtomicIntegerArray(10);

    public static void main(final String[] arguments) throws InterruptedException {

        System.out.println("Init Values: ");
        for (int i = 0; i < atomicIntegerArray.length(); i++) {
            atomicIntegerArray.set(i, i);
            System.out.print(atomicIntegerArray.get(i) + " ");
        }
        System.out.println();

        Thread t1 = new Thread(new Increment());
        Thread t2 = new Thread(new Compare());
        t1.start();
        t2.start();

        t1.join();
        t2.join();

        System.out.println("Final Values: ");
        for (int i = 0; i < atomicIntegerArray.length(); i++) {
            System.out.print(atomicIntegerArray.get(i) + " ");
        }
        System.out.println();
    }

    static class Increment implements Runnable {

        @Override
        public void run() {

            for (int i = 0; i < atomicIntegerArray.length(); i++) {
                int value = atomicIntegerArray.incrementAndGet(i);
                System.out.println(Thread.currentThread().getName() + ", index = " + i + ", value = " + value);
            }
        }

    }

    static class Compare implements Runnable {

        @Override
        public void run() {
            for (int i = 0; i < atomicIntegerArray.length(); i++) {
                boolean swapped = atomicIntegerArray.compareAndSet(i, 2, 3);
                if (swapped) {
                    System.out.println(Thread.currentThread().getName() + " swapped, index = " + i + ", value = 3");
                }
            }
        }

    }

}

五、属性更新器类型

更新器类支持基于反射机制的更新字段值的原子操作。

AtomicIntegerFieldUpdater - 整型字段的原子更新器。
AtomicLongFieldUpdater - 长整型字段的原子更新器。
AtomicReferenceFieldUpdater - 原子更新引用类型里的字段。

这些类的使用有一定限制：

因为对象的属性修改类型原子类都是抽象类，所以每次使用都必须使用静态方法 newUpdater() 创建一个更新器，并且需要设置想要更新的类和属性。
字段必须是 volatile 类型的；
不能作用于静态变量（static）；
不能作用于常量（final）；

public class AtomicReferenceFieldUpdaterDemo {

    static User user = new User("begin");

    static AtomicReferenceFieldUpdater<User, String> updater =
        AtomicReferenceFieldUpdater.newUpdater(User.class, String.class, "name");

    public static void main(String[] args) {
        ExecutorService executorService = Executors.newFixedThreadPool(3);
        for (int i = 0; i < 5; i++) {
            executorService.execute(new MyThread());
        }
        executorService.shutdown();
    }

    static class MyThread implements Runnable {

        @Override
        public void run() {
            if (updater.compareAndSet(user, "begin", "end")) {
                try {
                    TimeUnit.SECONDS.sleep(1);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
                System.out.println(Thread.currentThread().getName() + " 已修改 name = " + user.getName());
            } else {
                System.out.println(Thread.currentThread().getName() + " 已被其他线程修改");
            }
        }

    }

    static class User {

        volatile String name;

        public User(String name) {
            this.name = name;
        }

        public String getName() {
            return name;
        }

        public User setName(String name) {
            this.name = name;
            return this;
        }

    }

}