如何优雅的关闭线程池?
其实在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还是被打印出了,因为例子里提交的任务并不是整体相应中断信号的。由此也说明,我们在线程池里提交的任务尽量可以响应中断信号。