package com.gf.conn009;

import java.util.LinkedList;

import java.util.concurrent.atomic.AtomicInteger;

/**

 * 模拟Queue

 * @author huanchu

 *

 */

public class MyQueue {

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

	private final AtomicInteger count = new AtomicInteger(0);

	private final int maxSize;

	private final int minSize = 0;

	private final Object lock = new Object();

	public MyQueue(int maxSize){

		this.maxSize = maxSize;

	}

	public void put(Object obj){

		synchronized (lock) {

			while (count.get() == maxSize) {

				try {

					lock.wait();

				} catch (InterruptedException e) {

					e.printStackTrace();

				}

			}

			list.add(obj);

			count.incrementAndGet();

			System.out.println(" 元素 " + obj + " 被添加");

			lock.notify();

		}

	}

	public Object take(){

		Object temp = null;

		synchronized (lock) {

			while (count.get() == minSize) {

				try {

					lock.wait();

				} catch (InterruptedException e) {

					e.printStackTrace();

				}

			}

			count.decrementAndGet();

			temp = list.removeFirst();

			System.out.println(" 元素" + temp + "被消费");

			lock.notify();

		}

		return temp;

	}

	public int size(){

		return count.get();

	}

	public static void main(String[] args) throws InterruptedException {

		final MyQueue m = new MyQueue(5);

		m.put("a");

		m.put("b");

		m.put("c");

		m.put("d");

		m.put("e");

		System.out.println("当前元素的个数：" + m.size());

		Thread t1 = new Thread(new Runnable() {

			@Override

			public void run() {

				m.put("h");

				m.put("i");

			}

		},"t1");

		Thread t2 = new Thread(new Runnable() {

			@Override

			public void run() {

				try {

					Thread.sleep(1000);

					Object t1 = m.take();

					Thread.sleep(1000);

					Object t2 = m.take();

				} catch (InterruptedException e) {

					e.printStackTrace();

				}

			}

		},"t2");

		t1.start();

		Thread.sleep(1000);

		t2.start();

	}

}