Java如何實(shí)現(xiàn)簡易生產(chǎn)者消費(fèi)者模型

package test.exception.producer_consumer_model;

/*
假設(shè)為產(chǎn)品為筆
 */

public class Production {
  private String type = "";
  private String color = "";
  private long code = 0; // 產(chǎn)品編號
  private boolean isProduced = false; // 是否生產(chǎn)完成 初始狀態(tài)為未生產(chǎn)狀態(tài)
  private boolean isContinueProduce = true; // 是否停產(chǎn)該產(chǎn)品

  public void setContinueProduce(boolean continueProduce) {
    isContinueProduce = continueProduce;
  }

  public void setCode(long code) {
    this.code = code;
  }

  public Production(){
  }

  public boolean isContinueProduce() {
    return isContinueProduce;
  }

  public void setType(String type) {
    this.type = type;
  }

  public void setColor(String color) {
    this.color = color;
  }

  public void setProduced(boolean produced) {
    isProduced = produced;
  }

  public boolean isProduced() {
    return isProduced;
  }

  @Override
  public String toString() {
    return color + type + "-" + code;
  }
}

package test.exception.producer_consumer_model;

public class Producer implements Runnable {
  private final Production pen; // 產(chǎn)品

  public Producer(Production pen) {
    this.pen = pen;
  }

  // 生產(chǎn)
  public void produce() {
    long code = 0;
    while (this.pen.isContinueProduce()) {
      synchronized (this.pen) {
        if (this.pen.isProduced()) {
          try {
            this.pen.wait(); // 等待消費(fèi)者消費(fèi)
          } catch (InterruptedException e) {
            e.printStackTrace();
          }
        }
        // 開始生產(chǎn)
        this.pen.setType("鉛筆");
        this.pen.setColor("藍(lán)色");
        this.pen.setCode(code++);
        this.pen.setProduced(true);
        System.out.println(this.pen + " is produced");
        this.pen.notify();
      }
    }
    System.out.println("finish producing");
  }
  @Override
  public void run() {
    produce();
  }
}

package test.exception.producer_consumer_model;

public class Consumer implements Runnable {
  private final Production pen;

  public Consumer(Production pen) {
    this.pen = pen;
  }

  // 持續(xù)消費(fèi)
  public void consumer() {
    while (this.pen.isContinueProduce()) {
      synchronized (this.pen) {
        if (!this.pen.isProduced()) {
          try {
            this.pen.wait(); // 等待生產(chǎn)者生產(chǎn)
          } catch (InterruptedException e) {
            e.printStackTrace();
          }
        }

        System.out.println(this.pen + " is consumed"); // 使用
        this.pen.setProduced(false); // 使用完后更新狀態(tài)
        this.pen.notify();
      }
    }
    // 確保停止生產(chǎn)后，能夠使用最后生產(chǎn)的一支筆
    if (this.pen.isProduced()) {
      System.out.println(this.pen + " is consumed");
    }

    System.out.println("finish using");
  }

  @Override
  public void run() {
    consumer();
  }
}

package test.exception.producer_consumer_model;

public class Demo {
  public static void main(String[] args) throws InterruptedException {
    Production pen = new Production();
    Consumer consumer = new Consumer(pen);
    Producer producer = new Producer(pen);
    new Thread(producer).start(); // 開啟生產(chǎn)者線程
    new Thread(consumer).start(); // 開啟消費(fèi)者線程

    Thread.sleep(10000);
    pen.setContinueProduce(false); // 10s后停止生產(chǎn)該類型的筆
  }
}