聊聊高并发（二十）解析java.util.concurrent各个组件（二） 12个原子变量相关类

这篇说说java.util.concurrent.atomic包里的类，总共12个，网上有很多文章解析这几个类，这里挑些重点说说。

这12个类可以分为三组：

1. 普通类型的原子变量

2. 数组类型的原子变量

3. 域更新器

普通类型的原子变量的6个，

1. 其中AtomicBoolean, AtomicInteger, AtomicLong, AtomicReference分别对应boolean, int,  long, object完成基本的原子操作

2. AtomicMarkableReference, AtomicStampedReference是AtomicReference的功能增强版本，前者可以把引用跟一个boolean绑定，后者可以把引用和一个int型的版本号绑定来完成时间戳的功能。

AtomicBoolean, AtomicInteger, AtomicLong, AtomicReference这几个类的结构都相似，有几个特点

1. 底层都是采用sun.misc.Unsafe类来完成实际的CAS操作

2. 使用sun.misc.Unsafe直接操作内存对象来完成类似反射机制的对象属性存取能力

3. volatile类型的value值保存状态

4. 原子的get(), set()方法

5. 基本的compareAndSet方法完成CAS操作

6. weakCompareAndSet，弱化版本的CAS操作，这是API设计是预留地差异化接口，但是目前没有实现，目前和compareAndSet是一样的功能

7. getAndSet方法是利用CAS操作无锁地完成读取并设置的功能

8. lazySet方法优化设置，lazySet的使用看这篇 聊聊高并发（十八）理解AtomicXXX.lazySet方法

原子变量在并发编程中是基本的工具，可以用来实现非阻塞的数据结构和构建相关的基础构件。有几种基本的用法：

1. 安全的计数器

2. compareAndSet方法可以实现“滤网”的功能，找到第一个成功操作的线程，从而做一些操作，可以看看自旋锁相关的实现

3. compareAndSet方法可以实现“判断操作是否成功”的功能，这里会有ABA的问题，可以采用AtomicStampedReference来避免ABA问题

4. getAndSet方法可以实现完全的“设置并返回之前值”的功能

