java多线程六：原子操作类Atomic

由于Java中变量会有多种类型，所以Atomic包中提供了13个类，属于4种类型的原子更新方式，分别是原子更新基本类型、原子更新数组、原子更新引用和原子更新属性（字段）。其实实现原理都是对Unsafe类中一些方法的包装。

1. 原子更新基本类型

使用原子的方式更新基本类型，Atomic包提供了一下3个类。

• AtomicBoolean：原子更新布尔类型。
• AtomicInteger：原子更新整形
• AtomicLong：原子更新长整型。

当然这3个类提供的方法几乎一模一样，所以就只说一下AtomicInteger，这个类常用的方法有：

• int addAndGet(int data)：以原子方式将输入的数值与实例中的值（AtomicInteger封装的value）相加，并返回结果。
• int getAndIncrement()：以原子方式将当前值加1，返回自增前的值。
• void lazySet(int newValue)：最终会设置成newValue，使用lazySet设置值后，可能导致其他线程在之后的一小段时间内还可以读到旧的值。
• int getAndSet(int newValue)：以原子方式设置为newValue的值，并返回旧值。
• boolean compareAndSet(int expect, int update)：若输入的数值等于预期值，则以原子方式将该值设置为输入的值。

1.1 递增和递减操作代码

包括getAndIncrement()、getAndDecrement()、incrementAndGet()、decrementAndGet()等方法。他们都是通过调用Unsafe的getAndAddInt()方法来实现操作。这个函数是一个原子性操作，第一个参数是AtomicInt实例的引用，第二个参数是value变量在AtomicInt中的偏移值，第三个参数是要设置的第二个变量的值，其代码如下：

	//AtomicInteger的getAndIncrement方法
	public final int getAndIncrement() {
        return unsafe.getAndAddInt(this, valueOffset, 1);
    }

	//Unsafe的getAndInt()方法
    public final int getAndAddInt(Object var1, long var2, int var4) {
        int var5;
        do {
            var5 = this.getIntVolatile(var1, var2);	//得到var1对象中的var2变量（当前值）
        } while(!this.compareAndSwapInt(var1, var2, var5, var5 + var4));	//更改不成功继续循环

        return var5;	//返回更改前的值
    }

其实在JDK1.7中，getAndIncrement()方法是自己实现的，也是使用的CAS不停检查并尝试更新。只是没有用Unsafe实现，其代码如下：

public final int getAndIncrement() {
	while (true) {
		int current = get();
		int next = current + 1;
		if (compareAndSet(current, next))
			return current;
	}
}

所以只是才JDK1.8中将这个方法用Unsafe封装了起来，本质上都是CAS操作。

1.2 public final boolean compareAndSet(int expect, int update)

此方法是基于Unsafe类的compareAndSwapInt方法实现的：

    public final boolean compareAndSet(int expect, int update) {
        return unsafe.compareAndSwapInt(this, valueOffset, expect, update);
    }
	
	//这是一个本地方法
	public final native boolean compareAndSwapInt(Object var1, long var2, int var4, int var5);

其实就是调用了unsafe.compareAndSwapInt方法，第一个参数this是对象引用，第二个参数相当于要更新的值在对象中的偏移地址，它总是会在类加载的时候计算出来。上面的getAndIncrement方法内部的CAS其实也是通过这个方法来不断进行尝试更改的过程如下：

    static {
        try {
        //调用这个方法得到类AtomicInteger中value域的偏移地址
            valueOffset = unsafe.objectFieldOffset		
                (AtomicInteger.class.getDeclaredField("value"));
        } catch (Exception ex) { throw new Error(ex); }
    }

    private volatile int value;		//存放值的变量

Atomic包提供了3种基本类型的原子更新，但是Java的基本类型还有char、float和double等，那其他基本类型如何实现原子更新呢？而且Unsafe只提供了3中CAS方法：compareAndSwapObject、compareAndSwapInt和compareAndSwapLong，看了一下AtomicBoolean源码，发现它是先将Boolean转为整型，再使用Unsafe的compareAndSwapInt进行CAS，所以原子更新char、float和double变量也可以用类似思路来实现。

2. 原子更新数组

通过原子方式更新数组里的某个元素，Atomic包提供了以下4个类：

• AtomicIntegerArray：原子更新整型数组里的元素。
• AtomicLongArray：原子更新长整型数组里的元素。
• AtomicReferenceArray：原子更新引用类型数组里的元素。

AtomicIntegerArray类主要是提供原子的方式更新数组里的整型，常用方法如下：

