PriorityBlockingQueue详解

2022-10-30 16:25:39 浏览数 (1)

PriorityBlockingQueue介绍

  【1】PriorityBlockingQueue是一个无界的基于数组的优先级阻塞队列,数组的默认长度是11,也可以指定数组的长度,且可以无限的扩充,直到资源消耗尽为止,每次出队都返回优先级别最高的或者最低的元素。默认情况下元素采用自然顺序升序排序,当然我们也可以通过构造函数来指定Comparator来对元素进行排序。需要注意的是PriorityBlockingQueue不能保证同优先级元素的顺序。

  【2】优先级队列PriorityQueue: 队列中每个元素都有一个优先级,出队的时候,优先级最高的先出。

PriorityBlockingQueue的源码分析

  【1】属性值

代码语言:javascript复制
//默认容量
private static final int DEFAULT_INITIAL_CAPACITY = 11;

//最大容量设定
private static final int MAX_ARRAY_SIZE = Integer.MAX_VALUE - 8;

//存放数据的数组
private transient Object[] queue;

//元素个数
private transient int size;

//排序规则(比较器)
private transient Comparator<? super E> comparator;

//独占锁
private final ReentrantLock lock;

//队列为空的时候的阻塞队列
private final Condition notEmpty;

//用于分配的CAS自旋锁
private transient volatile int allocationSpinLock;

//只用于序列化的普通优先队列
private PriorityQueue<E> q;

  【2】构造函数

代码语言:javascript复制
//没有指定容量,则容量默认11
public PriorityBlockingQueue() {
    this(DEFAULT_INITIAL_CAPACITY, null);
}

//有指定容量则容量为指定数值
public PriorityBlockingQueue(int initialCapacity) {
    this(initialCapacity, null);
}

//初始化所需要的属性值
public PriorityBlockingQueue(int initialCapacity, Comparator<? super E> comparator) {
    if (initialCapacity < 1)
        throw new IllegalArgumentException();
    this.lock = new ReentrantLock();
    this.notEmpty = lock.newCondition();
    this.comparator = comparator;
    this.queue = new Object[initialCapacity];
}

  【3】核心方法分析

    1)核心扩容函数

代码语言:javascript复制
//扩容函数
private void tryGrow(Object[] array, int oldCap) {
    lock.unlock(); //必须释放然后重新获得主锁,这一步的意义在于所有操作共享一把锁,在进行扩容时(因为写已满),释放锁(不能写,要等待扩容完才能写),提供给读操作
    Object[] newArray = null;
    // CAS 轻量级锁加锁,避免并发扩容
    if (allocationSpinLock == 0 && UNSAFE.compareAndSwapInt(this, allocationSpinLockOffset, 0, 1)) {
        try {
            // 扩容步长,旧值小于64时,变为两倍 2。大于64时,变为1.5倍。
            int newCap = oldCap   ((oldCap < 64) ? (oldCap   2) :  (oldCap >> 1));
            // 超过最大容量,内存溢出
            if (newCap - MAX_ARRAY_SIZE > 0) {    // possible overflow
                int minCap = oldCap   1;
                if (minCap < 0 || minCap > MAX_ARRAY_SIZE)
                    throw new OutOfMemoryError();
                newCap = MAX_ARRAY_SIZE;
            }
            // 创建新数组
            if (newCap > oldCap && queue == array)
                newArray = new Object[newCap];
        } finally {
            allocationSpinLock = 0;
        }
    }
    // 并发扩容时,线程让出 cpu 执行时间,给其他线程执行,自己稍后执行,原因:加锁不成功必然有其他线程也在扩容,在等待过程中让出资源给其他线程利用
    if (newArray == null) 
        Thread.yield();
    //重新加锁
    lock.lock();
    //变更内存指向,利用内存拷贝复制旧数据
    if (newArray != null && queue == array) {
        queue = newArray;
        System.arraycopy(array, 0, newArray, 0, oldCap);
    }
}

    2)入队方法

代码语言:javascript复制
public void put(E e) {
    offer(e); // never need to block
}
public boolean add(E e) {
    return offer(e);
}

public boolean offer(E e) {
    if (e == null)
        throw new NullPointerException();
    final ReentrantLock lock = this.lock;
    lock.lock();
    int n, cap;
    Object[] array;
    while ((n = size) >= (cap = (array = queue).length))
        //自旋扩容
        tryGrow(array, cap);
    try {
        Comparator<? super E> cmp = comparator;
        //根据比较器采用填充的方式
        if (cmp == null)
            siftUpComparable(n, e, array);
        else
            siftUpUsingComparator(n, e, array, cmp);
        size = n   1;
        notEmpty.signal();
    } finally {
        lock.unlock();
    }
    return true;
}

    4)队列填入方式

      代码展示

