阻塞队列--ArrayBlockingQueue

什么是阻塞队列----阻塞队列概述

ArrayBlockingQueue是一个用数组实现的有界阻塞队列,按先进先出的原则对元素进行排序。put和take方法分别为添加和删除的阻塞方法。默认情况下不保证线程公平。

ArrayBlockingQueue整体结构

首先了解一下ArrayBlockingQueue的成员变量:
1)ArrayBlockingQueue内部是用items数组来存储元素的;
2)takeIndex标记下一个take、poll、peek、remove方法被调用时获取数组元素的索引。putIndex标记下一个put、offer、add方法被调用时添加到数组中的索引。
3)ReentrantLock对象用来控制添加线程和移除线程的并发访问;
4)两个条件对象:notEmpty条件对象是用于存放等待或唤醒调用take方法的线程,通知他们队列中已有元素,可以执行获取操作。notFull条件对象则用于等待或唤醒调用put方法的线程,通知他们队列中未满,可以执行添加元素的操作。
5)Itrs变量是迭代器Itr和队列的共享状态,是队列同迭代器Itr中数据及状态的同步的桥梁。

    /** The queued items */
    final Object[] items;

    /** items index for next take, poll, peek or remove */
    int takeIndex;

    /** items index for next put, offer, or add */
    int putIndex;

    /** Number of elements in the queue */
    int count;

    /*
     * Concurrency control uses the classic two-condition algorithm
     * found in any textbook.
     */

    /** Main lock guarding all access */
    final ReentrantLock lock;

    /** Condition for waiting takes */
    private final Condition notEmpty;

    /** Condition for waiting puts */
    private final Condition notFull;

    /**
     * Shared state for currently active iterators, or null if there
     * are known not to be any.  Allows queue operations to update
     * iterator state.
     */
    transient Itrs itrs;

ArrayBlockingQueue内部的阻塞队列是通过一个重入锁ReentrantLock和两个Condition条件队列实现的,所以ArrayBlockingQueue中的元素存在公平访问与非公平访问的区别。

非阻塞插入元素

add和offer方法的实现比较简单,只是在队列满时插入失败抛异常或返回false,队列非满时,则将待插入元素保存在items[putIndex],同时更新putIndex值并发出队列非空的信号即完成了元素的插入操作。
更新putIndex值时,如果当前putIndex==数组长度时,则将其置为0,这是由于元素插入总在队尾,将其当成一个环形队列就很好理解了。

    public boolean add(E e) {
        if (offer(e))
            return true;
        else
            throw new IllegalStateException("Queue full");
    }

    public boolean offer(E e) {
        Objects.requireNonNull(e);
        final ReentrantLock lock = this.lock;
        lock.lock();
        try {
            //当队列满了,则返回false
            if (count == items.length)
                return false;
            else {
                //队列非满,可执行元素插入
                enqueue(e);
                return true;
            }
        } 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();
    }

阻塞插入元素

有时限阻塞和无时限阻塞的执行逻辑基本相同,我们这里就只看无时限阻塞。
如果当前队列已满,则条件对象notFull调用await方法,将当前线程挂起加入到等待队列中,直到有线程移除了元素,调用了notFull.signal,才会将等待队列中的线程唤醒,继续执行添加操作。
如果当前队列非满,则直接调用enqueue完成元素的插入。

    public void put(E e) throws InterruptedException {
        Objects.requireNonNull(e);
        final ReentrantLock lock = this.lock;
        lock.lockInterruptibly();
        try {
            while (count == items.length)
                notFull.await();
            enqueue(e);
        } finally {
            lock.unlock();
        }
    }

移除元素

poll和take的执行逻辑同前面元素插入部分的逻辑相同。poll在队列为空时则直接返回null,take则会在队列为空时,调用notEmpty.await,使得当前线程挂起进入等待队列。在队列非空时,则调用dequeue方法移除元素。
dequeue方法也比较简单,取出takeIndex位置的元素后将该位置置空,同时更新takeIndex的值。调用notFull.signal唤醒在等待队列的因调用put方法进入阻塞状态的线程。

    public E poll() {
        final ReentrantLock lock = this.lock;
        lock.lock();
        try {
            return (count == 0) ? null : dequeue();
        } finally {
            lock.unlock();
        }
    }

    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")
        E x = (E) items[takeIndex];
        items[takeIndex] = null;
        if (++takeIndex == items.length) takeIndex = 0;
        count--;
        //更新迭代器中的元素数据
        if (itrs != null)
            itrs.elementDequeued();
        notFull.signal();
        return x;
    }

remove(Object o)方法用于删除队列中的指定元素,删除逻辑相对复杂一点。
该方法同样需要在重入锁的保护下执行。从takeIndex开始环形遍历,查找到要删除的元素的位置,再调用removeAt方法。
removeAt方法中,如果待删除的元素不是队列头元素,则执行循环操作,从待删除的元素之后的元素都往前移一个位置,后一个元素覆盖了待删除的元素,也就完成了删除操作,同时更新putIndex的值。

    public boolean remove(Object o) {
        if (o == null) return false;
        final ReentrantLock lock = this.lock;
        lock.lock();
        try {
            if (count > 0) {
                final Object[] items = this.items;
                final int putIndex = this.putIndex;
                int i = takeIndex;
                do {
                    //环形遍历队列,查找待删除元素的位置
                    if (o.equals(items[i])) {
                        removeAt(i);
                        return true;
                    }
                    if (++i == items.length) i = 0;
                } while (i != putIndex);
            }
            return false;
        } finally {
            lock.unlock();
        }
    }
    void removeAt(final int removeIndex) {
        final Object[] items = this.items;
        //如果待删除元素位置恰好是下一个删除元素位置,则按直接删除并更新takeIndex
        if (removeIndex == takeIndex) {
            // removing front item; just advance
            items[takeIndex] = null;
            if (++takeIndex == items.length) takeIndex = 0;
            count--;
            if (itrs != null)
                itrs.elementDequeued();
        } else {
            //
            for (int i = removeIndex, putIndex = this.putIndex;;) {
                int pred = i;
                if (++i == items.length) i = 0;
                if (i == putIndex) {
                    items[pred] = null;
                    this.putIndex = pred;
                    break;
                }
                items[pred] = items[i];
            }
            count--;
            if (itrs != null)
                itrs.removedAt(removeIndex);
        }
        notFull.signal();
    }

小结

ArrayBlockingQueue内部以一个数组对象来存储元素,同时使用一个重入锁ReentrantLock来实现多个线程之间插入和删除元素的同步;两个条件对象用于实现阻塞逻辑,通过调用其await和signal方法实现等待和唤醒。
元素的插入删除操作同普通的数组操作相同,这里也就不赘述了。

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 203,547评论 6 477
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,399评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 150,428评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,599评论 1 274
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,612评论 5 365
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,577评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,941评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,603评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,852评论 1 297
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,605评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,693评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,375评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,955评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,936评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,172评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 43,970评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,414评论 2 342