队列的学习(三) 手写一个阻塞队列
本文将介绍如何手写一个阻塞队列。阻塞队列是一种线程安全的队列,当队列为空时,消费者线程将被阻塞直到队列中有元素可供消费;当队列已满时,生产者线程将被阻塞直到队列有空闲位置可供插入元素。
阻塞队列的实现
我们可以使用 Java 提供的 Lock
和 Condition
接口来实现阻塞队列。具体实现如下:
import java.util.LinkedList;
import java.util.concurrent.locks.Condition;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;
public class BlockingQueue<T> {
private LinkedList<T> queue;
private int capacity;
private Lock lock;
private Condition notFull;
private Condition notEmpty;
public BlockingQueue(int capacity) {
this.capacity = capacity;
queue = new LinkedList<>();
lock = new ReentrantLock();
notFull = lock.newCondition();
notEmpty = lock.newCondition();
}
public void put(T element) throws InterruptedException {
lock.lock();
try {
while (queue.size() == capacity) {
notFull.await();
}
queue.add(element);
notEmpty.signal();
} finally {
lock.unlock();
}
}
public T take() throws InterruptedException {
lock.lock();
try {
while (queue.isEmpty()) {
notEmpty.await();
}
T element = queue.remove();
notFull.signal();
return element;
} finally {
lock.unlock();
}
}
}
在上述代码中,我们使用了 LinkedList
作为队列的底层数据结构,使用 Lock
和 Condition
接口来实现阻塞队列,其中:
lock
用于同步队列的插入和删除操作;notFull
用于在队列已满时阻塞生产者线程;notEmpty
用于在队列为空时阻塞消费者线程。
阻塞队列的使用
使用阻塞队列时,我们需要先创建一个阻塞队列对象,指定队列的容量,然后在生产者线程中调用 put
方法向队列中插入元素,在消费者线程中调用 take
方法从队列中获取元素。具体使用方法如下:
public static void main(String[] args) throws InterruptedException {
BlockingQueue<Integer> queue = new BlockingQueue<>(10);
new Thread(() -> {
for (int i = 0; i < 20; i++) {
try {
queue.put(i);
System.out.println("Producer put: " + i);
} catch (InterruptedException e) {
e.printStackTrace();
}
}
}).start();
new Thread(() -> {
for (int i = 0; i < 20; i++) {
try {
Integer element = queue.take();
System.out.println("Consumer take: " + element);
} catch (InterruptedException e) {
e.printStackTrace();
}
}
}).start();
}
在上述代码中,我们创建了一个容量为 10 的阻塞队列,启动了一个生产者线程和一个消费者线程,生产者线程向队列中插入了 20 个元素,消费者线程从队列中取出了 20 个元素。
总结
阻塞队列是一种常用的线程安全队列,可以有效地协调生产者和消费者线程的工作。本文介绍了如何手写一个阻塞队列,并提供了使用示例。希望本文能够对大家学习阻塞队列有所帮助!