博客
关于我
技术专家面试实战: 用数组实现一个阻塞队列?
阅读量:587 次
发布时间:2019-03-09

本文共 2999 字,大约阅读时间需要 9 分钟。

ArrayBlockingQueue是一种基于数组实现的阻塞队列,广泛应用于生产者消费者模式中的任务缓冲,如线程池中的工作队列。那么,如何用数组实现这种阻塞队列呢?让我们一步步深入探讨。

ArrayBlockingQueue的核心功能

ArrayBlockingQueue作为一个队列,需要具备入队和出队的基本操作。由于它是阻塞队列,当队列已满时,入队操作应阻塞,直到队列有空位可用;而当队列为空时,出队操作也应阻塞,直到队列中有元素可供获取。同时,由于ArrayBlockingQueue通常在多线程环境中使用,线程安全性是其核心需求。

实现ArrayBlockingQueue的步骤

我们可以将问题分解为两个主要部分:用数组实现队列以及实现条件阻塞和线程安全。

用数组实现队列

实现队列的入队和出队操作,可以通过循环数组的方式来处理。这种方式避免了每次出队都需要移动大量元素的问题,时间复杂度为O(1)。具体来说,我们可以使用两个索引变量,分别记录下一个要入队和下一个要出队的位置。当入队到数组末尾时,从0开始重新计算;同样,当出队到末尾时,也从0开始重新计算。同时,使用一个计数器来存储当前队列的元素数量。

实现条件阻塞和线程安全

在Java中,实现条件等待有两种主要方式:synchronized + Object.wait和Lock + Condition.await。选择后者可以更好地支持线程安全,因为它不仅支持中断,还能支持多个条件。

通过Lock和Condition,我们可以定义两个条件:一个用于检测队列是否不空(notEmpty),另一个用于检测队列是否已满(notFull)。当入队时,先检查队列是否已满,如果满了则调用notFull.await()进行条件等待;入队完成后,唤醒等待不空条件的线程。同样,出队操作也是类似的逻辑。

ArrayBlockingQueue的具体实现

下面是一个实现ArrayBlockingQueue的代码示例:

class ArrayBlockingQueue
{
private final Object[] items;
private int takeIndex;
private int putIndex;
private int count;
private ReentrantLock lock;
private final Condition notEmpty;
private final Condition notFull;
public ArrayBlockingQueue(int capacity) {
if (capacity <= 0) {
throw new IllegalArgumentException();
}
this.items = new Object[capacity];
lock = new ReentrantLock();
notEmpty = lock.newCondition();
notFull = lock.newCondition();
}
private void enqueue(E e) {
Object[] items = this.items;
if (putIndex < items.length && count < items.length) {
items[putIndex] = e;
putIndex = (putIndex + 1) % items.length;
count++;
notFull.signal();
}
}
private E dequeue() {
Object[] items = this.items;
E e = (E) items[takeIndex];
items[takeIndex] = null;
takeIndex = (takeIndex + 1) % items.length;
count--;
notEmpty.signal();
return e;
}
public void put(E e) throws InterruptedException {
lock.lockInterruptibly();
try {
while (count == items.length) {
notFull.await();
}
enqueue(e);
notEmpty.signal();
} finally {
lock.unlock();
}
}
public E take() throws InterruptedException {
lock.lockInterruptibly();
try {
while (count == 0) {
notEmpty.await();
}
E e = dequeue();
notFull.signal();
return e;
} finally {
lock.unlock();
}
}
}

代码解释

  • 初始化:构造函数初始化数组、锁和条件。检查容量是否合理,否则抛出异常。

  • 入队操作:在锁的保护下,判断队列是否已满。如果已满,调用notFull.await()进行条件等待;否则,执行入队操作,并调用notEmpty.signal()唤醒等待线程。

  • 出队操作:在锁的保护下,判断队列是否为空。如果为空,调用notEmpty.await()进行条件等待;否则,执行出队操作,并调用notFull.signal()唤醒等待线程。

  • 代码中的细节

  • put方法中的局部lock变量:这是为了优化性能,避免在方法内频繁访问实例变量时引起的性能 overhead。

  • while循环的锁内执行:确保在多个线程下正确的进入和退出队列状态,防止多个线程同时入队或出队。

  • 通过上述实现,我们可以清晰地看到ArrayBlockingQueue如何在数组中实现高效的入队和出队操作,同时确保线程安全和条件阻塞。这种设计在多线程环境中非常有用,特别是在需要生产者和消费者之间交换任务的情况下。

    转载地址:http://hmtpz.baihongyu.com/

    你可能感兴趣的文章
    MySQL 中随机抽样:order by rand limit 的替代方案
    查看>>
    MySQL 为什么需要两阶段提交?
    查看>>
    mysql 为某个字段的值加前缀、去掉前缀
    查看>>
    mysql 主从
    查看>>
    mysql 主从 lock_mysql 主从同步权限mysql 行锁的实现
    查看>>
    mysql 主从互备份_mysql互为主从实战设置详解及自动化备份(Centos7.2)
    查看>>
    mysql 主从关系切换
    查看>>
    MYSQL 主从同步文档的大坑
    查看>>
    mysql 主键重复则覆盖_数据库主键不能重复
    查看>>
    Mysql 事务知识点与优化建议
    查看>>
    Mysql 优化 or
    查看>>
    mysql 优化器 key_mysql – 选择*和查询优化器
    查看>>
    MySQL 优化:Explain 执行计划详解
    查看>>
    Mysql 会导致锁表的语法
    查看>>
    mysql 使用sql文件恢复数据库
    查看>>
    mysql 修改默认字符集为utf8
    查看>>
    Mysql 共享锁
    查看>>
    MySQL 内核深度优化
    查看>>
    mysql 内连接、自然连接、外连接的区别
    查看>>
    mysql 写入慢优化
    查看>>