[Java并发编程实战] 阻塞队列 BlockingQueue(含代码，生产者-消费者模型)

见贤思齐焉，见不贤而内自省也。—《论语》

Java5.0 增加了两种新的容器类型，它们是指：Queue 和 BlockingQueue。Queue 用来临时保存一组等待处理的元素。BlockingQueue 扩张了 Queue 接口，增加了可阻塞的插入和获取等操作。

BlockingQueue 通常运用于一个线程生产对象放入队列，另一个线程从队列获取对象并消费，这是典型的生产者消费者模型。

生产者线程持续生产新对象并插入队列，如果队列已满，那么插入对象的操作会一直阻塞，直到队列中出现可用的空间。
消费者线程持续从队列获取对象，如果队列为空，那么获取操作会一直阻塞，直到队列中出现可用的新对象。BlockingQueue 简化了生产者-消费者设计的实现过程，它支持任意数量的生产者和消费者。

BlockingQueue 的核心方法：

offer(E): 向队列插入元素，并返回插入成功与否。本方法不阻塞当前执行线程。

put(E) : 向队列插入元素，如果队列已满，则会阻塞当前线程直至元素加入队列。

take() : 获取队列的首位元素，如果队列为空，则阻塞当前线程直至队列有元素并取走。

poll():获取队列首个元素，指定时间内一旦数据可取，则立即返回；否则返回失败。

remove(E):删除队列中的元素，返回成功与否。

BlockingQueue 的实现

BlockingQueue是一个接口，所以你必须使用它的实现来使用它。它的实现包括以下几个：

• ArrayBlockingQueue：基于数组实现的有界队列(FIFO)，使用一把全局锁并行对 queue 读写操作，同时使用两个 Condition 阻塞队列为空时的取操作和队列为满时的写操作。

• LinkedBlockingQueue：基于已链接节点的，范围上限为 Integer.MAX_VALUE 的 blocking queue(FIFO)。主要操作 put 和 take 都是阻塞的。

• DelayQueue：当指定的延迟时间到了，才能够从队列中获取元素。它没有大小限制，因此插入元素时不会阻塞，而只有获取元素时才会被阻塞。它的用法可以参考下面两篇博客：
  http://www.cnblogs.com/jobs/archive/2007/04/27/730255.html，
  http://www.cnblogs.com/sunzhenchao/p/3515085.html

• PriorityBlockingQueue: 基于优先级的阻塞队列,但需要注意的是PriorityBlockingQueue并不会阻塞数据生产者，而只会在没有可消费的数据时，阻塞数据的消费者。

• SynchronoutQueue：它的内部同时只能够容纳单个元素。如果该队列已有一个元素的话，试图向队列插入一个新元素线程会阻塞，知道另一个线程将该元素从队列中拿走。同样，如果该队列为空，试图向队列中抽取一个元素的线程将会阻塞，知道另一个线程向队列中插入一个新的元素。SynchronousQueue适合一对一的匹配场景，没有容量，无法缓存。它的用法强烈推荐博客：
  http://www.cnblogs.com/leesf456/p/5560362.html

BlockingQueue的使用

这是一个使用 BlockingQueue 的例子，本例使用 ArrayBlockingQueue 实现。首先，BlockingQueueTest 创建一个生产者线程 Procucer, 把字符存放进共享队列。然后创建三个消费者线程 Consumer，把字符串从队列中取出。Consumer 取到最后一个字符串时，中断所有消费者线程，结束程序。

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.Vector;
import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.BlockingQueue;

public class BlockingQueueTest {
    //队列容量
    private static final int SIZE = 5;
    private static final int CONSUMER_SIZE = 3;
    //消费者线程退出标志
    private static String endString = "num:" + (SIZE*2-1);
    //存放消费者线程
    private static List list = new ArrayList<Thread>();

    public static void main(String[] args) throws Exception{
        //创建固定长度的阻塞队列
        BlockingQueue q = new ArrayBlockingQueue<String>(SIZE);

        //创建生产者
        Producer producer = new Producer(q);
        //启动生产者线程，生产对象
        producer.start();

        //启动消费者线程，获取队列对象
        for(int i = 0; i < CONSUMER_SIZE; i++) {
            list.add(new Consumer(q));
            ((Thread) list.get(i)).start();
        }
    }

    //中断线程
    public static void shutDownThread() {
        for(int i = 0; i < CONSUMER_SIZE; i++) {
            ((Thread) list.get(i)).interrupt();
        }
    }

    static class Producer extends Thread{

        private BlockingQueue queue = null;

        public Producer(BlockingQueue q) {
            this.queue = q;
        }

        @Override
        public void run() {
            // TODO Auto-generated method stub
            try {
                //生产10个对象，放进队列
                for(int i = 0; i < SIZE*2; i++) {
                    String str = "num:" + i;
                    System.out.println(Thread.currentThread().getName() +":"+"IN: " + str);
                    queue.put(str);
                    Thread.sleep(100);
                }
            } catch (InterruptedException e) {
                // TODO Auto-generated catch block
                e.printStackTrace();
            }

        }
    }

    //消费者线程
    static class Consumer extends Thread{

        private BlockingQueue queue = null;

        public Consumer(BlockingQueue q) {
            this.queue = q;
        }

        @Override
        public void run() {
            // TODO Auto-generated method stub
            try {
                //获取队列的元素，队列为空时会阻塞
                while(true) {
                    String str = (String) queue.take();
                    System.out.println(Thread.currentThread().getName() + ":" + "OUT " + str);
                    //已经取出最后一个元素，消费者线程应该退出,否则程序一直在运行
                    if(str.equals(endString)) {
                        shutDownThread();//中断线程
                    }
                }
            } catch (InterruptedException e) {
                // TODO Auto-generated catch block
                e.printStackTrace();
            }
        }
    }
}

执行结果：

总结

java.util.concurrent 中实现的阻塞队列包含了足够的同步机制，从而能够安全的将对象从生产者线程发布到消费者线程。对于可变对象，生产者-消费者模型，把对象的所有权安全的从生产者交付给消费者。转移后，消费者线程获得这个对象的所有权（独占访问权,可以任意修改它），并且生产者线程不会再访问它。同时，阻塞队列负责所有的控制流，使得消费者生产者的代码更加简单和清晰。

本文完毕，如对你有帮助，请关注我，谢谢~

参考

1. https://blog.csdn.net/defonds/article/details/44021605/
2. http://www.cnblogs.com/leesf456/p/5428630.html

3. 《并发编程实战》

   本文原创首发于微信公众号 [ 林里少年 ]，欢迎关注第一时间获取更新。