【Java】线程池（四）关闭线程池

如何优雅的关闭线程池？

其实在java源代码注释里有：大致是先调用shutdown方法，然后再调用awaitTermination等待正在执行的任务的结束。如果在一段时间内仍未结束，再调用shutdownnow方法强制停止。

guava的线程池库中有一个实现：

MoreExecutors::shutdownAndAwaitTermination方法。

    public static boolean shutdownAndAwaitTermination(ExecutorService service, long timeout, TimeUnit unit) {
        long halfTimeoutNanos = unit.toNanos(timeout) / 2L;
        service.shutdown();

        try {
            if (!service.awaitTermination(halfTimeoutNanos, TimeUnit.NANOSECONDS)) {
                service.shutdownNow();
                service.awaitTermination(halfTimeoutNanos, TimeUnit.NANOSECONDS);
            }
        } catch (InterruptedException var7) {
            Thread.currentThread().interrupt();
            service.shutdownNow();
        }

        return service.isTerminated();
    }

该方法是一个static方法，入参是一个线程池实例，等待时间t以及时间单位。

实现是先等待t/2，如果仍未结束则强制结束，再等剩余的t/2时间。

那么这个方法该在哪里调用呢？

我们的程序在正常退出前，如果使用了线程池，那么需要在退出前等待线程池执行结束。程序的退出是因为外部的信号导致，我们可以在退出信号上的hook里加上上述方法的调用。

例子：

    public static void main(String args[]) {
        ExecutorService executorService = Executors.newCachedThreadPool();
        executorService.submit(() -> {
            for (int i = 0; i < 10; i ++) {
                System.out.println(i);
                try {
                    Thread.sleep(2000);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        });
        Runtime.getRuntime().addShutdownHook(new Thread(() -> {
            System.out.println("enter hook");
            MoreExecutors.shutdownAndAwaitTermination(executorService, 20, TimeUnit.SECONDS);
        }));
    }

使用java的Runtime的addShutdownHook方法即可，该方法会在jvm进程退出前调用注册的hook。当然这里针对正常退出，如果是断电或者-9，则hook无法被调用，这也是不推荐使用-9结束服务器进程的原因。

打印如下：

0
1
enter hook
2
3
4
5
6
7
java.lang.InterruptedException: sleep interrupted
    at java.lang.Thread.sleep(Native Method)
    at com.liyao.s.Main2.lambda$main$0(Main2.java:16)
    at java.util.concurrent.Executors$RunnableAdapter.call(Executors.java:511)
    at java.util.concurrent.FutureTask.run(FutureTask.java:266)
    at java.util.concurrent.ThreadPoolExecutor.runWorker(ThreadPoolExecutor.java:1142)
    at java.util.concurrent.ThreadPoolExecutor$Worker.run(ThreadPoolExecutor.java:617)
    at java.lang.Thread.run(Thread.java:745)
8
9

Process finished with exit code 130 (interrupted by signal 2: SIGINT)

在打印2时关闭的进程，后面打印了3-7，正好是10s，此时已经等了t/2时间，所以后面发出了shutdownnow，可以看到sleep的进程被中断了。但是后面的8和9还是被打印出了，因为例子里提交的任务并不是整体相应中断信号的。由此也说明，我们在线程池里提交的任务尽量可以响应中断信号。