spring-rabbitmq优雅关机

1、添加一个jvm关闭状态标识status

2、在rmq的消费者端代码创建aop切面，过滤onMessage方法，并添加判断status如果处于关闭状态则执行reject+requeue逻辑。（可以适当延时执行，防止消息循环通知）

3、注册个JVM关闭的钩子

这样的好处是防止业务处理过程中JVM关闭出现异常。

 

程序正常退出
使用System.exit()
终端使用Ctrl+C触发的中断
系统关闭
OutOfMemory宕机
使用Kill pid命令干掉进程（注：在使用kill -9 pid时，是不会被调用的）
下面是JDK1.7中关于钩子的定义：

 

    public void addShutdownHook(Thread hook)
参数：
    hook - An initialized but unstarted Thread object 
抛出： 
    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")
从以下版本开始： 
    1.3 
另请参见：
    removeShutdownHook(java.lang.Thread), halt(int), exit(int)

首先来测试第一种，程序正常退出的情况：

 

package com.hook;
 
import java.util.concurrent.TimeUnit;
 
public class HookTest
{
    public void start()
    {
        Runtime.getRuntime().addShutdownHook(new Thread(new Runnable() {
            @Override
            public void run()
            {
                System.out.println("Execute Hook.....");
            }
        }));
    }
    
    public static void main(String[] args)
    {
        new HookTest().start();
        System.out.println("The Application is doing something");
        
        try
        {
            TimeUnit.MILLISECONDS.sleep(5000);
        }
        catch (InterruptedException e)
        {
            e.printStackTrace();
        }
    }
}
运行结果：

 

 

The Application is doing something
Execute Hook.....
如上可以看到，当main线程运行结束之后就会调用关闭钩子。

 

下面再来测试第五种情况（顺序有点乱，表在意这些细节）：

 

package com.hook;
 
import java.util.concurrent.TimeUnit;
 
public class HookTest2
{
    public void start()
    {
        Runtime.getRuntime().addShutdownHook(new Thread(new Runnable() {
            @Override
            public void run()
            {
                System.out.println("Execute Hook.....");
            }
        }));
    }
    
    public static void main(String[] args)
    {
        new HookTest().start();
        System.out.println("The Application is doing something");
        byte[] b = new byte[500*1024*1024];
        try
        {
            TimeUnit.MILLISECONDS.sleep(5000);
        }
        catch (InterruptedException e)
        {
            e.printStackTrace();
        }
    }
 
}
运行参数设置为：-Xmx20M  这样可以保证会有OutOfMemoryError的发生。

 

运行结果：

 

The Application is doing something
Exception in thread "main" java.lang.OutOfMemoryError: Java heap space
    at com.hook.HookTest2.main(HookTest2.java:22)
Execute Hook.....
可以看到程序遇到内存溢出错误后调用关闭钩子，与第一种情况中，程序等待5000ms运行结束之后推出调用关闭钩子不同。

 

接下来再来测试第三种情况：

package com.hook;
 
import java.util.concurrent.TimeUnit;
 
public class HookTest3
{
    public void start()
    {
        Runtime.getRuntime().addShutdownHook(new Thread(new Runnable() {
            @Override
            public void run()
            {
                System.out.println("Execute Hook.....");
            }
        }));
    }
    
    public static void main(String[] args)
    {
        new HookTest3().start();
        Thread thread = new Thread(new Runnable(){
 
            @Override
            public void run()
            {
                while(true)
                {
                    System.out.println("thread is running....");
                    try
                    {
                        TimeUnit.MILLISECONDS.sleep(100);
                    }
                    catch (InterruptedException e)
                    {
                        e.printStackTrace();
                    }
                }
            }
            
        });
        thread.start();
    }
 
}
在命令行中编译：javac com/hook/HookTest3.java

 

在命令行中运行：java com.hook.HookTest3  （之后按下Ctrl+C）

运行结果：
可以看到效果如预期。
还有几种情况就不一一列出了，有兴趣的读者可以试一下。

欢迎跳转到本文的原文链接：https://honeypps.com/java/java-shutdown-hook/

欢迎支持笔者新作：《深入理解Kafka:核心设计与实践原理》和《RabbitMQ实战指南》，同时欢迎关注笔者的微信公众号：朱小厮的博客。
————————————————
版权声明：本文为CSDN博主「朱小厮」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/u013256816/article/details/50394923