代码语言:javascript复制
private static <T> void siftUpComparable(int k, T x, Object[] array) {
    Comparable<? super T> key = (Comparable<? super T>) x;
    while (k > 0) {
        int parent = (k - 1) >>> 1;
        Object e = array[parent];
        if (key.compareTo((T) e) >= 0)
            break;
        array[k] = e;
        k = parent;
    }
    array[k] = key;
}
//雷同上面的不过比较器采用传入的
private static <T> void siftUpUsingComparator(int k, T x, Object[] array,  Comparator<? super T> cmp) {
    while (k > 0) {
        int parent = (k - 1) >>> 1;
        Object e = array[parent];
        if (cmp.compare(x, (T) e) >= 0)
            break;
        array[k] = e;
        k = parent;
    }
    array[k] = x;
}

      图解说明

        1.利用了满二叉树的理念,(k - 1) >>> 1可以获得存入节点的父节点下标,然后进行比较。若判断应该上升,则将父节点置于k存储的地方。

        2.重复循环,知道二叉树root节点,或者已经找到了合适的位置

    4)出队方法

代码语言:javascript复制
public E poll() {
    final ReentrantLock lock = this.lock;
    lock.lock();
    try {
        return dequeue();
    } finally {
        lock.unlock();
    }
}

public E take() throws InterruptedException {
    final ReentrantLock lock = this.lock;
    lock.lockInterruptibly();
    E result;
    try {
        while ( (result = dequeue()) == null)
            notEmpty.await();
    } finally {
        lock.unlock();
    }
    return result;
}

private E dequeue() {
    int n = size - 1;
    if (n < 0)
        return null;
    else {
        Object[] array = queue;
        //将头节点取出返回
        E result = (E) array[0];
        //将末尾节点当做向头节点位置存入的节点
        E x = (E) array[n];
        array[n] = null;
        Comparator<? super E> cmp = comparator;
        if (cmp == null)
            siftDownComparable(0, x, array, n);
        else
            siftDownUsingComparator(0, x, array, n, cmp);
        size = n;
        return result;
    }
}

    5)队列修正方式

      代码展示

代码语言:javascript复制
private static <T> void siftDownComparable(int k, T x, Object[] array, int n) {
    if (n > 0) {
        Comparable<? super T> key = (Comparable<? super T>)x;
        //过滤掉最后一层的元素(以为满二叉树中,最后一层占据的元素就是n/2)
        int half = n >>> 1;           // loop while a non-leaf
        while (k < half) {
            //获取头节点的左节点
            int child = (k << 1)   1; // assume left child is least
            Object c = array[child];
            int right = child   1;
            //进行左右节点的比较,取小的一边作为比较,以为优先队列要确保头节点是最小的(换而言之,保证子树下面的头节点为子树里面最小的即可)
            if (right < n && ((Comparable<? super T>) c).compareTo((T) array[right]) > 0)
                c = array[child = right];
            if (key.compareTo((T) c) <= 0)
                break;
            array[k] = c;
            k = child;
        }
        array[k] = key;
    }
}

private static <T> void siftDownUsingComparator(int k, T x, Object[] array, int n,Comparator<? super T> cmp) {
    if (n > 0) {
        int half = n >>> 1;
        while (k < half) {
            int child = (k << 1)   1;
            Object c = array[child];
            int right = child   1;
            if (right < n && cmp.compare((T) c, (T) array[right]) > 0)
                c = array[child = right];
            if (cmp.compare(x, (T) c) <= 0)
                break;
            array[k] = c;
            k = child;
        }
        array[k] = x;
    }
}

      图解说明

        1.采用尾结点代替头节点,然后利用下沉的方式来修正优先队列里面的数据。

        2.下沉限制在倒数第二层,以为倒数第二层就会与倒数第一层进行比较了,所以应该进行限制(下标位置超出倒数第二层的最大下标就应该停止了)

        3.其次看的时候,可以把左右子树都当做一个小的满二叉树,不断逐层划分,这样条理更清晰。

PriorityBlockingQueue总结

  【1】一个支持优先级排序的无界阻塞队列,优先级高的先出队,优先级低的后出队

  【2】数据结构:数组 二叉堆(默认容量11,可指定初始容量,会自动扩容,最大容量是(Integer.MAX_VALUE - 8))

  【3】锁:ReentrantLock,存取是同一把锁

  【4】阻塞对象:NotEmpty,出队,队列为空时阻塞

  【5】入队,不阻塞,永远返回成功,无界;根据比较器进行堆化(排序)自下而上,如果比较器为 null,则按照自然顺序排序,传入比较器对象就按照比较器的顺序排序。

  【6】出队,优先级最高的元素在堆顶(弹出堆顶元素),弹出前比较两个子节点再进行堆化(自上而下)。

  【7】应用场景:1.业务办理排队叫号,VIP客户插队;2.电商抢购活动,会员级别高的用户优先抢购到商品;

0 人点赞