wait/notify模拟queu

使用wait/notify模拟queue

BlockingQueue：顾名思义，首先它是一个队列，并且支持阻塞的机制，阻塞的放入和得到数据。我们要实现LinkedBlockingQueue，下面两个简单的方法put和take。

put（anObject）：把anObject加到BlockingQueue，如果BlockingQueue中没有空间，则调用此方法的线程被阻断，直到BlockingQueue里有空间再继续。

take：取走BlockingQueue里排在首位的对象，若BlockingQueue为空，阻断进入等待状态直到BlockingQueue有新的数据被加入

package com.java.day02_notify_wait;

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

/**
 * 用wait和notify模仿queue
 * 
 * @author syousetu
 *
 */
public class MyQueue {
    // 先创建一个容器，有上限和下限

    // 1.创建一个列表
    private volatile LinkedList<Object> list = new LinkedList<Object>();

    // 2.设置列表的上限
    private final int max;

    // 3.设置列表的下限
    private int min = 0;

    // 构造器
    public MyQueue(int len) {
        this.max = len;
    }

    // 计数器
    private AtomicInteger count = new AtomicInteger(0);

    // 要使用到synchronized 所以要创建一个锁
    private final Object lock = new Object();

    // put(Object)

    public void put(Object obj) {
        synchronized (lock) {
            // 队列队伍已满，请等待
            while (count.get() == max) {
                try {
                    lock.wait();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }

            // 给列表添加元素
            list.add(obj);
            // 计数器增加一
            count.incrementAndGet();
            // 通知另外一个线程可以进行取元素
            lock.notify();
            System.out.println("已成功添加元素：" + obj);

        }

    }

    // take()

    public Object take() {
        Object obj = null;

        synchronized (lock) {
            // 队列里没有元素，请等待，释放锁
            while (count.get() == min) {
                try {
                    lock.wait();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }

            // 移除第一个元素
            obj = list.removeFirst();
            // 通知另一个线程可以进行存放
            lock.notify();
            // 计数器减一
            count.decrementAndGet();
            System.out.println("已成功取出元素");

        }

        return obj;
    }

    public int getSize() {
        return list.size();
    }

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

        Thread t1  = new Thread(new Runnable() {
            public void run() {
                mq.put("f");
                mq.put("g");
            }
        },"t1");
        
        
        Thread t2  = new Thread(new Runnable() {
            public void run() {
                for (int i = 0; i < 3; i++) {
                    System.out.println(Thread.currentThread().getName()+"线程取出元素："+mq.take());
                }
                
            }
        });
        
        t1.start();
        t2.start();
        
    }

}

运行结果：

已成功添加元素：a
已成功添加元素：b
已成功添加元素：c
已成功添加元素：d
已成功添加元素：e
当前队列的大小：5
已成功取出元素
Thread-0线程取出元素：a
已成功添加元素：f
已成功取出元素
Thread-0线程取出元素：b
已成功添加元素：g
已成功取出元素
Thread-0线程取出元素：c