public class Test {
    public static void main(String[] args){
        //創建一個阻塞隊列,邊界為1
        BlockingQueue<String> queue = new ArrayBlockingQueue<String>(1);
        new Thread(new PThread(queue)).start();
        new Thread(new CThread(queue)).start();
    }
}

/**
 * 生產者
 */
class PThread implements Runnable {
    private BlockingQueue<String> queue;
    private AtomicInteger count = new AtomicInteger();

    public PThread(BlockingQueue<String> queue) {
        this.queue = queue;
    }

    @Override
    public void run() {
        while (true) {
            String data = count.getAndIncrement() + "";
            System.out.println("生產數據:"+data);
            try {
                queue.put(data);
                Thread.sleep(2000);//sleep,便於觀察
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }

    }
}

/**
 * 消費者
 */
class CThread implements Runnable {
    private BlockingQueue<String> queue;

    public CThread(BlockingQueue<String> queue) {
        this.queue = queue;
    }

    @Override
    public void run() {
        while (true) {
            try {
                String data = queue.take();
                System.out.println("消費數據:"+data);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }
}
 

因為ArrayBlockingQueue是一個阻塞的有界隊列,可以直接使用它.

多線程模擬生產者消費者示例之BlockQuue