手写一个简单的消息队列实现

2020-01-13

本文作者：FUNKYE(陈健斌),杭州某互联网公司主程。

DEMO

public class TestQueue {
    static ArrayList<Object> objects = new ArrayList<Object>(10);

    void add(Object msg) {
        synchronized (TestQueue.class) {
            if (objects.size() == 10) {
                System.out.println(Thread.currentThread().getName() + "队列满了,等待取");
                try {
                    TestQueue.class.wait();
                } catch (InterruptedException e) {
                    // TODO Auto-generated catch block
                    e.printStackTrace();
                }
            }
            System.out.println(Thread.currentThread().getName() + "存入:" + msg);
            objects.add(msg);
            TestQueue.class.notifyAll();
        }

    }

    void get() {
        synchronized (TestQueue.class) {
            if (objects.size() == 0) {
                System.out.println(Thread.currentThread().getName() + "队列空了,等待存");
                try {
                    TestQueue.class.wait();
                } catch (InterruptedException e) {
                    // TODO Auto-generated catch block
                    e.printStackTrace();
                }
                System.out.println(Thread.currentThread().getName() + "读取线程被唤醒");
            }
            System.out.println(Thread.currentThread().getName() + "取出:" + objects.get(0));
            objects.remove(0);
            TestQueue.class.notifyAll();
        }
    }
    public static void main(String[] args) {
        TestQueue testQueue = new TestQueue();
        new Thread(() -> {
            for (int i = 0; i < 10; i++) {
                testQueue.add(i);
            }
        }, "存储线程").start();
        new Thread(() -> {
            for (int i = 0; i < 10; i++) {
                testQueue.get();
            }
        }, "读取线程").start();
    }
}

查看日志输出:

存储线程存入:0
读取线程取出:0
读取线程队列空了,等待存
存储线程存入:1
存储线程存入:2
存储线程存入:3
存储线程存入:4
存储线程存入:5
存储线程存入:6
存储线程存入:7
存储线程存入:8
存储线程存入:9
读取线程读取线程被唤醒
读取线程取出:1
读取线程取出:2
读取线程取出:3
读取线程取出:4
读取线程取出:5
读取线程取出:6
读取线程取出:7
读取线程取出:8
读取线程取出:9

分析原理

其实非常简单,用锁锁住存取代码块,如果没有可取的数据,读取数据的线程进入等待状态,存储线程存完数据后会唤醒读取线程进行读取,如果存储线程瞬间就存满了也会进入等待状态,等待读取线程的读取后唤醒他.