在上一节,介绍了ArrayBlockingQueue的添加元素的方法,本节,结合源码给大家介绍一下获取元素的方法。
获取元素的方法有下述几种
E poll() 立刻返回,如果队列为空,返回null
E take() 如果队列不为空,返回队首元素,否则阻塞到队列不为空
E poll(long timeout, TimeUnit unit) 等待timeout 时间的poll
E peek() 获得队首的元素,并不将这个元素弹出
分别看一下这几个方法的源码
1.E poll()
public E poll() {
final ReentrantLock lock = this.lock;
lock.lock();
try {
return (count == 0) ? null : dequeue();
} finally {
lock.unlock();
}
}
此方法先获得lock,如果lock获得成功,会先去判断count 的值,如果这个值为0,那就说明此队列为空,立刻返回null
2.E take()
public E take() throws InterruptedException {
final ReentrantLock lock = this.lock;
lock.lockInterruptibly();
try {
while (count == 0)
notEmpty.await();
return dequeue();
} finally {
lock.unlock();
}
}
take 方法和poll方法的区别就在于,如果count为0,那么在notEmpty上等待,直到被唤醒,这里有个需要遵循的原则,await/wait方法调用的时候,一定要确保本线程持有依赖的状态(这里边为count)的锁,并且被唤醒