生产者消费者

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();;
		}
	}
}