多线程编程最简单的例子就是计数器,如下所示
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