如何设置全局钩子检测Java OutOfMemoryError以实现快速服务重启？

当然可以通过全局机制捕获OutOfMemoryError并触发重启，无需在代码各处编写重复的catch逻辑，以下是两种最可靠的实现方式：

1. JVM层面的OOM钩子（推荐） #

直接通过JVM启动参数配置，当发生OutOfMemoryError时自动执行指定脚本，这种方式不受代码中异常捕获逻辑的影响，能覆盖所有场景（包括JVM内部线程触发的OOM）。

使用方式 #

在服务的JVM启动参数中添加：

-XX:OnOutOfMemoryError="/path/to/your-restart-script.sh %p"

• %p会被JVM替换为当前进程的PID，脚本可利用这个PID完成进程清理、通知负载均衡器下线等操作，最后重启服务。
• 脚本要尽量轻量化，避免执行耗内存的操作，确保在OOM场景下能顺利执行。

2. 代码层面设置全局未捕获异常处理器 #

通过设置全局的UncaughtExceptionHandler，捕获所有线程中未被显式捕获的Throwable（包括Error类型），当检测到OutOfMemoryError时触发重启。

代码示例 #

public class GlobalOomHandler {
    public static void setup() {
        Thread.setDefaultUncaughtExceptionHandler((thread, throwable) -> {
            if (throwable instanceof OutOfMemoryError) {
                System.err.printf("线程[%s]触发OutOfMemoryError，启动紧急重启%n", thread.getName());
                // 优先调用系统脚本重启，避免在内存不足时执行复杂逻辑
                try {
                    Runtime.getRuntime().exec("/path/to/your-restart-script.sh");
                } catch (Exception e) {
                    // 极端情况直接退出进程
                    System.exit(1);
                }
                System.exit(1);
            }
        });
    }
}

在服务启动时调用GlobalOomHandler.setup()即可生效。

注意事项 #

• 这种方式无法捕获被代码显式catch(OutOfMemoryError)的场景，所以如果代码中有这类捕获逻辑，需要确保在catch块中也触发重启逻辑。
• OOM发生后尽量避免分配新对象，所以重启逻辑要尽可能简洁，优先调用外部脚本处理。

额外建议 #

• 重启前可先调用负载均衡器的节点下线API，避免重启过程中还有新请求被转发到故障节点，减少虚假40X错误。
• 生产环境务必模拟OOM场景验证钩子的有效性，确保触发逻辑可靠。

内容的提问来源于stack exchange，提问作者eshalev