阻塞队列、原子操作的原理分析

生产者消费者的实际使用

我相信大家都有使用过分布式消息队列，比如 ActiveMQ、kafka、RabbitMQ 等等，消息队列的是有可以使得程序之间实现解耦，提升程序响应的效率。

如果我们把多线程环境比作是分布式的话，那么线程与线程之间是不是也可以使用这种消息队列的方式进行数据通信和解耦呢？

阻塞队列的使用案例

注册成功后增加积分

假如我们模拟一个场景，就是用户注册的时候，在注册成功以后发放积分。这个场景在一般来说，我们会这么去实现

但是实际上，我们需要考虑两个问题

1.性能，在注册这个环节里面，假如添加用户需要花费 1 秒钟，增加积分需要花费 1 秒钟，那么整个注册结果的返

回就可能需要大于 2 秒，虽然影响不是很大，但是在量比较大的时候，我们也需要做一些优化

2.耦合，添加用户和增加积分，可以认为是两个领域，也就是说，增加积分并不是注册必须要具备的功能，但是一旦增加积分这个逻辑出现异常，就会导致注册失败。这种耦合在程序设计的时候是一定要规避的因此我们可以通过异步的方式来实现改造之前的代码逻辑

public class UserService {
public boolean register(){

User user=new User();

user.setName(“Mic”);

addUser(user);

sendPoints(user);

return true;

}

public static void main(String[] args) { new UserService().register();

}

private void addUser(User user){ System.out.println(“添加用户：”+user); try {

Thread.sleep(1000);

} catch (InterruptedException e) {

e.printStackTrace();

}

}

private void sendPoints(User user){ System.out.println(" 发 送 积 分 给 指 定 用户:"+user);

try {

Thread.sleep(1000);

} catch (InterruptedException e) { e.printStackTrace();

}

}
}

改造之后的逻辑

public class UserService {

private final ExecutorService single =

Executors.newSingleThreadExecutor();

private volatile boolean isRunning = true;

ArrayBlockingQueue arrayBlockingQueue=new ArrayBlockingQueue(10);
{
init();
}

public void init(){

single.execute(()->{

while(isRunning){

try {

User

user=(User)arrayBlockingQueue.take();//阻塞的方式获取队列中的数据

sendPoints(user);

} catch (InterruptedException e)

{

e.printStackTrace();

}

}

});

}

public boolean register(){

User user=new User();

user.setName(“Mic”);

addUser(user);

arrayBlockingQueue.add(user);//添加到异步队列

return true;

}

public static void main(String[] args) { new UserService().register();

}

private void addUser(User user){ System.out.println(“添加用户：”+user); try {

Thread.sleep(1000);

} catch (InterruptedException e) { e.printStackTrace();

}

}

private void sendPoints(User user){ System.out.println(" 发 送 积 分 给 指 定 用户:"+user);

try {

Thread.sleep(1000);

} catch (InterruptedException e) { e.printStackTrace();

}

}
}

优化以后，整个流程就变成了这样

在这个案例中，我们使用了 ArrayBlockingQueue 基于数组的阻塞队列，来优化代码的执行逻辑。

阻塞队列的应用场景

阻塞队列这块的应用场景，比较多的仍然是对于生产者消费者场景的应用，但是由于分布式架构的普及，是的大家更多的关注在分布式消息队列上。所以其实如果把阻塞队列比作成分布式消息队列的话，那么所谓的生产者和消费者其实就是基于阻塞队列的解耦。

另外，阻塞队列是一个 fifo 的队列，所以对于希望在线程级别需要实现对目标服务的顺序访问的场景中，也可以使用

J.U.C 中的阻塞队列

J.U.C 提供的阻塞队列

在 Java8 中，提供了 7 个阻塞队列

阻塞队列的操作方法

在阻塞队列中，提供了四种处理方式

1.插入操作

add(e) ：添加元素到队列中，如果队列满了，继续插入元素会报错，IllegalStateException。

