模拟阻塞 Queue

BlockingQueue 即阻塞队列，它是基于 ReentrantLock，依据它的基本原理，我们可以实现 Web 中的长连接聊天功能，当然其最常用的还是用于实现生产者与消费者模式，大致如下图所示：

在 Java 中，BlockingQueue是一个接口，它的实现类有 ArrayBlockingQueue、 DelayQueue 、 LinkedBlockingDeque 、 LinkedBlockingQueue 、 PriorityBlockingQueue、SynchronousQueue 等，它们的区别主要体现在存储结构上或对元素操作上的不同，但是对于 take 与 put 操作的原理，却是类似的。 put(an Object)：把 an Object 加到 BlockingQueue 里，如果 BlockQueue 没有空间，则调用此方法的线程被阻断，直到 BlockingQueue 里面有空间再继续。
take：取走 BlockingQueue 里排在首位的对象，若 BlockingQueue 为空，阻断进入等待状态直到 BlockingQueue 有新的数据被加入。
模拟阻塞队列代码如下：

public class MyQueue {

    // 1 需要一个承装元素的集合
    private LinkedList<Object> list = new LinkedList<>();

    // 2 需要一个计数器
    private AtomicInteger count = new AtomicInteger(0);

    // 3 需要制定上限和下限
    private final int minSize = 0;

    private final int maxSize;

    // 4 构造方法
    public MyQueue(int size) {
        this.maxSize = size;
    }

    // 5 初始化一个对象 用于加锁
    private final Object LOCK = new Object();

    // put(an Object): 把an Object加到BlockingQueue里,如果BlockQueue没有空间,则调用此方法的线程被阻断，直到BlockingQueue里面有空间再继续.
    public void put(Object obj) {
        synchronized (LOCK) {
            // 空间满了，线程阻塞
            while (count.get() == this.maxSize) {
                try {
                    LOCK.wait();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
            // 1 加入元素
            list.add(obj);
            // 2 计数器累加
            count.incrementAndGet();
            // 3 通知另外一个线程（唤醒）
            LOCK.notify();
            System.out.println("新加入的元素为:" + obj);
        }
    }

    // take: 取走BlockingQueue里排在首位的对象,若BlockingQueue为空,阻断进入等待状态直到BlockingQueue有新的数据被加入.
    public Object take() {
        Object ret = null;
        synchronized (LOCK) {
            while (count.get() == this.minSize) {
                try {
                    LOCK.wait();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
            // 1 做移除元素操作
            ret = list.removeFirst();
            // 2 计数器递减
            count.decrementAndGet();
            // 3 唤醒另外一个线程
            LOCK.notify();
        }
        return ret;
    }

    public int getSize() {
        return this.count.get();
    }

    public static void main(String[] args) {

        final MyQueue mq = new MyQueue(5);
        mq.put("a");
        mq.put("b");
        mq.put("c");
        mq.put("d");
        mq.put("e");

        System.out.println("当前容器的长度:" + mq.getSize());// 5

        Thread t1 = new Thread(() -> {
            mq.put("f");
            mq.put("g");
        }, "t1");
        t1.start();


        Thread t2 = new Thread(() -> {
            Object o1 = mq.take();
            System.out.println("移除的元素为:" + o1);
            Object o2 = mq.take();
            System.out.println("移除的元素为:" + o2);
        }, "t2");

        try {
            TimeUnit.SECONDS.sleep(2);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

        t2.start();
    }

}

运行结果：

我们通过上面的代码就实现了这么一个功能：就是一个队列，先加入5个元素，然后队列没有空间了，这时候再添加元素，就会进行阻塞状态，并且会唤醒消费者线程进行读取，等到读取了元素之后，会唤醒生产者线程进行添加，当添加之后又会唤醒消费者进行读取，所以打印出来的效果就是这样。