队列的学习(三) 手写一个阻塞队列

简介: 队列的学习(三) 手写一个阻塞队列本文将介绍如何手写一个阻塞队列。阻塞队列是一种线程安全的队列,当队列为空时,消费者线程将被阻塞直到队列中有元素可供消费;当队列已满时,生产者线程将被阻塞直到队列有空闲位置可供插入元素。

队列的学习(三) 手写一个阻塞队列

本文将介绍如何手写一个阻塞队列。阻塞队列是一种线程安全的队列,当队列为空时,消费者线程将被阻塞直到队列中有元素可供消费;当队列已满时,生产者线程将被阻塞直到队列有空闲位置可供插入元素。

阻塞队列的实现

我们可以使用 Java 提供的 LockCondition 接口来实现阻塞队列。具体实现如下:

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 作为队列的底层数据结构,使用 LockCondition 接口来实现阻塞队列,其中:

  • 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 个元素。

总结

阻塞队列是一种常用的线程安全队列,可以有效地协调生产者和消费者线程的工作。本文介绍了如何手写一个阻塞队列,并提供了使用示例。希望本文能够对大家学习阻塞队列有所帮助!

 

相关文章
|
Java 调度
多线程学习(三)那些队列可用于线程池
多线程学习(三)那些队列可用于线程池
95 0
|
7月前
|
算法 Java
Java数据结构——队列
Java数据结构——队列
47 4
|
7月前
|
存储 Java
Java实现队列
Java实现队列
40 0
Java实现队列
|
7月前
|
存储 Java C++
Java集合篇之深度解析Queue,单端队列、双端队列、优先级队列、阻塞队列
Java集合篇之深度解析Queue,单端队列、双端队列、优先级队列、阻塞队列
59 0
|
Java
|
存储
什么是队列,如何实现?
什么是队列,如何实现?
113 0
什么是队列,如何实现?
|
缓存 算法 安全
面试题 | 徒手写一个非阻塞线程安全队列 ConcurrentLinkedQueue?
面试题 | 徒手写一个非阻塞线程安全队列 ConcurrentLinkedQueue?
97 0
|
存储 缓存 安全
面试题 | 徒手写一个非阻塞线程安全队列 ConcurrentLinkedQueue?(下)
面试题 | 徒手写一个非阻塞线程安全队列 ConcurrentLinkedQueue?
78 0
|
存储 Java
Java数据结构之队列
队列是一个有序列表,可以用数组或是链表来实现。 遵循先入先出的原则。即:先存入队列的数据,要先取出。后存入的要后取出
119 0