原子操作类AtomicInteger详解
为什么需要AtomicInteger原子操作类?
对于全局变量的数值类型操作 num++,若没有加synchronized关键字则是线程不安全的,num++解析为num=num+1,明显,这个操作不具备原子性,多线程时必然会出现问题。测试下:
public class AtomicIntegerTest1 {
public static int count = 0;
public static void main(String[] args) {
for (int i = 0; i < 10000; i++) {
new Thread() {
public void run() {
count++;
}
}.start();
}
System.out.println("count: " + count);
}
}
输出的结果为count: 9992,这个值不定,每次测试都可能不一样,很显然,100个线程跑++操作,结果并没有像预期的那样count: 10000。
要是换成volatile修饰count变量呢?
volatile修饰的变量能够在线程间保持可见性,能被多个线程同时读但是又能保证只被单线程写,并且不会读取到过期值(由java内存模型中的happen-before原则决定的)volatile修饰字段的写入操作总是优先于读操作,即使多个线程同时修改volatile变量字段,总能保证获取到最新的值。试试:
public class AtomicIntegerTest3 {
static volatile int count = 0;
public static void main(String[] args) throws InterruptedException {
for (int i = 0; i < 100; i++) {
new Thread() {
public void run() {
for (int j = 0; j < 100; j++) {
count++;
}
}
}.start();
}
Thread.sleep(1000);
System.out.println("volatile count: " + count);
}
}
结果似乎又失望了,试了大约10次后出现volatile count: 9984,果然还是出现问题了,volatile仅仅保证变量在线程间保持可见性,却依然不能保证非原子性的操作。
Value的定义和volatile
AtomicInteger 本身是个整型,所以最重要的属性就是value,我们看看它是如何声明value的
private volatile int value;
我们看到value使用了volatile
修饰符,那么什么是volatile
呢?
volatile相当于synchronized
的弱实现,也就是说volatile
实现了类似synchronized
的语义,却又没有锁机制。它确保对volatile
字段的更新以可预见的方式告知其他的线程。
用了AtomicInteger类后会变成什么样子呢?
把上面的代码改造成AtomicInteger原子类型,看看效果:
public class AtomicIntegerTest2 {
public static AtomicInteger count = new AtomicInteger(0);
public static void main(String[] args) throws InterruptedException {
for (int i = 0; i < 100; i++) {
new Thread() {
public void run() {
for (int j = 0; j < 100; j++) {
count.getAndIncrement();
}
}
}.start();
}
Thread.sleep(1000);
System.out.println("AtomicInteger count: " + count);
}
}
结果每次都输出"AtomicInteger count: 10000",没毛病。concurrent(我这里是jdk1.7)包下提供了12种原子操作类型,如下:
原子更新基本类型
atomic包下提供了AtomicBoolean/AtomicLong/AtomicInteger三个原子更新基本类型,以AtomicInteger为例,其他两种基本类似。以下是AtomicInteger囊括的大致方法:
public final int getAndSet(int newValue) //给AtomicInteger设置newValue并返回加oldValue
public final boolean compareAndSet(int expect, int update) //如果输入的值和期望值相等就set并返回true/false
public final int getAndIncrement() //对AtomicInteger原子的加1并返回当前自增前的value
public final int getAndDecrement() //对AtomicInteger原子的减1并返回自减之前的的value
public final int getAndAdd(int delta) //对AtomicInteger原子的加上delta值并返加之前的value
public final int incrementAndGet() //对AtomicInteger原子的加1并返回加1后的值
public final int decrementAndGet() //对AtomicInteger原子的减1并返回减1后的值
public final int addAndGet(int delta) //给AtomicInteger原子的加上指定的delta值并返回加后的值
以getAndIncrement为例看下源码:
public final int getAndIncrement() {
for (;;) {
//先取出AtomicInteger的当前值
int current = get();
//对当前值加1操作
int next = current + 1;
//这里很关键,通过compareAndSet方法比较当前值有没有被其它线程修改过,若修改过返回false则再次进入compareAndSet方法判断
if (compareAndSet(current, next))
return current;
}
}
compareAndSet方法里面是调用了Unsafe类的compareAndSwapInt方法:
public final native boolean compareAndSwapObject(Object var1, long var2, Object var4, Object var5);
public final native boolean compareAndSwapInt(Object var1, long var2, int var4, int var5);
public final native boolean compareAndSwapLong(Object var1, long var2, long var4, long var6);
Unsafe是Java HotSpot提供的操作内存和线程的"后门",官方或者对于生产环境并不建议使用Unsafe类,因为它的API不稳定、不安全,错误使用将给你的HotSpot jvm带来致命性的灾难。同时对于其他基本类型,比如char、float、double等并没有对应的上述判断是否被修改方法,故可以将其转为compareAndSwapInt来简介判断,因为在AtomicBoolean的源码中就是这么做的。
相应的,concurrent包下除了提供的原子更新基本类型,还有原子更新数据、原子更新引用类型、原子更新字段类,最常用的也就是原子更新基本类型了。
用CAS操作实现安全的自增
AtomicInteger中有很多方法,例如incrementAndGet()
相当于i++
和getAndAdd()
相当于i+=n
。从源码中我们可以看出这几种方法的实现很相似,所以我们主要分析incrementAndGet()
方法的源码。
源码如下:
public final int incrementAndGet() {
for (;;) {
int current = get();
int next = current + 1;
if (compareAndSet(current, next))
return next;
}
}
public final boolean compareAndSet(int expect, int update) {
return unsafe.compareAndSwapInt(this, valueOffset, expect, update);
}
incrementAndGet()
方法实现了自增的操作。核心实现是先获取当前值和目标值(也就是value+1),如果compareAndSet(current, next)
返回成功则该方法返回目标值。那么compareAndSet是做什么的呢?理解这个方法我们需要引入CAS操作。
在大学操作系统课程中我们学过独占锁和乐观锁的概念。独占锁就是线程获取锁后其他的线程都需要挂起,直到持有独占锁的线程释放锁;乐观锁是先假定没有冲突直接进行操作,如果因为有冲突而失败就重试,直到操作成功。其中乐观锁用到的机制就是CAS,Compare and Swap。
AtomicInteger 中的CAS操作就是compareAndSet()
,其作用是每次从内存中根据内存偏移量(valueOffset
)取出数据,将取出的值跟expect 比较,如果数据一致就把内存中的值改为update。
这样使用CAS就保证了原子操作。其余几个方法的原理跟这个相同,在此不再过多的解释。
下一篇: 如何通过Nginx日志实时封禁风险IP