一个朋友去面试,在CTO面试这一关了,遇到了一个手写程序题:第一个人开门阻塞; 第二个人开门成功 同时第一个人 也成功;第三个人、第四个人、以及后面所有人,直接开门成功。
考察点:写多线程、多线程通信、锁、double-check。
实现如下:
package thread;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;
/**
* <p>Title: 面试题:打开一个门,需要2个人才能打开,前2个人打开后,后续其他人不再需要重复打开。</p>
*
* <p>Description: </p>
*
* @author mahh
*
* @since:2018年11月22日 下午2:58:37
*
*/
public class Door {
/** 开门次数控制 */
private static final int OPEN_DOOR_TIME_SET = 2;
/** 记录开门次数 */
private AtomicInteger count = new AtomicInteger(0);
/** 开门锁,同一时间只有一个人真正开门。如果要支持分布式锁, 可以使用ZooKeeper实现接口Lock */
private Lock lock = new ReentrantLock();
/**
* @Description: 开门
* @param loopI
*/
public void openDoor(int loopI) {
if(isOpened()) { // 检查,如果已经开门了,不再开门
return;
}
if(loopI == 1 || loopI == 2) {
lock.lock();
try {
if(isOpened()) { // 检查,如果已经开门了,不再开门。再次检查
return;
}
// 模拟一次开门
System.out.println("============ 第 " + loopI + " 个人开门");
// 记录一次开门
count.addAndGet(1);
} finally {
lock.unlock();
}
}
// 如果门还没开,一直等待
while(!isOpened()) {
try {
Thread.sleep(50L);
} catch (InterruptedException e) {
throw new RuntimeException(e);
}
}
}
/**
* @Description: 门是否已打开。true:已打开。
*/
private boolean isOpened(){
return count.get() >= OPEN_DOOR_TIME_SET;
}
public static void main(String[] args) {
Door door = new Door();
// 模拟多线程场景
final int threadcnt = 20;
CountDownLatch cdl = new CountDownLatch(threadcnt); // 控制发起请求
for (int i = 1; i < threadcnt + 1; i++) {
final int loopI = i;
Thread t1 = new Thread(() -> {
// 多线程集合点,同时开始
try {
cdl.await();
} catch (InterruptedException e) {
e.printStackTrace();
}
door.openDoor(loopI);
System.out.println("终于开门了, " + loopI);
});
t1.start();
cdl.countDown();
}
}
}