详细分析Java并发集合ArrayBlockingQueue的用法
在上一章中,我们介绍了阻塞队列BlockingQueue,下面我们介绍它的常用实现类ArrayBlockingQueue。
一.用数组来实现队列
因为队列这种数据结构的特殊要求,所以它天然适合用链表的方式来实现,用两个变量分别记录链表头和链表尾,当删除或插入队列时,只要改变链表头或链表尾就可以了,而且链表使用引用的方式链接的,所以它的容量几乎是无限的。
那么怎么使用数组来实现队列,我们需要四个变量:Object[]array来存储队列中元素,headIndex和tailIndex分别记录队列头和队列尾,count记录队列的个数。
- 因为数组的长度是固定,所以当count==array.length时,表示队列已经满了,当count==0的时候,表示队列是空的。
- 当添加元素的时候,将array[tailIndex]=e将tailIndex位置设置成新元素,之后将tailIndex++自增,然后将count++自增。但是有两点需要注意,在添加之前必须先判断队列是否已满,不然会出现覆盖已有元素。当tailIndex的值等于数组最后一个位置的时候,需要将tailIndex=0,循环利用数组
- 当删除元素的时候,将先记录下array[headIndex]元素,之后将headIndex++自增,然后将count--自减。但是有两点需要注意要注意,在删除之前,必须先判断队列是否为空,不然可能会删除已删除的元素。
这里用了一个很巧妙的方式,我们知道当向队列中插入一个元素,那么就占用了数组的一个位置,当删除一个元素的时候,那么其实数组的这个位置就空闲出来了,表示这个位置又可以插入新元素了。
所以我们插入新元素前,必须检查队列是否已满,删除元素之前,必须检查队列是否为空。
二.ArrayBlockingQueue中重要成员变量
/**储存队列的中元素*/ finalObject[]items; /**队列头的位置*/ inttakeIndex; /**队列尾的位置*/ intputIndex; /**当前队列拥有的元素个数*/ intcount; /**用来保证多线程操作共享变量的安全问题*/ finalReentrantLocklock; /**当队列为空时,就会调用notEmpty的wait方法,让当前线程等待*/ privatefinalConditionnotEmpty; /**当队列为满时,就会调用notFull的wait方法,让当前线程等待*/ privatefinalConditionnotFull;
就是多了lock、notEmpty、notFull变量来实现多线程安全和线程等待条件的,它们三个是怎么操作的呢?
2.1lock的作用
因为ArrayBlockingQueue是在多线程下操作的,所以修改items、takeIndex、putIndex和count这些成员变量时,必须要考虑多线程安全问题,所以这里使用lock独占锁,来保证并发操作的安全。
2.2notEmpty与notFull的作用
因为阻塞队列必须实现,当队列为空或队列已满的时候,队列的读取或插入操作要等待。所以我们想到了并发框架下的Condition对象,使用它来控制。
在AQS中,我们介绍了这个类的作用:
- await系列方法,会释放当前锁,并让当前线程等待。
- signal与signalAll方法,会唤醒当前线程。其实它并不是唤醒当前线程,而是将在这个Condition条件上等待的线程,添加到lock锁上的等待线程池中,所以当锁被释放时,会唤醒lock锁上的等待线程池中一个线程。具体在AQS中有源码分析。
三.添加元素方法
3.1add(Ee)与offer(Ee)方法:
//调用AbstractQueue父类中的方法。 publicbooleanadd(Ee){ //通过调用offer来时实现 if(offer(e)) returntrue; else thrownewIllegalStateException("Queuefull"); } //向队列末尾新添加元素。返回true表示添加成功,false表示添加失败,不会抛出异常 publicbooleanoffer(Ee){ checkNotNull(e); finalReentrantLocklock=this.lock; //使用lock来保证,多线程修改成员属性的安全 lock.lock(); try{ //队列已满,添加元素失败,返回false。 if(count==items.length) returnfalse; else{ //调用enqueue方法将元素插入队列中 enqueue(e); returntrue; } }finally{ lock.unlock(); } }
add方法是调用offer方法实现的。在offer方法中,必须先判断队列是否已满,如果已满就直接返回false,而不会阻塞当前线程。如果不满就调用enqueue方法将元素插入队列中。
注意:这里使用lock.lock()是保证同一时间只有一个线程修改成员变量,防止出现并发操作问题。虽然它也会阻塞当前线程,但是它并不是条件等待,只是因为锁被其他线程持有,而ArrayBlockingQueue中方法操作时间都不长,这里相当于不阻塞线程。
3.2put方法
//向队列末尾新添加元素,如果队列已满,当前线程就等待。响应中断异常 publicvoidput(Ee)throwsInterruptedException{ checkNotNull(e); finalReentrantLocklock=this.lock; //使用lock来保证,多线程修改成员属性的安全 lock.lockInterruptibly(); try{ //队列已满,则调用notFull.await()方法,让当前线程等待,直到队列不是满的 while(count==items.length) notFull.await(); //调用enqueue方法将元素插入队列中 enqueue(e); }finally{ lock.unlock(); } }
与offer方法大体流程一样,只是当队列已满的时候,会调用notFull.await()方法,让当前线程阻塞等待,直到队列被别的线程移除了元素,队列不满的时候,会唤醒这个等待线程。
3.3offer(Ee,longtimeout,TimeUnitunit)方法
/** *向队列末尾新添加元素,如果队列中没有可用空间,当前线程就等待, *如果等待时间超过timeout了,那么返回false,表示添加失败 */ publicbooleanoffer(Ee,longtimeout,TimeUnitunit) throwsInterruptedException{ checkNotNull(e); //计算一共最多等待的时间值nanos longnanos=unit.toNanos(timeout); finalReentrantLocklock=this.lock; //使用lock来保证,多线程修改成员属性的安全 lock.lockInterruptibly(); try{ //队列已满 while(count==items.length){ //nanos<=0表示最大等待时间已到,那么不用再等待了,返回false,表示添加失败。 if(nanos<=0) returnfalse; //调用notFull.awaitNanos(nanos)方法,超时nanos时间会被自动唤醒, //如果被提前唤醒,那么返回剩余的时间 nanos=notFull.awaitNanos(nanos); } //调用enqueue方法将元素插入队列中 enqueue(e); returntrue; }finally{ lock.unlock(); } }
与put的方法大体流程一样,只不过是调用notFull.awaitNanos(nanos)方法,让当前线程等待,并设置等待时间。
四.删除元素方法
4.1remove()和poll()方法:
//调用AbstractQueue父类中的方法。 publicEremove(){ //通过调用poll来时实现 Ex=poll(); if(x!=null) returnx; else thrownewNoSuchElementException(); } //删除队列第一个元素(即队列头),并返回它。如果队列是空的,它不会抛出异常,而是会返回null。 publicEpoll(){ finalReentrantLocklock=this.lock; //使用lock来保证,多线程修改成员属性的安全 lock.lock(); try{ //如果count==0,列表为空,就返回null,否则调用dequeue方法,返回列表头元素 return(count==0)?null:dequeue(); }finally{ lock.unlock(); } }
remove方法是调用poll()方法实现的。在poll()方法中,如果列表为空,就返回null,否则调用dequeue方法,返回列表头元素。
4.2take()方法
/** *返回并移除队列第一个元素,如果队列是空的,就前线程就等待。响应中断异常 */ publicEtake()throwsInterruptedException{ finalReentrantLocklock=this.lock; //使用lock来保证,多线程修改成员属性的安全 lock.lockInterruptibly(); try{ //如果队列为空,就调用notEmpty.await()方法,让当前线程等待。 //直到有别的线程向队列中插入元素,那么这个线程会被唤醒。 while(count==0) notEmpty.await(); //调用dequeue方法,返回列表头元素 returndequeue(); }finally{ lock.unlock(); } }
take()方法当队列为空的时候,当前线程必须等待,直到有别的线程向队列中插入新元素,那么这个线程会被唤醒。
4.3poll(longtimeout,TimeUnitunit)方法
/** *返回并移除队列第一个元素,如果队列是空的,就前线程就等待。 *如果等待时间超过timeout了,那么返回false,表示获取元素失败 */ publicEpoll(longtimeout,TimeUnitunit)throwsInterruptedException{ //计算一共最多等待的时间值nanos longnanos=unit.toNanos(timeout); finalReentrantLocklock=this.lock; //使用lock来保证,多线程修改成员属性的安全 lock.lockInterruptibly(); try{ //队列为空 while(count==0){ //nanos<=0表示最大等待时间已到,那么不用再等待了,返回null。 if(nanos<=0) returnnull; //调用notEmpty.awaitNanos(nanos)方法让档期线程等待,并设置超时时间。 nanos=notEmpty.awaitNanos(nanos); } //调用dequeue方法,返回列表头元素 returndequeue(); }finally{ lock.unlock(); } }
与take()方法流程一样,只不过调用awaitNanos(nanos)方法,让当前线程等待,并设置等待时间。
五.查看元素的方法
5.1element()与peek()方法
//调用AbstractQueue父类中的方法。 publicEelement(){ Ex=peek(); if(x!=null) returnx; else thrownewNoSuchElementException(); } //查看队列头元素 publicEpeek(){ finalReentrantLocklock=this.lock; //使用lock来保证,多线程修改成员属性的安全 lock.lock(); try{ //返回当前队列头的元素 returnitemAt(takeIndex);//nullwhenqueueisempty }finally{ lock.unlock(); } }
六.其他重要方法
6.1enqueue和dequeue方法
//将元素x插入队列尾 privatevoidenqueue(Ex){ //assertlock.getHoldCount()==1; //assertitems[putIndex]==null;//当前putIndex位置元素一定是null finalObject[]items=this.items; items[putIndex]=x; //如果putIndex等于items.length,那么将putIndex重新设置为0 if(++putIndex==items.length) putIndex=0; //队列数量加一 count++; //因为插入一个元素,那么当前队列肯定不为空,那么唤醒在notEmpty条件下等待的一个线程 notEmpty.signal(); } //删除队列头的元素,返回它 privateEdequeue(){ //assertlock.getHoldCount()==1; //assertitems[takeIndex]!=null; finalObject[]items=this.items; //得到当前队列头的元素 @SuppressWarnings("unchecked") Ex=(E)items[takeIndex]; //将当前队列头位置设置为null items[takeIndex]=null; if(++takeIndex==items.length) takeIndex=0; //队列数量减一 count--; if(itrs!=null) itrs.elementDequeued(); //因为删除了一个元素,那么队列肯定不满了,那么唤醒在notFull条件下等待的一个线程 notFull.signal(); returnx; }
这两个方法分别代表,向队列中插入元素和从队列中删除元素。而且它们要唤醒等待的线程。当插入元素后,那么队列一定不为空,那么就要唤醒在notEmpty条件下等待的线程。当删除元素后,那么队列一定不满,那么就要唤醒在notFull条件下等待的线程。
6.2remove(Objecto)方法
//删除队列中对象o元素,最多删除一条 publicbooleanremove(Objecto){ if(o==null)returnfalse; finalObject[]items=this.items; //使用lock来保证,多线程修改成员属性的安全 finalReentrantLocklock=this.lock; lock.lock(); try{ //当队列中有值的时候,才进行删除。 if(count>0){ //队列尾下一个位置 finalintputIndex=this.putIndex; //队列头的位置 inti=takeIndex; do{ //当前位置元素与被删除元素相同 if(o.equals(items[i])){ //删除i位置元素 removeAt(i); //返回true returntrue; } if(++i==items.length) i=0; //当i==putIndex表示遍历完所有元素 }while(i!=putIndex); } returnfalse; }finally{ lock.unlock(); } }
从队列中删除指定对象o,那么就要遍历队列,删除第一个与对象o相同的元素,如果队列中没有对象o元素,那么返回false删除失败。
这里有两点需要注意:
如何遍历队列,就是从队列头遍历到队列尾。就要靠takeIndex和putIndex两个变量了。
为什么Object[]items=this.items;这句代码没有放到同步锁lock代码块内。items是成员变量,那么多线程操作的时候,不会有并发问题么?
这个是因为items是个引用变量,不是基本数据类型,而且我们对队列的插入和删除操作,都是针对这一个items数组,没有改变数组的引用,所以在lock代码中,items会得到其他线程对它最新的修改。但是如果这里将intputIndex=this.putIndex;方法lock代码块外面,就会产生问题。
removeAt(finalintremoveIndex)方法
//删除队列removeIndex位置的元素 voidremoveAt(finalintremoveIndex){ //assertlock.getHoldCount()==1; //assertitems[removeIndex]!=null; //assertremoveIndex>=0&&removeIndex在队列中删除指定位置的元素。需要注意的是删除之后的数组还能保持队列形式,分为两种情况:
- 如果删除位置是队列头,那么简单,只需要将队列头的位置元素设置为null,将将队列头位置加一。
- 如果删除位置不是队列头,那么麻烦了,这个时候,我们就要将从removeIndex位置后的元素全部左移一位,覆盖前一个元素。最后将原来队列尾的元素置位null。
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持毛票票。