CountDownLatch 实现主线程规定时间内返回结果

业务场景

阿里巴巴的一道面试题目

一个搜索引擎，前端发送一个关键字，后端那这个关键字返回搜狗、百度、谷歌3个搜索引擎的搜索结果。要求，200ms内返回结果，否则此次认为请求失败。

首先，主线程开三个线程分别取请求这三个引擎的搜索结果，存入对应的集合中。主线程等待200ms后，直接返回结果。

public class Main{
    public static void main(String[] args) throws InterruptedException {
        long start = System.currentTimeMillis();

        new Thread(() -> {
            try {
                Thread.sleep(50);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }

        }, "SouGou").start();
        new Thread(() -> {
            try {
                Thread.sleep(50);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }

        }, "google").start();
        new Thread(() -> {
            try {
                Thread.sleep(50);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }, "Baidu").start();
        Thread.sleep(150);
        long end = System.currentTimeMillis();
        System.out.println(end - start);
    }
}

这样的写法，业务要求已经实现，但是我们发现，这样的话，每次请求都会花费200ms的时间。不管3个搜索引擎线程花费多少时间。如果我们的3个搜索线程在200ms之前完成，我们的主线程需要白白等待一段时间。这样是非常不合理的。这时候可以用我们的第二种写法（CountDownLatch）版本：

import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;

public class Main{
    public static void main(String[] args) throws InterruptedException {
        long start = System.currentTimeMillis();
        CountDownLatch countDownLatch = new CountDownLatch(3);
        new Thread(() -> {
            try {
                Thread.sleep(50);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            countDownLatch.countDown();
        }, "SouGou").start();
        new Thread(() -> {
            try {
                Thread.sleep(50);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            countDownLatch.countDown();
        }, "google").start();
        new Thread(() -> {
            try {
                Thread.sleep(50);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            countDownLatch.countDown();
        }, "Baidu").start();
        countDownLatch.await(150, TimeUnit.MILLISECONDS);
        long end = System.currentTimeMillis();
        System.out.println(end - start);
    }
}

CountDownLatch 可以实现一个记数线程等待的功能。传入的初始值为3，countDownLatch.countDown();每次可以使这个值减一。当这个值减到0的时候，就会直接结束countDownLatch.await(150, TimeUnit.MILLISECONDS)方法，使主线程继续运行。