Java中阻塞队列代码示例解析

本篇文章小编给大家分享一下Java中阻塞队列代码示例解析，文章代码介绍的很详细，小编觉得挺不错的，现在分享给大家供大家参考，有需要的小伙伴们可以来看看。

什么是阻塞队列

在数据结构中，队列遵循FIFO（先进先出）原则。在java中，Queue接口定义了定义了基本行为，由子类完成实现，常见的队列有ArrayDeque、LinkedList等，这些都是非线程安全的，在java 1.5中新增了阻塞队列，当队列满时，添加元素的线程呈阻塞状态；当队列为空时，获取元素的线程呈阻塞状态。

生产者、消费者模型

生产者将元素添加到队列中，消费中获取数据后完成数据处理。两者通过队列解决了生产者和消费者的耦合关系；当生产者的生产速度与消费者的消费速度不一致时，可以通过大道缓冲的目的。

阻塞队列的使用场景

线程池

在线程池中，当工作线程数大于等于corePoolSize时，后续的任务后添加到阻塞队列中；

目前有那些阻塞队列

在java中，BlockingQueue接口定义了阻塞队列的行为，常用子类是ArrayBlockingQueue和LinkedBlockingQueue。

BlockingQueue继承了Queue接口，拥有其全部特性。在BlockingQueue的java doc中对其中的操作方法做了汇总

插入元素

add(e):当队列已满时，再添加元素会抛出异常IllegalStateException

offer(e):添加成功，返回true，否则返回false

put:(e):当队列已满时，再添加元素会使线程变为阻塞状态

offer(e, time,unit):当队列已满时，在末尾添加数据，如果在指定时间内没有添加成功，返回false，反之是true

删除元素

remove(e):返回true表示已成功删除，否则返回false

poll():如果队列为空返回null，否则返回队列中的第一个元素

take():获取队列中的第一个元素，如果队列为空，获取元素的线程变为阻塞状态

poll(time, unit)：当队列为空时，线程被阻塞，如果超过指定时间，线程退出

检查元素

element():获取队头元素，如果元素为null，抛出NoSuchElementException

peek():获取队头元素，如果队列为空返回null，否则返回目标元素

ArrayBlockingQueue

底层基于数组的有界阻塞队列，在构造此队列时必须指定容量；

构造函数

// 第一个	
public ArrayBlockingQueue(int capacity, boolean fair,Collection c) {
        this(capacity, fair);

        final ReentrantLock lock = this.lock;
        lock.lock(); // Lock only for visibility, not mutual exclusion
        try {
            int i = 0;
            try {
                for (E e : c) {
                    checkNotNull(e);
                    items[i++] = e;
                }
            } catch (ArrayIndexOutOfBoundsException ex) {
                throw new IllegalArgumentException();
            }
            count = i;
            putIndex = (i == capacity) ? 0 : i;
        } finally {
            lock.unlock();
        }
    }

	// 第二个
    public ArrayBlockingQueue(int capacity, boolean fair) {
        if (capacity <= 0)
            throw new IllegalArgumentException();
        this.items = new Object[capacity];
        lock = new ReentrantLock(fair);
        notEmpty = lock.newCondition();
        notFull =  lock.newCondition();
    }

	// 第三个
	public ArrayBlockingQueue(int capacity) {
        this(capacity, false);
    }

capacity:队列的初始容量

fair:线程访问队列的公平性。如果为true按照FIFO的原则处理，反之；默认为falsec:

已有元素的集合，类型于合并两个数组

put()方法

public void put(E e) throws InterruptedException {
         // 检查元素是否为null
        checkNotNull(e);
        final ReentrantLock lock = this.lock;
        // 获取锁
        lock.lockInterruptibly();
        try {
            // 如果当前队列为空，变为阻塞状态
            while (count == items.length)
                notFull.await();
            // 反之，就添加元素
            enqueue(e);
        } finally {
            // 解锁
            lock.unlock();
        }
    }

    private void enqueue(E x) {
        final Object[] items = this.items;
        items[putIndex] = x;
        if (++putIndex == items.length)
            putIndex = 0;
        count++;
        // 此时队列不为空，唤醒消费者
        notEmpty.signal();
    }

