简介:AtomicInteger是java并发包下面提供的原子类,主要操作的是int类型的整型,通过调用底层Unsafe的CAS等方法实现原子操作。
ps:
volatile无法解决多线程下count++的线程安全问题,因为volatile仅有两个作用:
(1)保证可见性,即一个线程对变量的修改另一个线程立即可见;
(2)禁止指令重排序;
这里有个很重要的问题,count++实际上是两步操作,第一步是获取count的值,第二步是对它的值加1。
使用volatile是无法保证这两步不被其它线程调度打断的,所以无法保证原子性。
这就引出了我们今天讲的AtomicInteger,它的自增调用的是Unsafe的CAS并使用自旋保证一定会成功,它可以保证两步操作的原子性。
下面开始对AtomicInteger的源码进行简单分析:
1.重要属性:
// 获取Unsafe的实例
private static final Unsafe unsafe = Unsafe.getUnsafe();
// 标识value字段的偏移量
private static final long valueOffset;
// 静态代码块,通过unsafe获取value的偏移量
static {
try {
valueOffset = unsafe.objectFieldOffset
(AtomicInteger.class.getDeclaredField("value"));
} catch (Exception ex) { throw new Error(ex); }
}
// 存储int类型值的地方,使用volatile修饰
private volatile int value;
(1)使用int类型的value存储值,且使用volatile修饰,volatile主要是保证可见性,即一个线程修改对另一个线程立即可见,主要的实现原理是内存屏障,这里不展开来讲,有兴趣的可以自行查阅相关资料。
(2)调用Unsafe的objectFieldOffset()方法获取value字段在类中的偏移量,用于后面CAS操作时使用。
2.重要方法:
(1)compareAndSet()方法:比较输入的预期值是否和内存中的值相同,相同则更新为新值后返回true,更新失败返回false;
public final boolean compareAndSet(int expect, int update) {
return unsafe.compareAndSwapInt(this, valueOffset, expect, update);
}
// Unsafe中的方法
public final native boolean compareAndSwapInt(Object var1, long var2, int var4, int var5);
调用Unsafe.compareAndSwapInt()方法实现,这个方法有四个参数:
操作的对象;var1
对象中字段的偏移量;var2
原来的值,即期望的值;var4
要修改的值;var5
可以看到,这是一个native方法,底层是使用C/C++写的,主要是调用CPU的CAS指令来实现,它能够保证只有当对应偏移量处的字段值是期望值时才更新,即类似下面这样的两步操作:
if(value == expect) {
value = newValue;
}
通过CPU的CAS指令可以保证这两步操作是一个整体,也就不会出现多线程环境中可能比较的时候value值是a,而到真正赋值的时候value值可能已经变成b了的问题。
(2)
public final int getAndIncrement() {
return unsafe.getAndAddInt(this, valueOffset, 1);
}
// Unsafe中的方法
public final int getAndAddInt(Object var1, long var2, int var4) {
int var5;
do {
var5 = this.getIntVolatile(var1, var2);
} while(!this.compareAndSwapInt(var1, var2, var5, var5 + var4));
return var5;
}
//先根据var1(当前对象值)和var2(对象字段的偏移量)获取主内存中最新的值var5,
//如果var1和var5相等则更新var4+var5返回,否则自旋。直到var1和var5相等。
getAndIncrement()方法底层是调用的Unsafe的getAndAddInt()方法,这个方法有三个参数:
操作的对象;var1
对象中字段的偏移量;var2
要增加的值;var4
查看Unsafe的getAndAddInt()方法的源码,可以看到它是先获取当前的值,然后再调用compareAndSwapInt()尝试更新对应偏移量处的值,如果成功了就跳出循环,如果不成功就再重新尝试,直到成功为止,这可不就是(CAS+自旋)的乐观锁机制么^^
AtomicInteger中的其它方法几乎都是类似的,最终会调用到Unsafe的compareAndSwapInt()来保证对value值更新的原子性。
总结
(1)AtomicInteger中维护了一个使用volatile修饰的变量value,保证可见性;
(2)AtomicInteger中的主要方法最终几乎都会调用到Unsafe的compareAndSwapInt()方法保证对变量修改的原子性。
参考:https://www.jb51.net/article/161882.htm