• int assAndGet(int i, int data)：以原子方式将输入值与数组中索引i的元素相加、
• boolean compareAndSet(int i, int expect, int update)：如果当前值等于预期值，则以原子方式将数组位置i的元素设置成update值。

同样我们只看一下AtomicIntegerArray的用法：

public class atomicArray {
    static int[] value = new int[] {1, 2};

    static AtomicIntegerArray ai = new AtomicIntegerArray(value);

    public static void main(String[] args) {
        ai.getAndSet(0, 3);
        System.out.println(ai.get(0));
        System.out.println(value[0]);
    }
}

输出结果为：

3
1

这是因为数组value通过构造方法传递进去，然后AtomicIntegerArray会将当前数组复制一份，所以当AtomicIntegerArray对内部的数组元素进行修改时，不会影响传入的数组。

3. 原子更新引用类型

原子更新引用类型主要就是可以直接替换掉一个引用所指的对象，提供了以下3个类：

• AtomicReference：原子更新引用类型
• AtomicReferenceFieldUpdater：原子更新引用类型里的字段。
• AtomicMarkableReference：原子更新带有标记位的引用类型。

我们只看一下AtomicReference的用法：

public class AtomicReferenceTest {

    public static AtomicReference<User> atomicUserRef = new AtomicReference<>();

    public static void main(String[] args) {
        User user = new User("conan", 15);
        atomicUserRef.set(user);
        User updateUser = new User("shinichi", 17);
        atomicUserRef.compareAndSet(user, updateUser);
        System.out.println(atomicUserRef.get().getName());
        System.out.println(atomicUserRef.get().getOld());
    }

    static class User {
        private String name;
        private int old;

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

        public String getName() {
            return name;
        }

        public int getOld() {
            return old;
        }
    }
}

代码输出如下：

shinichi
17

可以看出来一个AtomicReference只能对应一个对象，代码中将新建的user对象设置进AtomicReference中，最后调用compareAndSet方法进行原子更新操作，实现原理和AtomicInteger里的compareAndSet方法一样，只不过这里使用的是Unsafe的compareAndSwapObject方法，如下：

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

4. 原子更新字段类

若需要原子更新某个类里的某个字段时，则需要使用原子更新字段类，Atomic包提供了以下3个类进行原子字段更新。

• AtomicIntegerFieldUpdater：原子更新整型的字段的更新器。
• AtomicLongFieldUpdater：原子更新长整型字段的更新器。
• AtomicStampedReference：原子更新带有版本号的引用类型。该类将整数值与引用关联起来，可以用于原子的更新数据和数据的版本号，可以解决使用CAS进行原子更新时可能出现的ABA问题（查看后面补充内容）。

想要原子的更新字段类需要两步。第一步，因为原子更新字段类都是抽象类，每次使用时必须使用静态方法newUpdater()创建一个更新器，并且需要设置想要更新的类和属性。第二步，更新类的字段（属性）必须使用public volatile修饰符。

public class AtomicIntegerFieldUpdaterTest {
	//创建原子更新器，并设置需要更新的对象类和对象的属性
    private static AtomicIntegerFieldUpdater<User> a = AtomicIntegerFieldUpdater.newUpdater(User.class, "old");

    public static void main(String[] args) {
        User conan = new User("conan", 10);
        System.out.println(a.getAndIncrement(conan));
        System.out.println(a.get(conan));
    }

    static class User {
        private String name;
        public volatile int old;

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

        public String getName() {
            return name;
        }

        public int getOld() {
            return old;
        }
    }
}

•  

执行结果如下

10
11

•  

补充： ABA问题

这是CAS里面会出现的问题，因为我们知道CAS是比较如果值没有发生变化则进行更新，但是如果一个值原来是A，后来变成了，但是又变为A，那么使用CAS进行检查时会发现它并没有改变，但实际上确实改变了。那么ABA问题解决思路就是使用版本号，在变量前面追加上版本号，每次变量更新时版本号加1，哪儿吗A—>B—>A就成了A1—>B2—>A3。

所以AtomicStampedReference就是用来解决ABA问题的。这个类的compareAndSet方法作用是首先检查当前引用是否等于预期引用，并且检查当前标志是否等于预期标志，如果全部相等，则以原子方式将该引用和该标志的值设置为给定的更新值。

public boolean compareAndSet (
	V expectedReference,		//预期引用
	V newReference,				//更新后的引用
	int expectedStamp,			//预期标志
	int newStamp				//更新后的标志