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

生产者消费者的实际使用
我相信大家都有使用过分布式消息队列,比如 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 个阻塞队列
ArrayBlockingQueue: 数组实现的有界阻塞队列, 此队列按照先进先出(FIFO)的原则 对元素进行排序。
LinkedBlockingQueue: 链表实现的有界阻塞队列, 此队列的默认和最大长度为 Integer.MAX_VALUE。此队列按照先进先出的原则对元素进行 排序
PriorityBlockingQueue: 支持优先级排序的无界阻塞队列, 默认情况下元素采取自然顺序 升序排列。也可以自定义类实现 compareTo()方法来指定元素 排序规则,或者初始化 PriorityBlockingQueue 时,指定构造 参数    Comparator 来对元素进行排序。
DelayQueue: 优先级队列实现的无界阻塞队列
SynchronousQueue: 不存储元素的阻塞队列, 每一个 put 操作必须等待一个 take 操 作,否则不能继续添加元素。
LinkedTransferQueue: 链表实现的无界阻塞队列
LinkedBlockingDeque: 链表实现的双向阻塞队列
阻塞队列的操作方法
在阻塞队列中,提供了四种处理方式
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("Queuefull");
}
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")
  E x = (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);
}

 

转载于:https://www.cnblogs.com/qlsem/p/11497185.html

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值