Java多线程进阶—— J.U.C之atomic框架：Atomic数组

逃离我推掉我的手 2023-06-21 12:30 27阅读 0赞

## 一、Atomic数组简介 ##

Atomic数组，顾名思义，就是能以原子的方式，操作数组中的元素。

JDK提供了三种类型的原子数组：`AtomicIntegerArray`、`AtomicLongArray`、`AtomicReferenceArray`。

这三种类型大同小异，AtomicIntegerArray对应AtomicInteger，AtomicLongArray对应AtomicLong，AtomicReferenceArray对应AtomicReference。

其实阅读源码也可以发现，这些数组原子类与对应的普通原子类相比，只是多了通过索引找到内存中元素地址的操作而已。

> 注意：原子数组并不是说可以让线程以原子方式一次性地操作数组中所有元素的数组。  
> 而是指对于数组中的每个元素，可以以原子方式进行操作。

说得简单点，**原子数组类型**其实可以看成**原子类型组成的数组**。

比如：

AtomicIntegerArray  array = new AtomicIntegerArray(10);
    array.getAndIncrement(0);   // 将第0个元素原子地增加1

等同于

AtomicInteger[]  array = new AtomicInteger[10];
    array[0].getAndIncrement();  // 将第0个元素原子地增加1

## 二、AtomicIntegerArray原理 ##

本节将以**AtomicIntegerArray**为例，介绍下原子数组的原理，**AtomicLongArray**和**AtomicReferenceArray**的使用和源码与**AtomicIntegerArray**大同小异，读者可以自己查看[Oracle官方文档][Oracle]和源码。

**AtomicIntegerArray**其实和其它原子类区别并不大，只不过构造的时候传入的是一个**int\[\]**数组，然后底层通过**Unsafe**类操作数组：

![clipboard.png][]

可以看到，**AtomicIntegerArray**提供了两种构造器，本质都是内部利用**array**变量保存一个**int\[\]**数组引用。

另外，**AtomicIntegerArray**利用**Unsafe**类直接操作**int\[\]**对象的内存地址，以达到操作数组元素的目的，几个关键的变量解释如下：

`int base = unsafe.arrayBaseOffset(int[].class);`  
Unsafe类的 arrayBaseOffset 方法：返回指定类型数组的第一个元素地址相对于数组起始地址的偏移值。

`int scale = unsafe.arrayIndexScale(int[].class);`  
Unsafe类的**arrayIndexScale**方法：返回指定类型数组的元素所占用的字节数。比如int\[\]数组中的每个int元素占用4个字节，就返回4。

那么，通过`base + i * sacle` 其实就可以知道 索引**i**的元素在数组中的内存起始地址。  
但是，观察**AtomicIntegerArray**的**byteOffset**方法，是通过`i << shift + base` 的公式计算元素的起始地址的：

![20191212213917403.png][]

这里，

![20191212213952705.png][]

其实就等于**scale**。

`shift = 31 - Integer.numberOfLeadingZeros(scale)`，`Integer.numberOfLeadingZeros(scale)`是将**scale**转换为2进制，然后从左往右数连续0的个数。

读者可以自己计算下：  
`shift = 31 - Integer.numberOfLeadingZeros(4) = 31 - 29 =2`

之所以要这么绕一圈，其实是处于性能的考虑，通过移位计算乘法的效率往往更高。

## 三、AtomicIntegerArray接口声明 ##

![clipboard.png][clipboard.png 1]

![clipboard.png][clipboard.png 2]

[Oracle]: https://docs.oracle.com/javase/8/docs/api/
[clipboard.png]: https://imgconvert.csdnimg.cn/aHR0cHM6Ly9zZWdtZW50ZmF1bHQuY29tL2ltZy9iVmJlRkN1?x-oss-process=image/format,png
[20191212213917403.png]: https://img-blog.csdnimg.cn/20191212213917403.png
[20191212213952705.png]: https://img-blog.csdnimg.cn/20191212213952705.png
[clipboard.png 1]: https://imgconvert.csdnimg.cn/aHR0cHM6Ly9zZWdtZW50ZmF1bHQuY29tL2ltZy9iVmJlRkVV?x-oss-process=image/format,png
[clipboard.png 2]: https://imgconvert.csdnimg.cn/aHR0cHM6Ly9zZWdtZW50ZmF1bHQuY29tL2ltZy9iVmJlRkVX?x-oss-process=image/format,png