面试锦集：手写一个同步类容器之Lock/Condition await signalAll实现

互联网大厂面试必考题目：

手写一个同步类容器，支持多个生产者线程以及多个消费者线程的阻塞调用。采用 Lock/Condition await signalAll实现，更精确定位唤起的线程。我第一次去大厂面试都被难住了！！！

package com.container;

import java.util.LinkedList;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.concurrent.locks.Condition;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;

/**
 * 面试锦集:手写一个同步类容器(互联网大厂面试题目)
 * 
 * 支持多个生产者线程以及多个消费者线程的阻塞调用，采用Lock/Condition await signalAll实现 更精确定位唤起的线程
 * 
 * @author 小辉GE/小辉哥
 * 
 * 2019年8月8日 下午20:30:00
 */
public class ContainerReentrantLock {
	// 最少0个元素
	private final int MIN_SIZE = 0;
	// 最多20个元素
	private final int MAX_SIZE = 20;
	// 定义容器size的计算器
	private final AtomicInteger atomic = new AtomicInteger(0);
	// 定义装载容器
	private final LinkedList<Object> container = new LinkedList<Object>();
	// 定义Lock
	private Lock lock = new ReentrantLock();
	// 定义Condition
	private Condition producer = lock.newCondition();
	private Condition consumer = lock.newCondition();

	public void put(Object obj) {
		try {
			lock.lock();
			while (atomic.get() == MAX_SIZE) {
				producer.await();
			}
			// 添加元素
			container.add(obj);
			// 计数器
			atomic.incrementAndGet();
			// 通知所有消费者线程进行消费
			consumer.signalAll();
		} catch (InterruptedException e) {
			e.printStackTrace();
		} finally {
			lock.unlock();
		}
	}

	public Object get() {
		Object result = null;
		try {
			lock.lock();
			while (atomic.get() == MIN_SIZE) {
				consumer.await();
			}
			result = container.removeFirst();
			atomic.decrementAndGet();
			// 通知所有生产者线程进行生产
			producer.signalAll();
		} catch (InterruptedException e) {
			e.printStackTrace();
		} finally {
			lock.unlock();
		}
		return result;
	}

	public static void main(String[] args) {
		ContainerReentrantLock lock = new ContainerReentrantLock();
		// 启动消费者线程
		for (int i = 0; i < 10; i++) {
			new Thread(() -> {
				for (int j = 0; j < 10; j++) {
					System.out.println("消费者" + Thread.currentThread().getName() + "，获取数据为：" + lock.get());
				}
			}, "consumer" + i).start();
		}

		try {
			TimeUnit.SECONDS.sleep(1);
		} catch (InterruptedException e) {
			e.printStackTrace();
		}

		// 启动生产者线程
		for (int i = 0; i < 5; i++) {
			new Thread(() -> {
				for (int j = 0; j < 20; j++) {
					int data = (int) (Math.random() * 1000);
					lock.put(data);
					System.out.println("生产者" + Thread.currentThread().getName() + "，存放数据为：" + data);
				}
			}, "producer" + i).start();
		}
	}
}

测试输出结果如下（输出结果比较多，我们只截取部分）：

从结果可以看出生产者和消费者线程是交替输出的。关于Lock/Condition await signalAll实现方式还不明白的同学，详情可以在我的头条号‘小辉GE’查看对应‘手写一个同步类容器(互联网大厂面试题目)’视频，里面有详细讲解。

以上代码仅供参考，如有不当之处，欢迎指出！！！

更多干货，欢迎大家关注和联系我。期待和大家一起更好的交流、探讨技术！！！