Java 集合框架系列(五):Queue 接口及其实现类

Queue 队列,也是 Collection 的一个重要分支,是一个先进先出的数据结构,其一个重要的子接口为BlockingQueue(阻塞队列),阻塞队列在多线程的场景中有着广泛的应用,平时用的队列可以直接使用 LinkedList 来使用。

阻塞队列

阻塞队列中的阻塞意为等待,与非阻塞队列相比的话有如下的不同点:

  1. 入队:
  • 非阻塞队列:当队列满的时候,放入新的元素时,数据丢失。
  • 阻塞队列:当队列满的时候,放入新的元素时,线程阻塞等待,等待队列中有出队的元素,再继续运行放进去。
  1. 出队:
  • 非阻塞队列:当队列没有元素的时候,取数据时得到的是null。
  • 阻塞队列:当队列没有元素的时候,取数据时,线程阻塞等待,什么时候有元素入队了,才可以继续运行取出元素。

常用 API

  1. 添加元素
 // 不能添加 null,成功时返回 true,不成功时抛出 Queue Full 异常
public boolean add(E, e);
 // 不能添加null,成功时返回true,不成功时返回false
public boolean offer(E, e);
// 队列满的时候阻塞队列
public void put(E e); 
  1. 查询
// 获取并移除此队列的头部,在元素变得可用之前一直等待
public E take(); 
// 获取并移除此队列的头部,在指定时间内等待可用的元素
public E poll(long time, TimeUnit unit); 
// 获取队列的头部,不移除 
public E peek(); 

BlockingQueue 的常见实现类

ArrayBlockingQueue

ArrayBlockingQueue 底层是一个基于数组的有边界的阻塞队列,其只用了一把锁来同时阻塞读写操作,读写不分离。

  1. 类的定义和关键字段
 public class ArrayBlockingQueue<E> extends AbstractQueue<E>
         implements BlockingQueue<E>, java.io.Serializable {
     // 底层使用数组保存数据
     final Object[] items;
     // 取元素时用到的索引,初始为0
     int takeIndex;
     // 放元素时用到的索引,初始为0
     int putIndex;
     // 数组中元素的个数
     int count;
     // 可重入锁,用于出队和入队等操作中
     final ReentrantLock lock;
     // lock伴随的一个不为空的等待池,当队列有数据时会唤醒在该等待队列中的线程获取数据
     private final Condition notEmpty;
     // lock伴随的一个队列不满的等待池,当队列中还没有满时会唤醒在该等待队列中的线程往队列        添加数据
     private final Condition notFull;
     
     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();
     }
 }
  1. 入队
// 添加元素
 public void put(E e) throws InterruptedException {
     checkNotNull(e);    // 检查是否为空
     // 同步的进行添加元素
     final ReentrantLock lock = this.lock;
     lock.lockInterruptibly();
     try {
         while (count == items.length)
             // 元素满了的话将当前线程放入已满等待池中阻塞
             notFull.await();
         // 入队操作-可能会唤醒在非空等待池中阻塞的队列
         enqueue(e);
     } finally {
         lock.unlock();
     }
 }
 // 入队
 private void enqueue(E x) {
     final Object[] items = this.items;
     // 将x放入putIndex索引指向的位置
     items[putIndex] = x;
     if (++putIndex == items.length)
         // 循环数组的作用,可以不断的利用底层数组存放数据
         putIndex = 0;
     // 元素个数+1
     count++;
     // 添加一个元素则证明底层数组有数据了,唤醒在空队列的等待池中阻塞的线程取出数据,即唤醒take方法的阻塞
     notEmpty.signal();
 }
  1. 出队
 // 取出元素
 public E take() throws InterruptedException {
     final ReentrantLock lock = this.lock;
     lock.lockInterruptibly();
     try {
         while (count == 0)
             // 如果没有元素的话,则会将当前线程放入非空等待池中阻塞
             notEmpty.await();
         return dequeue();
     } finally {
         lock.unlock();
     }
 }
 
