Java--多线程之生产者消费者模型

package com.yuanlief;

public class Main {

    public static void main(String[] args) {
        ShareData s = new ShareData();//create a instance
        new Consumer(s).start();//use created that instance
        new Producer(s).start();//synchronized use created that instance
    }
}

//共享数据类
class MyData {
    public int data;
}

//共享数据控制类
class ShareData {
    private MyData data;
    private boolean writeable = true;

    public synchronized void setShareData(MyData data){
        if(!writeable){
            try {
                wait();//如果没有消费就等待
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }

        this.data = data;
        writeable = false;//标记已经生产
        notify();//通知消费者已经生产，可以消费了
    }
    public synchronized MyData getShareData() {
        if (writeable){
            try {
                wait();//如果没有生产就等待
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
        writeable = true;//标记已经消费
        notify();//通知需要生产
        return this.data;
    }
}

/**
 * 生产者线程类
 */
class Producer extends Thread {
    private ShareData s;
    Producer(ShareData s){
        this.s = s;
    }

    @Override
    public void run() {
        for (int i = 1;i <= 10;i++){
            try {
                Thread.sleep((int) (Math.random() * 10));//每增加一个数就睡一会，好让另一个线程处理
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            MyData myData = new MyData();
            myData.data = i;
            s.setShareData(myData);
            System.out.print("生产者产生一条数据：" + myData.data + ".\n");
        }
    }
}

/**
 * 消费者线程类
 */

class Consumer extends Thread {
    private ShareData s;

    Consumer(ShareData s) {
        this.s = s;
    }

    @Override
    public void run() {
        MyData myData;
        do {
            try {
                Thread.sleep((int) (Math.random() * 10));
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            myData = s.getShareData();
            System.out.print("消费者消费一条数据：" + myData.data + ".\n");
        }while (myData.data <= 10);
    }
}