BlockingQueue:顾名思义,首先它是一个队列,并且支持阻塞的机制,阻塞的放入和得到数据。我们要实现LinkedBlockingQueue下面两个简单的方法put和take。
Put:把anObject加到BlockingQueue里,如果BlockQueue没有空间,则调用此方法的线程被阻塞,直到BlockingQueue里面有空间再继续。
Take:取走BlockingQueue里排在首位的对象,若BlockingQueue为空,阻塞进入等待状态直到BlockingQueue有新的数被加入。
示例:
import java.util.LinkedList; import java.util.concurrent.TimeUnit; import java.util.concurrent.atomic.AtomicInteger; public class MyQueue { // 装载集合 private final LinkedList<Object> list = new LinkedList<Object>(); //计数器 线程安全的 private AtomicInteger count = new AtomicInteger(0); private final int minSize = 0; private final int maxSize ; public MyQueue(int size) { this.maxSize = size; } //初始化一个对象进行加锁 private final Object lock = new Object(); public void put(Object obj) { synchronized (lock) { if(count.get()>=maxSize) { try { lock.wait(); } catch (InterruptedException e) { // TOD O Auto-generated catch block e.printStackTrace(); } } // list.add(obj); //计数器自增 count.incrementAndGet(); System.out.println("添加一个元素.."); //唤醒取元素线程 lock.notify(); } } public Object take() { Object r = new Object(); synchronized (lock) { while(count.get()==this.minSize) { try { lock.wait(); } catch (InterruptedException e) { // TODO Auto-generated catch block e.printStackTrace(); } } //移除元素 r = list.removeFirst(); //计数器自减 count.decrementAndGet(); //唤醒添加线程 lock.notify(); } return r; } public int getSize() { return this.count.get(); } public static void main(String[] args) { //在线程中使用变量必须用final取修饰 final MyQueue mq = new MyQueue(5); mq.put("a"); mq.put("b"); mq.put("c"); mq.put("d"); mq.put("e"); System.out.println("当前队列元素数:" + mq.getSize()); Thread t1 = new Thread(new Runnable() {
@Override public void run() { // TODO Auto-generated method stub mq.put("f"); mq.put("g"); } }); t1.start(); Thread t2 = new Thread(new Runnable() {
@Override public void run() { // TODO Auto-generated method stub Object o1 = mq.take(); System.out.println("移除的元素为:" + o1); Object o2 = mq.take(); System.out.println("移除的元素为:" + o2); } }); try { //停止2秒后再执行t2线程,保证t1线程全部执行完 TimeUnit.SECONDS.sleep(2); } catch (InterruptedException e) { // TODO Auto-generated catch block e.printStackTrace(); } t2.start(); } } |
添加一个元素.. 添加一个元素.. 添加一个元素.. 添加一个元素.. 添加一个元素.. 当前队列元素数:5 移除的元素为:a 添加一个元素.. 移除的元素为:b 添加一个元素.. |