现在的位置: 首页 > 综合 > 正文

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

2017年12月19日 ⁄ 综合 ⁄ 共 6868字 ⁄ 字号 评论关闭

这篇说说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");

抱歉!评论已关闭.