Java中如何暂停程序关闭请求，等待方法返回或超时后再关闭？

问题场景 #

需求是拦截程序的任意关闭请求（比如Ctrl+C、系统信号），等待schedulerManager.shutdown()方法执行完成（或等待指定超时时间）后再退出程序，但当前实现中，shutdown()方法里的persistSchedulerState()因数据库流已关闭而执行失败。

当前代码示例 #

主命令类 #

import picocli.CommandLine;
// 其他导入
@CommandLine.Command(
        name = "main",
        description = "Example main command"
)
public class MainCommand implements Callable<Integer> {
    @Override
    public Integer call() throws Exception {
        
        // 注册关闭钩子
        Runtime.getRuntime().addShutdownHook(new Thread(() -> {
            try {
                schedulerManager.shutdown(); // 必须等待该方法返回或超时
            } catch (IOException | SchedulerException e) {
                Logger.error("Scheduler shutdown failed: " + e.getMessage());
            }
        }));
        
        while (true) {
            Thread.sleep(1000);  // 每秒休眠一次
        }
    }
}

调度器的shutdown方法 #

public void shutdown() throws SchedulerException, IOException {
    persistSchedulerState(); // 将任务持久化到数据库
    shutdownScheduler();     // 关闭调度器
}

问题原因 #

JVM触发关闭流程时，shutdown hook的执行顺序不固定。如果数据库连接池、IO流等资源被其他关闭逻辑（比如框架的销毁钩子、其他自定义shutdown hook）提前关闭，persistSchedulerState()就会因无法访问数据库而失败。

解决方案 #

1. 主动监听中断信号，提前触发shutdown #

在主线程中监听中断信号，收到关闭请求时主动执行shutdown逻辑，避免依赖shutdown hook的延迟执行：

import picocli.CommandLine;
import java.util.concurrent.TimeUnit;
// 其他导入
@CommandLine.Command(
        name = "main",
        description = "Example main command"
)
public class MainCommand implements Callable<Integer> {
    private volatile boolean running = true;

    @Override
    public Integer call() throws Exception {
        // 注册关闭钩子，标记运行状态为结束
        Runtime.getRuntime().addShutdownHook(new Thread(() -> {
            running = false;
            executeShutdown();
        }));

        while (running) {
            try {
                Thread.sleep(1000);
            } catch (InterruptedException e) {
                // 收到中断信号，主动终止循环并执行shutdown
                running = false;
                executeShutdown();
            }
        }
        return 0;
    }

    private void executeShutdown() {
        try {
            // 等待shutdown完成，最多30秒超时
            boolean shutdownSuccess = schedulerManager.shutdownWithTimeout(30, TimeUnit.SECONDS);
            if (!shutdownSuccess) {
                Logger.warn("Scheduler shutdown timed out, exiting program");
            }
        } catch (IOException | SchedulerException e) {
            Logger.error("Scheduler shutdown failed: " + e.getMessage());
        }
    }
}

2. 为shutdown方法添加超时机制 #

修改调度器的shutdown方法，支持超时控制，避免无限等待：

public boolean shutdownWithTimeout(long timeout, TimeUnit unit) throws SchedulerException, IOException {
    CountDownLatch latch = new CountDownLatch(1);
    Thread persistThread = new Thread(() -> {
        try {
            persistSchedulerState();
            shutdownScheduler();
        } catch (SchedulerException | IOException e) {
            Logger.error("Shutdown failed: " + e.getMessage());
        } finally {
            latch.countDown();
        }
    });
    persistThread.start();
    
    try {
        return latch.await(timeout, unit);
    } catch (InterruptedException e) {
        Thread.currentThread().interrupt();
        return false;
    }
}

3. 控制资源关闭顺序 #

确保数据库连接池等资源的关闭逻辑在调度器shutdown完成之后执行。比如使用Spring框架时，可通过调整@PreDestroy注解的执行顺序，或者手动管理资源生命周期，让调度器的shutdown优先执行。

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