高并发学习之08原子操作类

1. 简介

在前面文章synchronized中，我们介绍了synchonized 轻量级锁时，谈到其是CAS机制实现的，本文我们将讨论JDK为我们提供的13种原子操作类，他们的实现原理都是CAS机制。13种原子类在java.util.concurrent.atomic包中。

2. 原子更新基本类型类

2.1.原子更新基本类型类

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

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

以上3个类提供的方法几乎一模一样，所以本节仅以AtomicInteger为例进行讲解，AtomicInteger的常用方法如下：

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

import java.util.concurrent.atomic.AtomicInteger;
public class AtomicIntegerTest {
	static AtomicInteger ai = new AtomicInteger(1);
	public static void main(String[] args) {
		System.out.println(ai.getAndIncrement());
		System.out.println(ai.get());
	}
}

2.1.2 原子类实现原理

为了搞懂其原子类操作的原理，下面以AtomicInteger中getAndSet()方法为了例：

private static final Unsafe unsafe = Unsafe.getUnsafe();
public final int getAndSet(int newValue) {
        return unsafe.getAndSetInt(this, valueOffset, newValue);
    }

从上面的源码可以看出，其调用unsafe类中getAndSetInt方法。unsafe是sun.misc包下的工具类，原子类中操作都是调用unsafe中的方法。其源码为：

public native int getIntVolatile(Object var1, long var2);

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

 public final int getAndSetInt(Object var1, long var2, int var4) {
        int var5;
        do {
            var5 = this.getIntVolatile(var1, var2);
        } while(!this.compareAndSwapInt(var1, var2, var5, var4));

        return var5;
    }

使用do。。while方式判断，获取到的当前值与内存中当前值是否一致，如果一致则更新，不一致在重新获取。注意native 修饰的方法都是JVM底层方法，直接调用c++函数。关于native解释可以看下这篇文章native解释。
通过代码，我们发现Unsafe只提供了3种CAS方法：compareAndSwapObject、compare-AndSwapInt和compareAndSwapLong，再看AtomicBoolean源码，发现它是先把Boolean转换成整型，再使用compareAndSwapInt进行CAS，所以原子更新char、float和double变量也可以用类似的思路来实现。

3. 原子更新数组

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

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

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

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

以上几个类提供的方法几乎一样，举个例子：

public class AtomicIntegerArrayTest {
	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对内部的数组元素进行修改时，不会影响传入的数组。

4. 原子更新引用类型

原子更新基本类型的AtomicInteger，只能更新一个变量，如果要原子更新多个变量，就需要使用这个原子更新引用类型提供的类。Atomic包提供了以下3个类：

• AtomicReference：原子更新引用类型。
• AtomicReferenceFieldUpdater：原子更新引用类型里的字段。
• AtomicMarkableReference：原子更新带有标记位的引用类型。可以原子更新一个布尔类型的标记位和引用类型。构造方法是 public AtomicMarkableReference(V initialRef, boolean initialMark) ；
  以上几个类提供的方法几乎一样，所以本节仅以AtomicReference为例进行讲解，AtomicReference的使用示例代码如下所示。

public class AtomicReferenceTest {
	public static AtomicReference<User> atomicUserRef = new AtomicReference<User>();
    public static void main(String[] args) {
        User user = new User("herman",15);
        atomicUserRef.set(user);
        User updateUser = new User("ocan",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;
        }
    }
}
输出结果如下:
ocan
17

代码中首先构建一个user对象，然后把user对象设置进AtomicReferenc中，最后调用compareAndSet方法进行原子更新操作，实现原理同AtomicInteger里的compareAndSet方法。

5. 原子更新字段类

如果需原子地更新某个类里的某个字段时，就需要使用原子更新字段类，Atomic包提供了以下3个类进行原子字段更新:

• AtomicIntegerFieldUpdater：原子更新整型的字段的更新器。
• AtomicLongFieldUpdater：原子更新长整型字段的更新器。
• AtomicStampedReference：原子更新带有版本号的引用类型。

注意AtomicStampedReference将整数值与引用关联起来，可用于原子的更新数据和数据的版本号，可以解决使用CAS进行原子更新时可能出现的ABA问题。

要想原子地更新字段类需要两步：

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

上面3个类提供的方法几乎一样，所以仅以AstomicIntegerFieldUpdater为例进行讲解，

public class AtomicIntegerFieldUpdaterTest {
	// 创建原子更新器,并设置需要更新的对象类和对象的属性
    private static AtomicIntegerFieldUpdater<User> a = AtomicIntegerFieldUpdater.
            newUpdater(User.class, "age");
    public static void main(String[] args) {
        // 设置ocan age为10
        User conan = new User("ocan", 10);
        // age原子自增，输出上一次值
        System.out.println(a.getAndIncrement(conan));
        // 输出自增后的age
        System.out.println(a.get(conan));
    }
    public static class User {
        private String name;
        public volatile int age;
        public User(String name, int age) {
            this.name = name;
            this.age= age;
        }
        public String getName() {
            return name;
        }
        public int getAge() {
            return age;
        }
    }
}
代码执行后输出如下。
10
11