Java借助CountDownLatch完成异步回调

Posted 2020-10-04 张伯雨

public class AsyncDemo {

    private static void doSomeTask() {
        System.out.println("Hello World");
    }

    private static void onCompletion() {
        System.out.println("All tasks finished");
    }

    public static void main(String[] args) {
        ExecutorService executor = Executors.newCachedThreadPool();
        final CountDownLatch latch = new CountDownLatch(2);

        executor.execute(new Task(latch));
        executor.execute(new Task(latch));

        executor.execute(() -> {
            try {
                latch.await();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            onCompletion();
        });
        executor.shutdown();
    }

    private static class Task implements Runnable {

        /**
         * CountDownLatch 是JDK提供的一个简单的线程监测工具
         * 基于简单的计数，调用countDown()方法表明当前线程已经终止
         * 在监测线程中调用await()方法,该方法会一直挂起直到所有其它线程终止
         */
        private final CountDownLatch latch;

        public Task(CountDownLatch latch) {
            this.latch = latch;
        }

        @Override
        public void run() {
            try {
                doSomeTask();
            } catch (Exception e) {
                e.printStackTrace();
            } finally {
                latch.countDown();
            }
        }
    }
}

这里有两点需要补充：

1.如果你是用main方法启动的线程，这种调用方法是没有问题的，JDK会确保所有线程都终止以后main方法才退出。但是如果main方法不是异步任务的启动者（如JUnit,Spring,Tomcat），一旦启动之后laucher将会失去对线程的控制。如在JUnit中laucher提交完任务后就会被认为所有过程已完成，其它线程会被强行终止。

2.正因为如此，请根据环境使用正确的Executor。比如，在web环境中，应该选用tomcat（或Spring）管理的线程池作为Executor，这样才能确保web应用对于异步任务的整个生命周期具有控制权；如果你选用JDK的线程池有什么后果呢？任务也许可以正常执行，当一旦你终止web-app，正在执行的异步线程并不会被正常kill掉，并由此造成内存泄漏或其它不可预见的后果。

Executors.newCachedThreadPool() 中的线程是 isDaemon true.
我翻到了 13 年也研究了一下 CountDownLatch 协调线程 的使用 http://unmi.cc/countdownlatch-threads/。
CountDownLatch 是 Java 1.5 加入的 API, 到了 Java 8 我会选择使用 CompletableFuture， 它要等待所有线程是否完成可以简单调用
CompletableFuture.allOf(CompletableFuture...completableFutures)
就是其他框架的 waitAll() 的机制。