手写生产者消费者模式

四个类

• 仓库
• 生产者
• 消费者
• 测试类

1. 生产者

import java.util.Date;
import java.util.LinkedList;

public class Storage {
    private Integer maxSize;
    private LinkedList<Date> queue;

    public Storage() {
        maxSize = 10;
        queue = new LinkedList<>();
    }

    public synchronized void put() {
        //  如果满了，就等待
        while (queue.size() == maxSize) {
            try {
                wait();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
        // 如果没满，就继续往里放
        queue.offer(new Date());
        System.out.println("仓库里有了 " + queue.size() + " 个产品");
        // 通知消费者
        notify();
    }

    public synchronized void take() {
        // 如果为空，就等待
        while (queue.isEmpty()) {
            try {
                wait();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
        // 如果非空就往外拿
        System.out.println("拿到了 " + queue.poll() + "，仓库还剩下 " + queue.size() + " 个产品");
        // 通知生产者
        notify();
    }
}

2. 生产者

public class Producer implements Runnable {

    private Storage storage;

    public Producer(Storage storage) {
        this.storage = storage;
    }

    @Override
    public void run() {
        for (int i = 0; i < 100; i++) {
            try {
                Thread.sleep(50);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            storage.put();
        }
    }
}

3. 消费者

public class Consumer implements Runnable {

    private Storage storage;

    public Consumer(Storage storage) {
        this.storage = storage;
    }

    @Override
    public void run() {
        for (int i = 0; i < 100; i++) {
            storage.take();
        }
    }
}

4. 测试类

public class Test {
    public static void main(String[] args) {
        Storage storage = new Storage();
        Consumer consumer = new Consumer(storage);
        Producer producer= new Producer(storage);
        new Thread(consumer).start();
        new Thread(producer).start();
    }
}