1. 程式人生 > >java執行緒安全之wait/notify模擬Queue (九)

java執行緒安全之wait/notify模擬Queue (九)

概念理解

      BlockingQueue:顧名思義,首先他是一個佇列,並且支援阻塞的機制,阻塞的放入和得到資料。我們要實現LinkedBlockingQueue下面兩個簡單的方法put和take。

     put(anObject):把anObject加到BlockingQueue裡,如果BlockingQueue沒有空間,則呼叫此方法的執行緒被阻斷,直到BlockingQueue裡面有空間在繼續。

     take():取走BlockingQueue裡排在首位的物件,若BlockingQueue為空,阻斷進入等待狀態直到BlockingQueue有新的資料被加入。

案例


public class MyQueue {

    //1 需要一個承裝元素的集合 
    private LinkedList<Object> list = new LinkedList<Object>();

    //2 需要一個計數器
    private AtomicInteger count = new AtomicInteger(0);

    //3 需要制定上限和下限
    private final int minSize = 0;

    private final int maxSize ;

    //4 構造方法
    public
MyQueue(int size){ this.maxSize = size; } //5 初始化一個物件 用於加鎖 private final Object lock = new Object(); //put(anObject): 把anObject加到BlockingQueue裡,如果BlockQueue沒有空間,則呼叫此方法的執行緒被阻斷,直到BlockingQueue裡面有空間再繼續. public void put(Object obj){ synchronized (lock) { if
(count.get() == this.maxSize){ try { //釋放鎖 lock.wait(); } catch (InterruptedException e) { e.printStackTrace(); } } //1 加入元素 list.add(obj); //2 計數器累加 count.incrementAndGet(); //3 通知另外一個執行緒(喚醒) lock.notify(); System.out.println("新加入的元素為:" + obj); } } //take: 取走BlockingQueue裡排在首位的物件,若BlockingQueue為空,阻斷進入等待狀態直到BlockingQueue有新的資料被加入. public Object take(){ Object ret = null; synchronized (lock) { if(count.get() == this.minSize){ try { //釋放鎖 lock.wait(); } catch (InterruptedException e) { e.printStackTrace(); } } //1 做移除第一個元素操作 ret = list.removeFirst(); //2 計數器遞減 count.decrementAndGet(); //3 喚醒另外一個執行緒 lock.notify(); } return ret; } public int getSize(){ return this.count.get(); } public static void main(String[] args) { 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() { mq.put("f"); mq.put("g"); } },"t1"); t1.start(); Thread t2 = new Thread(new Runnable() { @Override public void run() { //取走第一個 Object o1 = mq.take(); System.out.println("移除的元素為:" + o1); Object o2 = mq.take(); System.out.println("移除的元素為:" + o2); } },"t2"); try { TimeUnit.SECONDS.sleep(2); } catch (InterruptedException e) { e.printStackTrace(); } t2.start(); } }

列印結果

新加入的元素為:a
新加入的元素為:b
新加入的元素為:c
新加入的元素為:d
新加入的元素為:e
當前容器的長度:5
新加入的元素為:f
移除的元素為:a
移除的元素為:b
新加入的元素為:g