文章目录
- 前言
- 实战演示
- 写在最后
前言
前几天看见一个高级Java面试题,我觉得很有代表意义。既考察了面试者的基本锁机制运用,也了解了阻塞队列的产生实现原理。先分享出来,以供鉴赏。
面试题:实现一个自定义的阻塞队列,它应该支持以下功能:
添加元素到队列末尾(如果队列已满,则阻塞添加线程直到有空间)
从队列头部移除元素(如果队列为空,则阻塞获取元素的线程直到有元素可消费)
查询队列是否为空
查询队列是否已满
要求使用Java的ReentrantLock和Condition类来实现线程间的同步。
实战演示
既然题目说到了使用Java的ReentrantLock和Condition类来实现线程间的同步,也就为我们指明了方向,直接可以按照题目的思路进行编码。当然,如果题目没有要求使用技术点,你是否知道还有哪些方式可以实现这个阻塞队列呢?
自定义阻塞队列编码如下:
/*** CustomBlockingQueue* @author senfel* @date 2024/2/21 12:50*/
public class CustomBlockingQueue<T> {private final LinkedList<T> queue = new LinkedList<>();private final int capacity;private final ReentrantLock lock = new ReentrantLock();private final Condition notEmpty = lock.newCondition();private final Condition notFull = lock.newCondition();public CustomBlockingQueue(int capacity) {this.capacity = capacity;}/*** put* @author senfel* @date 2024/2/21 12:52* @return void*/public void put(T item) throws InterruptedException {lock.lock();try {while (queue.size() == capacity) {notFull.await();}if (queue.isEmpty()) {notEmpty.signal(); // 如果之前有等待的消费者,现在可以唤醒了}queue.addLast(item);} finally {lock.unlock();}}/*** take* @author senfel* @date 2024/2/21 12:59* @return void*/public T take() throws InterruptedException {lock.lock();try {while (queue.isEmpty()) {notEmpty.await();}if (queue.size() == capacity) {notFull.signal(); // 如果之前有等待的生产者,现在可以唤醒了}return queue.removeFirst();} finally {lock.unlock();}}public boolean isEmpty() {lock.lock();try {return queue.isEmpty();} finally {lock.unlock();}}public boolean isFull() {lock.lock();try {return queue.size() == capacity;} finally {lock.unlock();}}
}
写在最后
这个CustomBlockingQueue类使用了Java并发包中的ReentrantLock作为锁机制,并通过其条件变量notEmpty和notFull分别管理队列空和队列满的情况。当队列满时,调用put方法会阻塞,直到有空间;当队列空时,调用take方法也会阻塞,直到有元素可消费。同时,当添加或删除元素改变队列状态时,会适当调整对应的条件变量以唤醒等待的线程。