Java借助CountDownLatch完成异步回调

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掉，并由此形成内存泄漏或其它不可预见的后果。