// 入队
private E dequeue() {
     final Object[] items = this.items;
     @SuppressWarnings("unchecked")
     E x = (E) items[takeIndex];
     items[takeIndex] = null;
     if (++takeIndex == items.length)
         // 循环数组
         takeIndex = 0;
     count--; // 元素个数-1
     if (itrs != null)
         itrs.elementDequeued();
     // 取出一个元素则证明队列中又有空间可以存放数据了,唤醒在满队列的等待池中阻塞的线程,可以往队列中继续添加数据,即唤醒put方法的阻塞
     notFull.signal();
     return x;
 }

Condition 等待队列。
总结:

  1. 两个 Conditional 变量的作用,notEmpty 用来阻塞获取元素操作,当队列非空时(enqueue 中唤醒)会唤醒该等待池中的线程;notFull 用来阻塞添加元素操作,当队列未满时(dequeue 中唤醒)会唤醒该等待池中的线程。
  2. 读写操作只用到一把锁,读写不分离。
  3. 必须指定初始化容量。对于 put 操作当队列已满的时候会阻塞线程,对于 take 操作当队列为空时也会阻塞线程。
    能否将while(count == items.length)和while(count == 0)中的while换成if?
    不能,因为 notFull 中的线程被唤醒的瞬间,假如这个时候有其他线程放入元素,此时队列又满了,线程仍然需要执行wait。如果采用 if 的话只会判断一次,线程被唤醒之后就会继续执行 enqueue 操作,但是此时队列是满的,执行 enqueue 就会造成数据的丢失了。因此需要不断的判断,直到唤醒的线程判断出当前队列真的有位置了才能继续运行。

LinkedBlockingQueue

底层基于链表结构,支持读写同时操作,并发情况下,性能比 ArrayBlockingQueue 高。是一个可选择的有界队列,可以指定链表的容量,当不指定时默认为整型的最大值。
其 API 的使用与 ArrayBlockingQueue 一样。

  1. 类的定义及关键字段
public class LinkedBlockingQueue<E> extends AbstractQueue<E>
         implements BlockingQueue<E>, java.io.Serializable {
     
     // 链表的节点结构
     static class Node<E> {
         E item;     // 数据域
         Node<E> next;   // 下个节点的指针域
         Node(E x) { item = x; }
     }
     // 链表的最大容量,不指定时默认为 Integer.MAX_VALUE
     private final int capacity;
     // 链表中元素的个数,因为读写操作是分离的,所以要使用原子操作来改变元素的个数
     private final AtomicInteger count = new AtomicInteger();
     // 头结点
     transient Node<E> head;
     // 尾节点
     private transient Node<E> last;
     // 获取数据的锁
     private final ReentrantLock takeLock = new ReentrantLock();
     // 队列非空等待池
     private final Condition notEmpty = takeLock.newCondition();
     // 添加数组的锁
     private final ReentrantLock putLock = new ReentrantLock();
     // 队列不满时的等待池
     private final Condition notFull = putLock.newCondition();
     
     public LinkedBlockingQueue() {
         // 默认容量为整型最大值
         this(Integer.MAX_VALUE);
     }
     public LinkedBlockingQueue(int capacity) {
         if (capacity <= 0) throw new IllegalArgumentException();
         this.capacity = capacity;
         last = head = new Node<E>(null);
     }
 }
  1. 入队
private void enqueue(Node<E> node) {
     // 尾插法,直接添加到最后一个元素,并改变last的指向
     last = last.next = node;
 }
  // 添加元素
 public void put(E e) throws InterruptedException {
     if (e == null) throw new NullPointerException();
     int c = -1;
     // 封装成新的节点结构
     Node<E> node = new Node<E>(e);
     final ReentrantLock putLock = this.putLock;
     final AtomicInteger count = this.count;
     putLock.lockInterruptibly();
     try {
         while (count.get() == capacity) {
             // 队列已满,则阻塞当期线程,采用 while 的原因与 ArrayBlockingQueue 相同
             notFull.await();
         }
         enqueue(node);
         // 递增元素个数,c返回的是旧值
         c = count.getAndIncrement();
         if (c + 1 < capacity)
             // 再次判断队列是否满的,因为count可能同时被读线程操作
             notFull.signal();
     } finally {
         putLock.unlock();
     }
     if (c == 0)
         // 队列中有元素了,唤醒take阻塞的线程,这里的c是旧值,count=1
         signalNotEmpty();
 }
  1. 出队
