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

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

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

BlockingQueue 通常运用于一个线程生产对象放入队列,另一个线程从队列获取对象并消费,这是典型的生产者消费者模型。
[Java并发编程实战] 阻塞队列 BlockingQueue(含代码,生产者-消费者模型)_第1张图片

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

BlockingQueue 的核心方法:

[Java并发编程实战] 阻塞队列 BlockingQueue(含代码,生产者-消费者模型)_第2张图片

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();

    public static void main(String[] args) throws Exception{
        //创建固定长度的阻塞队列
        BlockingQueue q = new ArrayBlockingQueue(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并发编程实战] 阻塞队列 BlockingQueue(含代码,生产者-消费者模型)_第3张图片

总结

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

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

参考

  1. https://blog.csdn.net/defonds/article/details/44021605/
  2. http://www.cnblogs.com/leesf456/p/5428630.html
  3. 《并发编程实战》

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

你可能感兴趣的:(Java并发编程实战)