并发编程复习（六）：使用wait和notify模拟阻塞队列

话不多说，代码如下

import java.util.LinkedList;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;

/**
* 使用wait和notify模拟阻塞队列......
*/
public class MyQueue {

private int minSize = 0;

private int maxSize;

public MyQueue(int size) {
this.maxSize = size;
}

private LinkedList<Object> list = new LinkedList<>();

private AtomicInteger count = new AtomicInteger();

private Object lock = new Object();

public void put(Object obj) {
synchronized (lock) {
while (count.intValue() == maxSize) {
try {
lock.wait();
} catch (InterruptedException e) {
e.printStackTrace();
}
}
list.add(obj);
System.out.println("添加的元素为："+obj);
count.incrementAndGet();
lock.notify();
}
}
public Object take() {
Object rel = null;
synchronized (lock) {
while (count.intValue() == minSize) {
try {
lock.wait();
} catch (InterruptedException e) {
e.printStackTrace();
}
}
rel = list.removeFirst();
System.out.println("移除的元素为："+rel);
count.decrementAndGet();
lock.notify();
}
return rel;
}

public Integer getSize() {
return list.size();
}

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 thread1 = new Thread(new Runnable() {
@Override
public void run() {
mq.put("f");
mq.put("g");
}
},"thread01:");
thread1.start();
Thread t2 = new Thread(new Runnable() {
@Override
public void run() {
Object o = mq.take();
System.out.println("取得的元素是："+o);
Object o1 = mq.take();
System.out.println("取得的元素是："+o1);
}
},"thread02:");
try {
TimeUnit.SECONDS.sleep(2);
} catch (InterruptedException e) {
e.printStackTrace();
}
t2.start();
}
}


输出结果：
添加的元素为：a
添加的元素为：b
添加的元素为：c
添加的元素为：d
添加的元素为：e
队列中的对象个数为：5
移除的元素为：a
添加的元素为：f
取得的元素是：a
移除的元素为：b
取得的元素是：b
添加的元素为：g