JVM关闭钩子 ---ShutdownHook

川长思鸟来 2022-05-27 03:52 234阅读 0赞

Java程序经常也会遇到进程挂掉的情况，一些状态没有正确的保存下来，这时候就需要在JVM关掉的时候执行一些清理现场的代码。JAVA中的ShutdownHook提供了比较好的方案。

JDK提供了Java.Runtime.addShutdownHook(Thread hook)方法，可以注册一个JVM关闭的钩子，这个钩子可以在一下几种场景中被调用：

1.  程序正常退出
2.  使用System.exit()
3.  终端使用Ctrl+C触发的中断
4.  系统关闭
5.  OutOfMemory宕机
6.  使用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)

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

**\[java\]**  [view plain][]   [copy][view plain]

1.  package com.hook;  
2.    
3.  import java.util.concurrent.TimeUnit;  
4.    
5.  public class HookTest  
6.  \{  
7.      public void start()  
8.      \{  
9.          Runtime.getRuntime().addShutdownHook(new Thread(new Runnable() \{  
10.             @Override  
11.             public void run()  
12.             \{  
13.                 System.out.println("Execute Hook.....");  
14.             \}  
15.         \}));  
16.     \}  
17.       
18.     public static void main(String\[\] args)  
19.     \{  
20.         new HookTest().start();  
21.         System.out.println("The Application is doing something");  
22.           
23.         try  
24.         \{  
25.             TimeUnit.MILLISECONDS.sleep(5000);  
26.         \}  
27.         catch (InterruptedException e)  
28.         \{  
29.             e.printStackTrace();  
30.         \}  
31.     \}  
32. \}

运行结果：

**\[plain\]**  [view plain][]   [copy][view plain]

1.  The Application is doing something  
2.  Execute Hook.....

如上可以看到，当main线程运行结束之后就会调用关闭钩子。

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

**\[java\]**  [view plain][]   [copy][view plain]

1.  package com.hook;  
2.    
3.  import java.util.concurrent.TimeUnit;  
4.    
5.  public class HookTest2  
6.  \{  
7.      public void start()  
8.      \{  
9.          Runtime.getRuntime().addShutdownHook(new Thread(new Runnable() \{  
10.             @Override  
11.             public void run()  
12.             \{  
13.                 System.out.println("Execute Hook.....");  
14.             \}  
15.         \}));  
16.     \}  
17.       
18.     public static void main(String\[\] args)  
19.     \{  
20.         new HookTest().start();  
21.         System.out.println("The Application is doing something");  
22.         byte\[\] b = new byte\[500\*1024\*1024\];  
23.         try  
24.         \{  
25.             TimeUnit.MILLISECONDS.sleep(5000);  
26.         \}  
27.         catch (InterruptedException e)  
28.         \{  
29.             e.printStackTrace();  
30.         \}  
31.     \}  
32.   
33. \}

运行参数设置为：-Xmx20M  这样可以保证会有OutOfMemoryError的发生。

运行结果：

**\[plain\]**  [view plain][]   [copy][view plain]

1.  The Application is doing something  
2.  Exception in thread "main" java.lang.OutOfMemoryError: Java heap space  
3.      at com.hook.HookTest2.main(HookTest2.java:22)  
4.  Execute Hook.....

可以看到程序遇到内存溢出错误后调用关闭钩子，与第一种情况中，程序等待5000ms运行结束之后推出调用关闭钩子不同。

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

**\[java\]**  [view plain][]   [copy][view plain]

1.  package com.hook;  
2.    
3.  import java.util.concurrent.TimeUnit;  
4.    
5.  public class HookTest3  
6.  \{  
7.      public void start()  
8.      \{  
9.          Runtime.getRuntime().addShutdownHook(new Thread(new Runnable() \{  
10.             @Override  
11.             public void run()  
12.             \{  
13.                 System.out.println("Execute Hook.....");  
14.             \}  
15.         \}));  
16.     \}  
17.       
18.     public static void main(String\[\] args)  
19.     \{  
20.         new HookTest3().start();  
21.         Thread thread = new Thread(new Runnable()\{  
22.   
23.             @Override  
24.             public void run()  
25.             \{  
26.                 while(true)  
27.                 \{  
28.                     System.out.println("thread is running....");  
29.                     try  
30.                     \{  
31.                         TimeUnit.MILLISECONDS.sleep(100);  
32.                     \}  
33.                     catch (InterruptedException e)  
34.                     \{  
35.                         e.printStackTrace();  
36.                     \}  
37.                 \}  
38.             \}  
39.               
40.         \});  
41.         thread.start();  
42.     \}  
43.   
44. \}

在命令行中编译：javac com/hook/HookTest3.java

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

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

[view plain]: https://blog.csdn.net/u013256816/article/details/50394923#
[20151224165056549]: /images/20220527/05698fab55d54e82a7d5a8fe94e18686.png