5. AtomicBoolean作为一种二元状态可以用来作为“开关”，实现找到一个打开开关的线程。比如 if(b.compareAndSet(false, true)){// dosomething}

来看几个典型的操作

public class AtomicBoolean implements java.io.Serializable {
    private static final long serialVersionUID = 4654671469794556979L;
    // setup to use Unsafe.compareAndSwapInt for updates
    private static final Unsafe unsafe = Unsafe.getUnsafe();
    private static final long valueOffset;

    // 使用Unsafe直接操作内存的方式设置属性的值
    static {
      try {
        valueOffset = unsafe.objectFieldOffset
            (AtomicBoolean.class.getDeclaredField("value"));
      } catch (Exception ex) { throw new Error(ex); }
    }
    // 使用volatile变量来保存状态
    private volatile int value;
    // 使用Unsafe的compareAndSwapXXX方法完成底层的CAS操作
    public final boolean compareAndSet(boolean expect, boolean update) {
        int e = expect ? 1 : 0;
        int u = update ? 1 : 0;
        return unsafe.compareAndSwapInt(this, valueOffset, e, u);
    }

    // 无锁地实现“设置并返回之前值”的功能，无锁的特点就是轮询加CAS操作
    public final boolean getAndSet(boolean newValue) {
        for (;;) {
            boolean current = get();
            if (compareAndSet(current, newValue))
                return current;
        }
    }
    // 优化volatie变量的写，再不需要保证可见性的场景下使用lazySet来优化，减少内存屏障
    public final void lazySet(boolean newValue) {
        int v = newValue ? 1 : 0;
        unsafe.putOrderedInt(this, valueOffset, v);
    }

AtomicMarkableReference和AtomicStampedReference都是对AtomicReference的增强，内部使用了不可变对象来保存一个二元状态<Reference, XXX>，当原子设置时，就创建新的对象，并把volaitle引用指向最新的不可变对象。更多内容请查看 聊聊高并发（十二）分析java.util.concurrent.atomic.AtomicStampedReference源码来看如何解决CAS的ABA问题

AtomicMarkableReference可以用来标记对象，常用来构建数据结构中表示节点，可以用boolean表示节点是否被删除

public class AtomicMarkableReference<V> {

    private static class Pair<T> {
        final T reference;
        final boolean mark;
        private Pair(T reference, boolean mark) {
            this.reference = reference;
            this.mark = mark;
        }
        static <T> Pair<T> of(T reference, boolean mark) {
            return new Pair<T>(reference, mark);
        }
    }

    private volatile Pair<V> pair;

    public AtomicMarkableReference(V initialRef, boolean initialMark) {
        pair = Pair.of(initialRef, initialMark);
    }

    public boolean compareAndSet(V       expectedReference,
                                 V       newReference,
                                 boolean expectedMark,
                                 boolean newMark) {
        Pair<V> current = pair;
        return
            expectedReference == current.reference &&
            expectedMark == current.mark &&
            ((newReference == current.reference &&
              newMark == current.mark) ||
             casPair(current, Pair.of(newReference, newMark)));
    } 

数组类型的原子变量有3个: AtomicIntegerArray, AtomicLongArray, AtomicReferenceArray, 它们是普通原子变量的数组版本，可以完成对数组中元素的原子操作。

基本的方法和普通原子类型类似，这里就不重复说了，提一下利用Unsafe读取数组元素的方法

1. 利用Unsafe.arrayBaseOffset得到数组的第一个元素的偏移量，因为有对象头的存在，所以offset不是从0开始

2. 利用Unsafe.arrayIndexScale得到数组中元素的长度

3. 利用移位操作代替乘法提高效率。所以先计算shift，比如8字节长度的元素，需要左移3位，相当与2的3次幂，4字节长度的元素左移2位，相当于2的2次幂

4. 计算数组中元素的实际位置 = index << shift + base，说白了，就是  index * 元素长度  + base

public class AtomicIntegerArray implements java.io.Serializable {

    private static final Unsafe unsafe = Unsafe.getUnsafe();
    // 获得数组第一个元素的偏移量offset，因为有对象头的存在，所以offset不是从0开始的 
    private static final int base = unsafe.arrayBaseOffset(int[].class);
    // 移位操作的基数，用移位操作代替乘法
    private static final int shift;
    private final int[] array;

    static {
        // 获取数组元素的长度，对于int[]数组， scale = 4
        int scale = unsafe.arrayIndexScale(int[].class);
        // 如果长度不是为2的幂，就报错
        if ((scale & (scale - 1)) != 0)
            throw new Error("data type scale not a power of two");
        // 31 - Integer.numberOfLeadingZeros(scale) 相当于求floor(log2(x)),这里为2，如果是Long,就是3
        // 其实就是用移位操作代替乘法，比如4字节长度，就要左移2位，8字节长度，就要左移3位。左移1位 = 乘 2
        shift = 31 - Integer.numberOfLeadingZeros(scale);
    }

    private long checkedByteOffset(int i) {
        if (i < 0 || i >= array.length)
            throw new IndexOutOfBoundsException("index " + i);

        return byteOffset(i);
    }
    // 用移位操作代替乘法，实际上求的是数组的第i个元素的偏移量，方便定位到数组元素的内存地址
    private static long byteOffset(int i) {
        return ((long) i << shift) + base;
    }

最后看看域更新器，有三个: AtomicIntegerFieldUpdate,  AtomicLongFieldUpdate,  AtomicReferenceFieldUpdate

域更新器是一种优化手段，它提供了现有volatile域的一种基于反射的视图，从而能对现有volatile域进行CAS操作，我们知道volatile字段只保证可见性，但是不保证原子性，

如果要想对volatile字段进行CAS操作，就要用到域更新器。它的好处是可以让volatile字段具备原子变量的能力，而不需要实际创建这么多的原子变量，毕竟volatile比起原子变量来说还是轻量级的。

域更新器没有提供对外的构造函数，它需要利用工厂方法的方式来创建，提供一个newUpdater(xxx)方法来返回一个新建的域更新器对象。

1. tclass指的是字段所在类的Class类型

2. vclass指的是字段的Class类型，需要注意的是字段必须是volatile标示的，不然会抛出异常

3. filedName字段名

4. 调用者的类型，可以用Reflection.getCallerClass()获得

 public static <U, W> AtomicReferenceFieldUpdater<U,W> newUpdater(Class<U> tclass, Class<W> vclass, String fieldName) {
        return new AtomicReferenceFieldUpdaterImpl<U,W>(tclass,
                                                        vclass,
                                                        fieldName,
                                                        Reflection.getCallerClass());
    }

  AtomicReferenceFieldUpdaterImpl(Class<T> tclass,
                                        Class<V> vclass,
                                        String fieldName,
                                        Class<?> caller) {
            Field field = null;
            Class fieldClass = null;
            int modifiers = 0;
            try {
                field = tclass.getDeclaredField(fieldName);
                modifiers = field.getModifiers();
                sun.reflect.misc.ReflectUtil.ensureMemberAccess(
                    caller, tclass, null, modifiers);
                sun.reflect.misc.ReflectUtil.checkPackageAccess(tclass);
                fieldClass = field.getType();
            } catch (Exception ex) {
                throw new RuntimeException(ex);
            }

            if (vclass != fieldClass)
                throw new ClassCastException();

            if (!Modifier.isVolatile(modifiers))
                throw new IllegalArgumentException("Must be volatile type");

            this.cclass = (Modifier.isProtected(modifiers) &&
                           caller != tclass) ? caller : null;
            this.tclass = tclass;
            if (vclass == Object.class)
                this.vclass = null;
            else
                this.vclass = vclass;
            offset = unsafe.objectFieldOffset(field);
        }

AtomicIntegerFieldUpdate这些更新器的接口和原子变量一致，都提供了compareAndSet操作，getAndSet操作等，这里不重复说。举个例子看看如何使用域更新器

1. Node类有一个volatile类型的next字段，它没有使用AtomicReference，使用了更轻量级的volatile

2. 如果想对volatile类型的next做CAS操作，就要创建域更新器AtomicReferenceFieldUpdater

private class Node<E>{
		private final E item;
		private volatile Node<E> next;
		
		public Node(E item){
			this.item = item;
		}
	}
	
	private static AtomicReferenceFieldUpdater<Node, Node> nextUpdate = AtomicReferenceFieldUpdater.newUpdater(Node.class, Node.class, "next");