利用钩子优雅关停kafka

背景

利用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();
                }
            });