package concurrent;
import java.util.LinkedList;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.locks.Condition;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;
/**
* 面试题:写一个固定容量的同步容器,拥有put和get方法,以及getCount方法
* 能够支持2个生产者线程以及10个消费者的阻塞调用
* 方法1:使用wait和notifyAll来实现 -- while
* 方法2:使用Lock和Condition的方式更加精确的指定哪些线程被唤醒
* @author BarryLee
* @2018年11月6日@下午3:57:59
*/
public class TestCondition {
LinkedList<Object>list = new LinkedList<>();//这个容器,使用LinkedList的好处就是可以removeFirst,List接口中并没有
final int MAX = 10;//容器最大容量
Lock lock = new ReentrantLock();//锁
Condition producer = lock.newCondition();//生产者
Condition consumer = lock.newCondition();//消费者
public int getCount() {
return list.size();
}
public void put(Object o) {
lock.lock();
try {
while(list.size()==MAX) {
System.out.println("生产者等待...");
producer.await();
}
Thread.sleep(2);
System.out.println("add");
list.add(o);
consumer.signalAll();//通知消费者
}catch (Exception e) {
e.printStackTrace();
}
finally {
lock.unlock();
}
}
public void get() {
lock.lock();
try {
// 必须使用while,如果使用if
// 如果正好多个consumer醒过来,他就不会继续再次判断size,而是直接remove,这是可能list为空
// 就会出现异常:Exception in thread "consumer - 9" java.util.NoSuchElementException
while(list.size()==0) {
System.out.println("消费者等待...");
consumer.await();
}
Thread.sleep(2);
System.out.println("get");
list.removeFirst();
producer.signalAll();
} catch (InterruptedException e) {
e.printStackTrace();
} finally {
lock.unlock();
}
}
public static void main(String[] args) {
TestCondition contation = new TestCondition();
//10个消费者
for(int i = 0;i<10;i++) {
new Thread(()->{//jdk8新特性
for(int j = 0;j<2;j++) {
contation.get();
}
},"consumer - "+i).start();;
}
//睡一会
try {
TimeUnit.SECONDS.sleep(1);
} catch (InterruptedException e) {
e.printStackTrace();
}
//2个生产者
for(int i = 0;i<2;i++) {
new Thread(()->{
for(int j = 0;j<10;j++) {
contation.put(new Object());
}
},"producer - "+i).start();;
}
}
}