offer(e) : 添加元素到队列，同时会返回元素是否插入成功的状态，如果成功则返回 true

put(e) ：当阻塞队列满了以后，生产者继续通过 put 添加元素，队列会一直阻塞生产者线程，知道队列可用 offer(e,time,unit) ：当阻塞队列满了以后继续添加元素，生产者线程会被阻塞指定时间，如果超时，则线程直接退出

2.移除操作

remove()：当队列为空时，调用 remove 会返回 false，

如果元素移除成功，则返回 true

poll(): 当队列中存在元素，则从队列中取出一个元素，如果队列为空，则直接返回 null take()：基于阻塞的方式获取队列中的元素，如果队列为空，则 take 方法会一直阻塞，直到队列中有新的数据可以消费poll(time,unit)：带超时机制的获取数据，如果队列为空，则会等待指定的时间再去获取元素返回

ArrayBlockingQueue 原理分析
构造方法
ArrayBlockingQueue 提供了三个构造方法，分别如下。capacity： 表示数组的长度，也就是队列的长度

fair：表示是否为公平的阻塞队列，默认情况下构造的是非公平的阻塞队列。

其中第三个构造方法就不解释了，它提供了接收一个几个作为数据初始化的方法

public ArrayBlockingQueue(int capacity) { this(capacity, false);

}

public ArrayBlockingQueue(int capacity, boolean fair) {

if (capacity <= 0)

throw new IllegalArgumentException(); this.items = new Object[capacity];
lock = new ReentrantLock(fair); //重入锁，出

队和入队持有这一把锁

notEmpty = lock.newCondition(); //初始化非空

等待队列

notFull = lock.newCondition(); //初始化非满

等待队列

}

关于锁的用途，大家在没有看接下来的源码之前，可以先思考一下他的作用。

items 构造以后，大概是一个这样的数组结构

Add 方法

以 add 方法作为入口，在 add 方法中会调用父类的 add 方法，也就是 AbstractQueue.如果看源码看得比较多的话，一般这种写法都是调用父类的模版方法来解决通用性问题

public boolean add(E e) {

return super.add(e);

}

从父类的 add 方法可以看到，这里做了一个队列是否满了的判

断，如果队列满了直接抛出一个异常

