import java.util.concurrent.atomic.AtomicInteger;
/**
 * @author lhd
 */
public class BlockQueue {
	/**
  * 生產(chǎn)者鎖對(duì)象
  */
	private final Object addLock = new Object();
	/**
  * 消費(fèi)者鎖對(duì)象
  */
	private final Object deleteLock = new Object();
	/**
  * 隊(duì)列總大小
  */
	private final Integer size = 30;
	/**
  * 數(shù)據(jù)存放
  */
	private Object[] queue = new Object[size];
	/**
  * 存放的數(shù)量,使用AtomicInteger是因?yàn)槠胀ǖ膇nt遞增遞減操作會(huì)存在非原子性的問題，會(huì)使數(shù)量異常
  */
	private AtomicInteger count = new AtomicInteger(0);
	/**
  * 生產(chǎn)
  * @param o 對(duì)象
  */
	public void add(Object o) {
		//獲取生產(chǎn)鎖,wait方法必須獲取到對(duì)象鎖后才可以調(diào)用，否則拋出異常
		synchronized (addLock){
			//判斷是否超過隊(duì)列大小,超過則進(jìn)入等待
			while (count.get() >= size){
				try {
					addLock.wait();
				}
				catch (InterruptedException e) {
					e.printStackTrace();
				}
			}
			//存放一個(gè)
			queue[count.get()] = o;
			//遞增
			int i = count.incrementAndGet();
			//打印一下日志
			String name = Thread.currentThread().getName();
			System.out.println(name + "生產(chǎn)了一個(gè)，現(xiàn)有數(shù)量" + i);
		}
		//如果隊(duì)列有數(shù)據(jù)，則調(diào)用notifyAll喚醒消費(fèi)者
		if (count.get() >= 1){
			//notifyAll、notify都需要先獲取對(duì)象鎖,否則會(huì)拋出異常
			synchronized (deleteLock){
				deleteLock.notifyAll();
			}
		}
	}
	/**
  * 消費(fèi)
  * @return
  */
	public Object poll(){
		Object o;
		//先獲取對(duì)象鎖，和生產(chǎn)者類似
		synchronized (deleteLock){
			//隊(duì)列里沒有數(shù)據(jù)則等待
			while (count.get() <= 0){
				try {
					deleteLock.wait();
				}
				catch (InterruptedException e) {
					e.printStackTrace();
				}
			}
			//獲取數(shù)據(jù)
			o = queue[count.get()];
			//遞減
			int i = count.decrementAndGet();
			String name = Thread.currentThread().getName();
			System.out.println(name + "消費(fèi)了一個(gè)，現(xiàn)有數(shù)量" + i);
		}
		//如果隊(duì)列沒有滿，則可以喚醒生產(chǎn)者
		if (count.get() < size){
			//需要先獲取到鎖
			synchronized (addLock){
				addLock.notifyAll();
			}
		}
		return o;
	}
	/**
  * 簡單的測試
  * @param args
  */
	public static void main(String[] args) {
		BlockQueue blockQueue = new BlockQueue();
		Thread t1 = new Thread(()-> {
			while (true){
				blockQueue.add(new Object());
			}
		}
		);
		Thread t2 = new Thread(()-> {
			while (true){
				blockQueue.add(new Object());
			}
		}
		);
		Thread t3 = new Thread(()-> {
			while (true){
				blockQueue.add(new Object());
			}
		}
		);
		Thread t4 = new Thread(()-> {
			while (true){
				blockQueue.poll();
			}
		}
		);
		Thread t5 = new Thread(()-> {
			while (true){
				blockQueue.poll();
			}
		}
		);
		Thread t6 = new Thread(()-> {
			while (true){
				blockQueue.poll();
			}
		}
		);
		t1.start();
		t2.start();
		t3.start();
		t4.start();
		t5.start();
		t6.start();
	}
}

/**
 * @author lhd
 */
public class JoinTest {


 public static void main(String[] args) throws InterruptedException {
  Thread t1 = new Thread(() -> System.out.println(1));
  Thread t2 = new Thread(()-> System.out.println(2));
  Thread t3 = new Thread(()-> System.out.println(3));

  t1.start();
  t1.join();

  t2.start();
  t2.join();

  t3.start();
  t3.join();
 }
}