// 出队
 private E dequeue() {
     Node<E> h = head;
     // 第一个节点,会成为新的头节点,
     Node<E> first = h.next;
     h.next = h; // help GC,这样操作没有别的引用h这个节点,垃圾回收会进行回收
     head = first;
     // 返回第一个节点的数据
     E x = first.item;
     // head的数据域为空
     first.item = null;
     return x;
 }
  // 取出元素
 public E take() throws InterruptedException {
     E x;
     int c = -1;
     final AtomicInteger count = this.count;
     final ReentrantLock takeLock = this.takeLock;
     takeLock.lockInterruptibly();
     try {
         while (count.get() == 0) {
             notEmpty.await();
         }
         x = dequeue();
         // 原子的递减元素个数
         c = count.getAndDecrement();
         if (c > 1)
             // 再次判断是否有元素
             notEmpty.signal();
     } finally {
         takeLock.unlock();
     }
     if (c == capacity)
         // c是count的旧值,c==capacity就证明队列有空余出来的位置了,唤醒put阻塞的线程
         signalNotFull();
     return x;
 }

LinkedBlockingQueue 在 FixedThreadPool 和 SingleThreadExecutor 线程池中被使用。

为什么使用了锁还要使用 while 进行排队队列是否为空或者队列是否满了?
这是因为底层使用的是 ReetrantLock,支持可重复入的。

ArrayBlockingQueue和LinkedBlockingQueue的对比:
假如有 10000 个线程,分别有 5000 个线程进行读操作,有 5000 个线程进行写操作,其性能对比如下:

public static void main(String[] args) throws InterruptedException {
     ArrayBlockingQueue<String> blockingQueue = new ArrayBlockingQueue<>(10000);
     // LinkedBlockingQueue<String> linkedBlockingQueue = new LinkedBlockingQueue<>(10000);
     List<String> dataList = new Vector<>();
     // 创建10000个线程,分别进行500次的读写操作
     long start = System.currentTimeMillis();
     for (int i = 0; i < 10000; i++) {
         if (i % 2 == 0) {
             new Thread(() -> {
                 for (int j = 0; j < 500; j++) {
                     try {
                         blockingQueue.put(Thread.currentThread().getName() + "j");
                     } catch (InterruptedException e) {
                         e.printStackTrace();
                     }
                 }
             }, "Thread-" + i).start();
         } else {
             new Thread(() -> {
                 for (int j = 0; j < 500; j++) {
                     try {
                         dataList.add(blockingQueue.take());
                     } catch (InterruptedException e) {
                         e.printStackTrace();
                     }
                 }
             }, "Thread-" + i).start();
         }
 
     }
     long end = System.currentTimeMillis();
     System.out.println("用时:" + (end - start) + "ms");
     System.out.println(dataList.size());
 }

结果:
ArrayBlockingQueue用时:2331ms
LinkedBlockingQueue用时:2319ms
两者性能差不多,有时 LinkedBlockingQueue 的性能还比不上 ArrayBlockingQueue。

SynchronousQueue

使用这个队列必须先从队列中取出元素(即先调用 take 方法),才可以向队列中加入元素(再调用 put 方法),很像管道。SynchronousQueue 队列中没有任何容量,甚至一个容量都没有。可以理解一个标记,当一个线程从队列获取数据的时候就会打上一个标记,之后如果有另外一个线程放入数据,数据就会直接传送给获取数据的线程。
如果先向队列中添加元素则会抛出异常 Queue Full,因为队列是没有容量的。使用 put 方法的话阻塞,因为一开始队列就是满的。注意取出元素的方法不能用 peek,因为 peek 不会将元素从队列中拿走,只是查看的作用。
优点:方便高效的进行线程间的数据传送,效率高,不会产生队列中数据争抢问题。

public static void main(String[] args) throws InterruptedException {
     SynchronousQueue<String> synchronousQueue = new SynchronousQueue<>();
     new Thread(() -> {
         try {
             System.out.println(Thread.currentThread().getName() + ":" + synchronousQueue.take());
         } catch (InterruptedException e) {
             e.printStackTrace();
         }
     }, "Thread-take").start();
     new Thread(() -> {
         try {
             synchronousQueue.put("abc");
         } catch (InterruptedException e) {
             e.printStackTrace();
         }
     }, "Thread-put").start();

 }

