java執行緒安全之wait/notify模擬Queue (九)
阿新 • • 發佈:2019-01-29
概念理解
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