模拟生产者和消费者

1、使用多线程模拟生产者和消费者

class ShareData {private int number = 0;private Lock lock = new ReentrantLock();private Condition condition = lock.newCondition();public void increment() throws Exception {lock.lock();try {//多线程的判断必须要用whilewhile (number != 0) {//释放当前锁，进入等待状态，不能生产//唤醒线程从await()返回后需要重新获得锁condition.await();}//生产number++;System.out.println(Thread.currentThread().getName() + "\t" + number);//通知唤醒所有等待线程condition.signalAll();} catch (InterruptedException e) {e.printStackTrace();} finally {lock.unlock();}}public void decrement() throws Exception {lock.lock();try {//多线程的判断必须要用whilewhile (number != 1) {//释放当前锁，进入等待状态，不能消费//唤醒线程从await()返回后需要重新获得锁condition.await();}//消费number--;System.out.println(Thread.currentThread().getName() + "\t" + number);//通知唤醒所哟等待线程condition.signalAll();} catch (InterruptedException e) {e.printStackTrace();} finally {lock.unlock();}}
}public class ProducerConsumer {public static void main(String[] args) {ShareData shareData = new ShareData();new Thread(()->{for (int i = 0; i < 10; i++) {try {shareData.increment();} catch (Exception e) {e.printStackTrace();}}}, "A").start();new Thread(()->{for (int i = 0; i < 10; i++) {try {shareData.decrement();} catch (Exception e) {e.printStackTrace();}}}, "B").start();}
}

2、使用阻塞队列方式模拟生产者和消费者

class ShareResource {//默认开启，进行生产+消费private volatile boolean FLAG = true;private AtomicInteger atomicInteger = new AtomicInteger();BlockingQueue<String> blockingQueue = null;public ShareResource(BlockingQueue<String> blockingQueue) {this.blockingQueue = blockingQueue;System.out.println(blockingQueue.getClass().getName());}public void produce() throws InterruptedException {String data = null;boolean retValue;while (FLAG) {data = atomicInteger.incrementAndGet() + "";retValue = blockingQueue.offer(data, 2L, TimeUnit.SECONDS);if (retValue) {System.out.println(Thread.currentThread().getName() + "\t 插入队列" + data + "成功");} else {System.out.println(Thread.currentThread().getName() + "\t 插入队列" + data + "失败");}TimeUnit.SECONDS.sleep(1);}System.out.println(Thread.currentThread().getName() + "\tFLAG=false，生产结束");}public void consume() throws InterruptedException {String data = null;while (FLAG) {data = blockingQueue.poll(2L, TimeUnit.SECONDS);if (null == data || data.equalsIgnoreCase("")) {FLAG = false;System.out.println(Thread.currentThread().getName() + "\t 超过2秒钟没有取到数据，消费退出");System.out.println();return;}System.out.println(Thread.currentThread().getName() + "\t消费数据" + data + "成功");}}public void stop() {this.FLAG = false;}
}public class ProducerAndConsumer {public static void main(String[] args) throws InterruptedException {ShareResource shareResource = new ShareResource(new ArrayBlockingQueue<String>(10));new Thread(() -> {System.out.println(Thread.currentThread().getName() + "生产线程启动");try {shareResource.produce();} catch (InterruptedException e) {e.printStackTrace();}}, "Producer").start();new Thread(() -> {System.out.println(Thread.currentThread().getName() + "消费线程启动");try {shareResource.consume();} catch (InterruptedException e) {e.printStackTrace();}}, "Consumer").start();TimeUnit.SECONDS.sleep(5);System.out.println("5秒钟时间到，活动结束");shareResource.stop();}
}