JUC源码系列之AtomicInteger源码解析

多线程编程最简单的例子就是计数器，如下所示

public class Test {

    private static int i = 0;

    public static void main(String[] args) {
        IntStream.range(0, 10000).forEach(v -> new Thread(Test::add).start());
    }
    private static void add() {
        i++;
    }
}

学习了最基础的多线程编程都应该知道，这个计数器的最终结果是不确定的，因为这个程序并没有正确同步。

怎么样才能正确同步？

• 将 add 方法声明为 synchronized 方法，这个终极大招，肯定是可以解决同步问题，但这样会对 Test 类加锁，效率不太高。
• 将变量 i 声明为 volatile 类型，volatile 效率上会比 synchronized 稍高，但这个场景下并不适用，因为 i++ 并不是一个原子操作。

那么还有没有一种不用加锁，效率高，又能正确同步的解法呢？

JDK 提供了 AtomicInteger 可以很简单实现正确同步，代码实现如下所示，

public class Test {

    private static AtomicInteger i = new AtomicInteger(0);

    public static void main(String[] args) {
        IntStream.range(0, 10000).forEach(v -> new Thread(Test::add).start());
    }
    private static void add() {
        i.incrementAndGet();
    }
}

源码分析

进入 AtomicInteger 类源码，首先查看类定义及变量定义。

public class AtomicInteger extends Number implements java.io.Serializable {
    // 实现
    private static final Unsafe unsafe = Unsafe.getUnsafe();
    private volatile int value;
    //...
}

可以看到一个重要的变量 value，这个变量用于存储使用的数值，该变量被声明为 volatile 类型，意味着对该变量的操作都是线程安全的。get / set 方法是对变量 value 的简单赋值，所以 volatile 的特性保证了该方法的线程安全性。

public final int get() {
    return value;
}

public final void set(int newValue) {
    value = newValue;
}

AtomicInteger 还可以对数值进行计算，计算都是基于原数值进行计算，比如自增，自减方法，这种操作正好是 volatile 所不支持的操作

public final int getAndIncrement() {
    return unsafe.getAndAddInt(this, valueOffset, 1);
}

public final int getAndDecrement() {
    return unsafe.getAndAddInt(this, valueOffset, -1);
}

可以看到 AtomicInteger 使用了 Unsafe 类的 getAndAddInt 方法进行处理，而 getAndAddInt 方法又是通过 compareAndSwapInt 方法进行计算，而最终的 compareAndSwapInt 方法是一个 native 方法，这个方法由C实现，通过cas机器指令保证线程安全。

public final int getAndAddInt(Object obj, long offset, int var4) {
    int value;
    do {
        value = this.getIntVolatile(obj, offset);
    } while(!this.compareAndSwapInt(obj, offset, value, value + var4));
    return value;
}

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

结束语

AtomicInteger 实现线程同步的原理还是比较简单的。

• 单步操作：利用 volatile 特性保证线程安全。
• 多步操作：利用 Unsafe 类的 compareAndSwapInt 方法保证线程安全。

---

Atomic简介

Java并发包下提供了一系列线程安全的原子操作类，这些类名都以 Atomic 开头，统一存放在 java.util.concurrent.atomic 包路径下，主要包括以下类：

• 标量类：AtomicBoolean，AtomicInteger，AtomicLong，AtomicReference
• 数组类：AtomicIntegerArray，AtomicLongArray，AtomicReferenceArray
• 更新器类：AtomicLongFieldUpdater，AtomicIntegerFieldUpdater，AtomicReferenceFieldUpdater
• 复合变量类：AtomicMarkableReference，AtomicStampedReference