标签:check 必须 value bool https ror 内存布局 length start
JDK官方提供了3个原子数组,它们提供了原子更新数组中元素的能力,它们主要借助Unsafe类实现其核心功能。
AtomicIntegerArray:原子更新整型数组里的元素
AtomicLongArray:原子更新长整型数组里的元素。
AtomicReferenceArray:原子更新引用类型数组里的元素。
在分析原子数组之前,我们先来了解Java对象的内存布局,Java对象由对象头和实例数据两部分组成。
下图中MarkWord包含对象的hashCode、锁信息、垃圾回收的分代信息等,占32/64位;Class Metadata Pointer表示一个此对象数据类型的Class对象(虚拟机中的Klass对象)的指针,占32/64位;ArrayLength是数组对象特有的内容,表示数组的长度,占32位。数组对象的实例数据是各个元素的值或引用,普通对象的实例数据是各实例字段的值或引用。另外为了快速内存分配、快速内存寻址、提高性能,Java语言规范要求Java对象要做内存对齐处理,每个对象占用的内存字节数必须是8的倍数,若不是则要填零补足对齐。
从下图可以看出,字段与对象头之间的偏移量是固定的,只要知道字段的相对偏移量和对象起始地址,我们就能获取此字段的绝对内存地址(fieldAddress=objAddress+fieldOffset),根据此绝对内存地址,我们就能忽略访问修饰符的限制而可直接读取/修改此字段的值或引用。
数组对象的元素内存定址,相对对于普通对象的字段定址有些不一样,它要先计算出对象头的长度,作为基础偏移量;由于数组元素的数据类型是相同的,每个元素的值或引用所占内存空间是相同的,因此将元素值或引用或占内存作为每两相邻元素的相对偏移量。根据对象起始位置、基础偏移量、相邻元素相对偏移量及数组下标,就可以获取到某个元素值或引用的绝对内存地址(itemAddress=arrayAddress+baseOffset+index*indexOffset),进而通过绝对内存地址读取或修改此元素的值或引用。
这里以JDK1.8的AtomicIntegerArray为例,对原子数组做出简单的实现原理分析。
////Unsafe是所有并发工具的基础工具类
private static final Unsafe unsafe = Unsafe.getUnsafe();
////数组对象头到数组首元素间的地址偏移量
private static final int base = unsafe.arrayBaseOffset(int[].class);
//数组中相邻元素的地址偏移量的位移表示形式(若shift=x,那么相邻元素的偏移量就是"1<<shift")
private static final int shift;
//实际存放元素的数组
private final int[] array;
静态块主要是对静态常量shift初始化。
static {
int scale = unsafe.arrayIndexScale(int[].class);//表示相邻元素的地址偏移量
if ((scale & (scale - 1)) != 0)
throw new Error("data type scale not a power of two");//非法的,scale一定是2的幂次方
shift = 31 - Integer.numberOfLeadingZeros(scale);
}
scale表示相邻元素的地址偏移量,因为内存对齐的原因,scale一定是2n。而shift是相邻元素的地址偏移量的位移表示形式,Integer.numberOfLeadingZeros(scale)
取scale进制前导零的个数,所以shift是scale二进制有效位数减1,即有效的0
的个数。
如scale=16,即scale=0b00000000_00000000_00000000_00010000 , 那么shift=4。
构造方法
AtomicIntegerArray(int)
创建指定长度的原子数组,AtomicIntegerArray(int)
利用克隆(这里是深度克隆,源数组和原子数组互不影响)将指定的普通数组包装成原子数组.
public AtomicIntegerArray(int length) {
array = new int[length];
}
public AtomicIntegerArray(int[] array) {
// Visibility guaranteed by final field guarantees
this.array = array.clone();
}
获取指定下标元素get(int)
get()方法主要利用byteOffset根据公式offset=base+i*scale求出数组对象的起始位置下标元素的偏移量offset,然后利用unsafe.getIntVolatile根据公式indexAddr=arrayAddr+offset求出出指定偏移位置的元素值
public final int get(int i) {
return getRaw(checkedByteOffset(i));
}
private long checkedByteOffset(int i) {//下标检查,并计算数组对象起始位置至此下标元素的偏移量
if (i < 0 || i >= array.length)
throw new IndexOutOfBoundsException("index " + i);
return byteOffset(i);
}
private static long byteOffset(int i) {//计算偏移量
return ((long) i << shift) + base; //i*2^shift +base 即,"i*元素间距+对象头长度"
}
private int getRaw(long offset) {
return unsafe.getIntVolatile(array, offset);//利用unsafe方法求出指定偏移位置的int值
}
更新指定下标元素set(int,int)
set()方法与get方法类似,都要先调用checkedByteOffset计算偏移量offset,不过此set方法要调用putIntVolatile在指定内存位置修改值而已。
public final void set(int i, int newValue) {
unsafe.putIntVolatile(array, checkedByteOffset(i), newValue);
}
getAndSet(int,int)
先获取值再设置新值,同样先调用checkedByteOffset计算偏移量offset,然后getAndSetInt进行CAS自旋更新指定的元素。
public final int getAndSet(int i, int newValue) {
return unsafe.getAndSetInt(array, checkedByteOffset(i), newValue);
}
public final int getAndSetInt(Object o, long offset, int newValue) {//unsafe的方法
int v;
do {
v = getIntVolatile(o, offset);
} while (!compareAndSwapInt(o, offset, v, newValue));
return v;
}
compareAndSet()CAS更新期望的元素,先调用checkedByteOffset计算偏移量offset,然后调用compareAndSwapInt尝试CAS更新元素。
public final boolean compareAndSet(int i, int expect, int update) {
return compareAndSetRaw(checkedByteOffset(i), expect, update);
}
private boolean compareAndSetRaw(long offset, int expect, int update) {
return unsafe.compareAndSwapInt(array, offset, expect, update);
}
class AtomicIntArrayDemo {
public static void main(String[] args) {
AtomicIntegerArray array = new AtomicIntegerArray(new int[]{45,23,13,47,12,42});
for (int i = 0; i <array.length() ; i++) {
final int j=i;
new Thread(()->{
array.compareAndSet(j, 13, 31);
array.getAndAdd(j, 2);
array.decrementAndGet(j);
array.getAndSet(j, array.get(j) - 1);
}).start();
}
System.out.println(array);
}
}
标签:check 必须 value bool https ror 内存布局 length start
原文地址:https://www.cnblogs.com/gocode/p/analysis-source-code-of-AtomicIntegerArray.html