前言
用Java如何设计一个阻塞队列,这个问题是在面滴滴的时候被问到的。当时确实没回答好,只是说了用个List,然后消费者再用个死循环一直去监控list的是否有值,有值的话就处理List里面的内容。回头想想,自己真是一个大傻X,也只有我才会这么设计一个阻塞队列(再说,我这也不是阻塞的队列)。
结果自己面试完之后,也没去总结这部分知识,然后过了一段时间,某教育机构的面试又被问到类似的问题了,只不过是换了一个形式,“请用wait方法和notify方法实现一套有生产者和消费者的这种逻辑”。然后我就又蒙圈了,追悔莫及,为啥我没有去了解一下这部分知识,所以这次我准备好好总结一下这部分内容。
具体实现
如果说实现一个队列,那么一个LinkedList的这种实现了Queue接口的都可以直接使用,或者自己写一个先进先出的Array都可以。
但是要做到阻塞就还需要进行阻塞的实现,就是说当队列是空时,如果再继续从队列中获取数据,将会被阻塞,直到有新的数据入队列才停止阻塞;还有当队列已经满了(到达设置的最大容量),再往队列里添加元素的操作也会被阻塞,直到有数据从队列中被移除。
这里首先要有一个锁,保证同时只能有一个线程执行出队列、同时只能有一个线程执行入队列。而执行出队列和入队列的线程的阻塞和唤醒,是靠wait()方法和notifyAll()方法来实现的。
代码实现如下:
public class MyBlockQueue {
/**
* 队列长度默认为10
*/
private int limit = 10;
private Queue queue = new LinkedList<>();
/**
* 初始化队列容量
* @param limit 队列容量
*/
public MyBlockQueue(int limit){
this.limit = limit;
}
/**
* 入队列
* @param object 队列元素
* @throws InterruptedException
*/
public synchronized boolean push(Object object) throws InterruptedException{
// 如果队列已满,再来添加队列的线程就直接阻塞等待。
while (this.queue.size() == this.limit){
wait();
}
// 如果队列为空了,就唤醒所有阻塞的线程。
if(this.queue.size() == 0){
notifyAll();
}
// 入队
boolean add = this.queue.offer(object);
return add;
}
/**
* 出队列
* @return
* @throws InterruptedException
*/
public synchronized Object pop() throws InterruptedException{
// 如果出队列时,队列为空,则阻塞队列。
while (this.queue.size() == 0){
wait();
}
// 如果队列重新满了之后,唤醒阻塞的所有线程。
if(this.queue.size() == this.limit){
notifyAll();
}
Object poll = this.queue.poll();
return poll;
}
}
Java中阻塞队列的实现
首先我们先来归纳一下,Java中有哪些已经实现好了的阻塞队列:
队列 |
描述 </ |