Java多线程之模拟一个阻塞队列

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

public class MyQueue {


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

    //需要一个计数器，统计加入List集合元素的个数

    private AtomicInteger count = new AtomicInteger(0);

    //需要指定上限和下限

    private int minSize = 0;
    private int maxSize ;
    //创建queue时指定最大长度
    public MyQueue(int size){
        this.maxSize=size;
    }
    //初始化一个对象，用于加锁
    private Object lock = new Object();

    public void put(Object obj){
        synchronized (lock){
            while(count.get() == this.maxSize){
                try {
                    lock.wait();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
            list.add(obj);
            //计数器加一
            count.incrementAndGet();
            System.out.println("新添加的元素为："+obj);
            //通知其他线程
            lock.notify();
        }

    }
    public Object take(){
        Object ret =null;
        synchronized (lock){
                //注意这里不使用list.getSize()
            while(count.get()==this.minSize){
                try {
                    lock.wait();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
            //移除元素并返回
            ret =list.removeFirst();
            count.decrementAndGet();
            lock.notify();
        }

        return ret ;
    }
    public int getSize(){

        return this.count.get();
    }

    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());
        new Thread(new Runnable() {
            @Override
            public void run() {
                //此时容器已经满了,会阻塞
                mq.put("f");
                mq.put("g");
                System.out.println(Thread.activeCount());
            }
        }).start();
        // 休眠两秒钟
        try {
            Thread.sleep(2000);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        new Thread(new Runnable() {
            @Override
            public void run() {
                //从容器里面拿元素，同时唤醒添加元素的线程
                Object o1= mq.take();
                System.out.println("当前移除的元素"+o1);
                Object o2= mq.take();
                System.out.println("当前移除的元素"+o2);
            }
        }).start();
    }
}

打印结果为

新添加的元素为：a
新添加的元素为：b
新添加的元素为：c
新添加的元素为：d
新添加的元素为：e
当前容器的长度：5
//这里停顿两秒钟，说明第一个添加元素的线程还没有执行，在第二个线程执行完后，才会执行第一个线程
新添加的元素为：f
当前移除的元素a
当前移除的元素b
新添加的元素为：g
3