背景
利用kafka进行消息生产和消费,生产线程或消费线程在tomcat关闭的时候无法优雅的停止,一定要强制退出才行。很不体面。
Runtime.addShutdownHook解释
如果你想在jvm关闭的时候
进行内存清理、对象销毁等操作,或者仅仅想起个线程然后这个线程不会退出,你可以使用Runtime.addShutdownHook。
这个方法的作用就是在JVM中增加一个关闭的钩子。当程序正常退出、系统调用 System.exit方法或者虚拟机被关闭时才会执行系统中已经设置的所有钩子,当系统执行完这些钩子后,JVM才会关闭
。所谓钩子,就是一个已初始化但并不启动的线程。JVM退出通常通过两种事件。
- 程序正常退出,例如最后一个非守护进程退出、使用System.exit()退出等
- 程序异常退出,例如使用Ctrl+C触发的中断、用户退出或系统关闭等系统事件等 该方法的说明如下,详细说明参见官方文档。
void java.lang.Runtime.addShutdownHook(Thread hook)
Parameters:
hook - An initialized but unstarted Thread object
Throws:
IllegalArgumentException - If the specified hook has already been registered, or if it can be determined that the hook is already running or has already been run
IllegalStateException - If the virtual machine is already in the process of shutting down
SecurityException - If a security manager is present and it denies RuntimePermission("shutdownHooks")
kafka代码示例
在start的时候加个钩子即可
Runtime.getRuntime().addShutdownHook(new Thread() {
@Override
public void run() {
KafkaConsumer.this.close();
}
});