take()方法

 public E take() throws InterruptedException {
        final ReentrantLock lock = this.lock;
        // 获取锁
        lock.lockInterruptibly();
        try {
            // 如果队列为空，消费者变为阻塞状态
            while (count == 0)
                notEmpty.await();
            // 不为空，就获取数据
            return dequeue();
        } finally {
            // 解锁
            lock.unlock();
        }
    }

        private E dequeue() {
        final Object[] items = this.items;
        @SuppressWarnings("unchecked")
        // 获取队头元素x
        E x = (E) items[takeIndex];
        items[takeIndex] = null;
        if (++takeIndex == items.length)
            takeIndex = 0;
        count--;
        if (itrs != null)
            itrs.elementDequeued();
         // 此时队列没有满，同时生产者继续添加数据
        notFull.signal();
        return x;
    }

LinkedBlockingQueue

底层基于单向链表的无界阻塞队列，如果不指定初始容量，默认为Integer.MAX_VALUE，否则为指定容量

构造函数

// 不指定容量     
	public LinkedBlockingQueue() {
        this(Integer.MAX_VALUE);
    }
	// 指定容量
    public LinkedBlockingQueue(int capacity) {
        if (capacity <= 0) throw new IllegalArgumentException();
        this.capacity = capacity;
        last = head = new Node(null);
    }

	// 等同于合并数组
    public LinkedBlockingQueue(Collection c) {
        this(Integer.MAX_VALUE);
        final ReentrantLock putLock = this.putLock;
        putLock.lock(); // Never contended, but necessary for visibility
        try {
            int n = 0;
            for (E e : c) {
                if (e == null)
                    throw new NullPointerException();
                if (n == capacity)
                    throw new IllegalStateException("Queue full");
                enqueue(new Node(e));
                ++n;
            }
            count.set(n);
        } finally {
            putLock.unlock();
        }
    }

put()方法

 public void put(E e) throws InterruptedException {
        // 元素为空，抛出异常
        if (e == null) throw new NullPointerException();
        int c = -1;
        Node node = new Node(e);
        final ReentrantLock putLock = this.putLock;
        // 获取队列中的数据量
        final AtomicInteger count = this.count;
        // 获取锁
        putLock.lockInterruptibly();
        try {
            // 队列满了，变为阻塞状态
            while (count.get() == capacity) {
                notFull.await();
            }
            // 将目标元素添加到链表的尾端
            enqueue(node);
            // 总数增加
            c = count.getAndIncrement();
            // 队列还没有满，继续添加元素
            if (c + 1 < capacity)
                notFull.signal();
        } finally {
            // 解锁
            putLock.unlock();
        }
        if (c == 0)
            signalNotEmpty();
    }

take()方法

public E take() throws InterruptedException {
        E x;
        int c = -1;
        // 获取队列中的工作数
        final AtomicInteger count = this.count;
        final ReentrantLock takeLock = this.takeLock;
        // 获取锁
        takeLock.lockInterruptibly();
        try {
            // 如果队列为空，变为阻塞状态
            while (count.get() == 0) {
                notEmpty.await();
            }
            // 获取队头元素
            x = dequeue();
            // 递减
            c = count.getAndDecrement();
            // 通知消费者
            if (c > 1)
                notEmpty.signal();
        } finally {
            // 解锁
            takeLock.unlock();
        }
        if (c == capacity)
            // 
            signalNotFull();
        return x;
    }

对比

相同点

两者都是通过Condition通知生产者和消费者完成元素的添加和获取

都可以指定容量

不同点

ArrayBlockingQueue基于数据，LinkedBlockingQueue基于链表

ArrayBlockingQueue内有一把锁，LinkedBlockingQueue内有两把锁

自己动手实现一个阻塞队列

通过分析源码可以知道，阻塞队列其实是通过通知机制Condition完成生产者和消费的互通。也可以通过Object类中的wait()和notify、notifyAll实现。下面是自己写的一个阻塞队列

public class BlockQueue {
    // 对象锁
    public static final Object LOCK = new Object();
    // 控制变量的值 来通知双方
    public boolean condition;
    
    public void put() {
        synchronized (LOCK) {
            while (condition) {
                try {
                    // 满了
                    System.out.println("put   队列满了，开始阻塞");
                    LOCK.wait();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
            condition = true;
            System.out.println("put   改为true，唤醒消费者");
            LOCK.notifyAll();
        }
    }


    public void take() {
        synchronized (LOCK) {
            while (!condition) {
                // 没满
                System.out.println("take   队列没满，开始阻塞");
                try {
                    LOCK.wait();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
            condition = false;
            System.out.println("take   改为false，唤醒生产者");
            LOCK.notifyAll();
        }
    }
}