因此一般要多个线程配合使用才行。常用于 CachedThreadPool。

PriorityBlockingQueue

带有优先级的阻塞队列,队列中的元素有不同的优先级;没有界限限制(整型最大值),但是可以指定初始化的长度,不指定时默认为11,底层会进行扩容操作。放入的元素必须实现内部比较器,或者在创建 PriorityBlockingQueue 时构造方法中指定外部比较器,元素的优先级根据比较器来决定。

public static void main(String[] args) throws InterruptedException {
     PriorityBlockingQueue<String> blockingQueue = new PriorityBlockingQueue<>(11, new Comparator<String>() {
         @Override
         public int compare(String o1, String o2) {
             return o1.compareTo(o2);
         }
     });
 
     blockingQueue.put("bbb");
     blockingQueue.put("aaa");
     blockingQueue.put("ccc");
     System.out.println(blockingQueue);
     System.out.println("------------------------------------------------------");
     System.out.println(blockingQueue.take());
     System.out.println(blockingQueue.take());
     System.out.println(blockingQueue.take());
 
 }

DelayQueue

DelayQueue 是一个无界的 BlockingQueue,用于放置实现了 Delayed 接口的对象,存放在DelayQueue 队列中的对象只能在其到期时才能从队列中取走。
当生产者线程调用 put 方法添加元素时,会触发 Delayed 接口中的 compareTo 方法进行排序,也就是队列中的元素的顺序是按到期时间排序的,而非进入队列的顺序。排在队列头部的元素最早到期,越往后到期时间越晚。
消费者线程查看队列头部的元素,然后调用元素的 getDelay 方法,如果此方法的返回值 ≤0,则消费者线程会从队列中取出此元素处理。如果 getDealy 方法返回值大于0,则消费者线程 wait 返回的时间值后,再次从队列头部取出元素,此时元素到期可以取出。也就是说会线程会按照设定的时间进行阻塞。
注意:不能将 null 元素放置到这种队列中。
使用场景:

  1. 淘宝订单业务:下单之后如果30分钟之内没有付款就自动取消订单。
  2. 饿了么订餐通知:下单成功后60S之内给用户发送短信通知。
  3. 关闭空闲连接:服务器中客户端的连接空闲一段时间后会自动关闭。
  4. 缓存:缓存中的对象超过空闲时间限制,需要从缓存中移出。
  5. 任务超时处理:在tcp协议的滑动窗口中,用来处理超时未响应的请求。
 public class TestDelayQueue {
     public static void main(String[] args) throws InterruptedException {
         DelayQueue<Student> delayQueue = new DelayQueue<>();
         delayQueue.put(new Student("zhangsan", 20, System.currentTimeMillis() + 5000));
         delayQueue.put(new Student("lisi", 19, System.currentTimeMillis() + 2000));
         delayQueue.put(new Student("wangwu", 18, System.currentTimeMillis() + 1000));
         System.out.println(delayQueue);
         System.out.println(delayQueue.take());
         System.out.println(delayQueue.take());
         System.out.println(delayQueue.take());
     }
 }
 // Student.java
 public class Student implements Delayed {
 
     private String name;
     private int age;
     private long endTime;
 
     public Student() {}
 
     public Student(String name, int age, long endTime) {
         this.name = name;
         this.age = age;
         this.endTime = endTime;
     }
 
     @Override
     public long getDelay(TimeUnit unit) {
         return endTime - System.currentTimeMillis();
     }
 
     @Override
     public int compareTo(Delayed o) {
         Student other = (Student) o;
         return age - other.getAge();
     }
 
     public String getName() {
         return name;
     }
 
     public void setName(String name) {
         this.name = name;
     }
 
     public int getAge() {
         return age;
     }
 
     public void setAge(int age) {
         this.age = age;
     }
 
     @Override
     public String toString() {
         return "Student{" +
                 "name='" + name + '\'' +
                 ", age=" + age +
                 ", endTime=" + endTime +
                 '}';
     }
 }

实现 Delayed 接口的实现类需要重写 getDelay 和 compareTo 方法。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值