public boolean add(E e) {

if (offer(e))

return true;

else

throw new IllegalStateException("Queue

full");

}

offer 方法

add 方法最终还是调用 offer 方法来添加数据，返回一个添

加成功或者失败的布尔值反馈

这段代码做了几个事情

1.判断添加的数据是否为空

2.添加重入锁

3.判断队列长度，如果队列长度等于数组长度，表示满了直接返回 false

4.否则，直接调用 enqueue 将元素添加到队列中

public boolean offer(E e) {

checkNotNull(e); //对请求数据做判断 final ReentrantLock lock = this.lock; lock.lock();

try {

if (count == items.length)

return false;

else {

enqueue(e);

return true;

}

} finally { lock.unlock();

}

}

enqueue

这个是最核心的逻辑，方法内部通过 putIndex 索引直接将元素添加到数组 items

private void enqueue(E x) {

//assert lock.getHoldCount() == 1;

//assert items[putIndex] == null;

final Object[] items = this.items;

items[putIndex] = x; //通过 putIndex 对数据赋

值

if (++putIndex == items.length) // 当 putIndex 等于数组长度时，将 putIndex 重置为 0

putIndex = 0;

count++;//记录队列元素的个数

notEmpty.signal();//唤醒处于等待状态下的线程，表

示当前队列中的元素不为空,如果存在消费者线程阻塞，就可以

开始取出元素

}

这里大家肯定会有一个疑问，putIndex 为什么会在等于数组长度的时候重新设置为 0。

因为 ArrayBlockingQueue 是一个 FIFO 的队列，队列添加元素时，是从队尾获取 putIndex 来存储元素，当 putIndex 等于数组长度时，下次就需要从数组头部开始添加了。

下面这个图模拟了添加到不同长度的元素时，putIndex 的变化，当 putIndex 等于数组长度时，不可能让 putIndex 继续累加，否则会超出数组初始化的容量大小。同时大家还需要思考两个问题

1.当元素满了以后是无法继续添加的，因为会报错

2.其次，队列中的元素肯定会有一个消费者线程通过 take 或者其他方法来获取数据，而获取数据的同时元素也会从队列中移除

put 方法

put 方法和 add 方法功能一样，差异是 put 方法如果队列满了，会阻塞。这个在最开始的时候说过。接下来看一下它的实现逻辑

public void put(E e) throws

InterruptedException {

checkNotNull(e);

final ReentrantLock lock = this.lock;

lock.lockInterruptibly(); //这个也是获得锁，但

是和 lock 的区别是，这个方法优先允许在等待时由其他线程调

用等待线程的 interrupt 方法来中断等待直接返回。而 lock

方法是尝试获得锁成功后才响应中断

try {

while (count == items.length)

notFull.await();//队列满了的情况下，当前

线程将会被 notFull 条件对象挂起加到等待队列中

enqueue(e);

} finally { lock.unlock();

}

}

take 方法

take 方法是一种阻塞获取队列中元素的方法

它的实现原理很简单，有就删除没有就阻塞，注意这个阻塞是可以中断的，如果队列没有数据那么就加入 notEmpty 条件队列等待(有数据就直接取走，方法结束)，如果有新的 put 线程添加了数据，那么 put 操作将会唤醒 take 线程，执行 take 操作。

public E take() throws InterruptedException { final ReentrantLock lock = this.lock;

lock.lockInterruptibly();

try {

while (count == 0)

notEmpty.await(); //如果队列为空的情况

下，直接通过 await 方法阻塞

return dequeue();

} finally { lock.unlock();

}

}

如果队列中添加了元素，那么这个时候，会在 enqueue 中调用 notempty.signal 唤醒 take 线程来获得元素

dequeue 方法

这个是出队列的方法，主要是删除队列头部的元素并发返回给客户端

takeIndex，是用来记录拿数据的索引值

private E dequeue() {

//assert lock.getHoldCount() == 1;

//assert items[takeIndex] != null;

final Object[] items = this.items; @SuppressWarnings(“unchecked”)
Ex = (E) items[takeIndex]; //默认获取 0 位置

的元素

items[takeIndex] = null;//将该位置的元素设置为

空

if (++takeIndex == items.length)//这里的作用也是一样，如果拿到数组的最大值，那么重置为 0，继续从头部

位置开始获取数据

takeIndex = 0;

count–;//记录 元素个数递减

if (itrs != null)

itrs.elementDequeued();//同时更新迭代器中

的元素数据

notFull.signal();//触发 因为队列满了以后导致的

被阻塞的线程

return x;

}

itrs.elementDequeued();

ArrayBlockingQueue 中，实现了迭代器的功能，也就是可以通过迭代器来遍历阻塞队列中的元素

所以 itrs.elementDequeued() 是用来更新迭代器中的元素数据的

takeIndex 的索引变化图如下，同时随着数据的移除，会唤醒处于 put 阻塞状态下的线程来继续添加数据

remove 方法

remove 方法是移除一个指定元素。看看它的实现代码

public boolean remove(Object o) {

if (o == null) return false;

final Object[] items = this.items; //获取数

组元素

final ReentrantLock lock = this.lock; lock.lock(); //获得锁

try {

if (count > 0) { //如果队列不为空

final int putIndex = this.putIndex;

//获取下一个要添加元素时的索引

int i = takeIndex;//获取当前要被移除的

元素的索引

do {

if (o.equals(items[i])) {//从

takeIndex 下标开始，找到要被删除的元素

removeAt(i);//移除指定元素

return true;//返回执行结果

}

//当前删除索引执行加 1 后判断是否与数组

长度相等

//若为 true，说明索引已到数组尽头，将 i 设置为 0

if (++i == items.length)

i = 0;

} while (i != putIndex);//继续查找，直

到找到最后一个元素

}

return false;

} finally { lock.unlock();

}

}

原子操作类

原子性这个概念，在多线程编程里是一个老生常谈的问题。所谓的原子性表示一个或者多个操作，要么全部执行完，要么一个也不执行。不能出现成功一部分失败一部分的情况。
在多线程中，如果多个线程同时更新一个共享变量，可能会得到一个意料之外的值。比如 i=1 。A 线程更新 i+1 、 B 线程也更新 i+1。
通过两个线程并行操作之后可能 i 的值不等于 3。而可能等于 2。因为 A 和 B 在更新变量 i 的时候拿到的 i 可能都是 1 这就是一个典型的原子性问题

前面我们讲过，多线程里面，要实现原子性，有几种方法，其中一种就是加 Synchronized 同步锁。

而从 JDK1.5 开始，在 J.U.C 包中提供了 Atomic 包，提供了对于常用数据结构的原子操作。它提供了简单、高效、以及线程安全的更新一个变量的方式

J.U.C 中的原子操作类

由于变量类型的关系，在 J.U.C 中提供了 12 个原子操作的类。这 12 个类可以分为四大类

1.原子更新基本类型

AtomicBoolean、AtomicInteger、AtomicLong

2.原子更新数组

AtomicIntegerArray 、 AtomicLongArray 、 AtomicReferenceArray

3.原子更新引用

AtomicReference 、 AtomicReferenceFieldUpdater 、AtomicMarkableReference（更新带有标记位的引用类型）

4.原子更新字段

AtomicIntegerFieldUpdater、AtomicLongFieldUpdater、 AtomicStampedReference

AtomicInteger 原理分析

接下来，我们来剖析一下 AtomicInteger 的实现原理，仍然是基于我们刚刚在前面的案例中使用到的方法作为突破口

getAndIncrement

getAndIncrement 实际上是调用 unsafe 这个类里面提供的方法，

Unsafe 类我们前面在分析 AQS 的时候讲过，这个类相当于是一个后门，使得 Java 可以像 C 语言的指针一样直接操作内存空间。当然也会带来一些弊端，就是指针的问题。实际上这个类在很多方面都有使用，除了 J.U.C 这个包以外，还有 Netty、kafka 等等

这个类提供了很多功能，包括多线程同步(monitorEnter)、

CAS 操 作 (compareAndSwap) 、 线 程 的 挂 起 和 恢 复(park/unpark)、内存屏障(loadFence/storeFence)

内存管理（内存分配、释放内存、获取内存地址等.）

public final int getAndIncrement() {

return unsafe.getAndAddInt(this,

valueOffset, 1);

}

valueOffset，也比较熟了。通过 unsafe.objectFieldOffset() 获取当前 Value 这个变量在内存中的偏移量，后续会基于这个偏移量从内存中得到 value 的值来和当前的值做比较，实现乐观锁

private static final long valueOffset;
static {
try {
valueOffset = unsafe.objectFieldOffset(AtomicInteger.class.getDeclaredField(“value”) );
} catch (Exception ex) { throw new Error(ex); }

}
getAndAddInt

通过 do/while 循环，基于 CAS 乐观锁来做原子递增。实际上前面的 valueOffset 的作用就是从主内存中获得当前 value 的值和预期值做一个比较，如果相等，对 value 做递增并结束循环

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;

}

get 方法
get 方法只需要直接返回 value 的值就行，这里的 value 是

通过 Volatile 修饰的，用来保证可见性

public final int get() {

return value;

}

其他方法

AtomicInteger 的实现非常简单，所以我们可以很快就分析完它的实现原理，当然除了刚刚分析的这两个方法之外，还有其他的一些

比 如 它提 供了 compareAndSet ，允许客 户端 基于 AtomicInteger 来实现乐观锁的操作

public final boolean compareAndSet(int expect, int update) {

return unsafe.compareAndSwapInt(this, valueOffset, expect, update); }