自由手記

檔案下載:39238473

import java.util.ArrayList;
import java.util.List;

/**
* 通過wait和notify實現的生產者消費者demo
* User: zhangb
* Date: 12-12-1
* Time: 下午7:19
*/
public class ProducerAndCustomerDemo {
    private static int capacity = 150;
    private static List<String> basket = new ArrayList<String>(capacity);

    public static void main(String[] args) {
        // 多生產者與多消費者
        int producerSize = 2;
        Thread[] ps = new Thread[producerSize];
        for (int i = 0, step = 500; i < producerSize; i ++) {
            ps[i] = new Thread(new Producer((i) * step, (i+1) * step), "生產-->線程--" + (i+1));
            ps[i].start();
        }

        int customerSize = 10;
        Thread[] cs = new Thread[customerSize];
        for (int i = 0; i < customerSize; i ++) {
            cs[i] = new Thread(new Customer(), "消費線程--" + (i+1));
            cs[i].start();
        }

        // 等待生產線程結束並中斷消費線程
        for (int i = 0; i < producerSize; i ++) {
            try {
                ps[i].join();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
        for (int i = 0; i < customerSize; i ++) {
            cs[i].interrupt();
        }
    }

    static class Producer implements Runnable {
        private int start;
        private int end;

        Producer(int start, int end) {
            this.start = start;
            this.end = end;
        }

        @Override
        public void run() {
            for (int i = start; i < end; i ++) {
                synchronized (basket) {
                    try {
                        while (basket.size() == capacity) {
                            basket.wait();
                        }
                        String p = " PRO" + i;
                        System.out.println(Thread.currentThread().getName() + p);
                        basket.add(p);
                        basket.notifyAll();
                        Thread.yield(); // 讓出當前線程的執行權,有利於看出交替線程運行的效果
                    } catch (InterruptedException e) {
                        e.printStackTrace();
                        break;
                    }
                }
            }
        }
    }

    static class Customer implements Runnable {
        @Override
        public void run() {
            while (true) {
                synchronized (basket) {
                    try{
                        while (basket.size() == 0) {
                            basket.wait();
                        }
                        System.out.println(Thread.currentThread().getName() + basket.remove(0));
                        basket.notifyAll();
                    } catch (InterruptedException e) {
                        System.out.println(Thread.currentThread().getName() + "退出");
                        break;
                    }
                }
            }
        }
    }
}