Java添加关闭钩子里面的方法

时间:2016-03-23 09:26:14

标签: java executorservice shutdown-hook completion-service

在我的代码中,我使用CompletionService和ExecutorService来启动一堆Thread来执行某些任务(这可能需要很长时间)。 所以我有一个创建ExecutorService和CompletionService的方法,然后开始提交线程然后获取结果。 我想添加一个关闭钩子,以便优雅地关闭执行程序(我知道可能我应该处理释放资源而不是执行器关闭但在我的情况下每个线程都有自己的资源,所以优雅地关闭它们可以是一个很好的解决方案我假设)。

因此我编写以下代码

public Class myClass{
...
private CompletionService<ClusterJobs> completion;
final long SHUTDOWN_TIME = TimeUnit.SECONDS.toSeconds(10);

...
public Message executeCommand(Message request){

final ExecutorService executor = Executors.newFixedThreadPool(30);

completion = new ExecutorCompletionService<ClusterJobs>(executor);

....//submit and take results

Runtime.getRuntime().addShutdownHook(new Thread(){
            @Override
            public void run() {
                logger.debug("Shutting down executor");

                try {
                    if (!executor.awaitTermination(SHUTDOWN_TIME, TimeUnit.SECONDS)) {
                        logger.debug("Executor still not terminate after waiting time...");
                        List<Runnable> notExecuted= executor.shutdownNow();
                        logger.debug("List of dropped task has size " + droppedTasks.size());
                    }
                }catch(InterruptedException e){
                    logger.error("",e);
                }
            }
        });

}
}

您认为这是一个合理的解决方案,还是使用本地类注册和取消注册关闭钩子是不安全的?

提前致谢

此致

1 个答案:

答案 0 :(得分:1)

来自Design of the Shutdown Hooks API

简单的关闭挂钩通常可以写为匿名内部类,如下例所示:

Runtime.getRuntime().addShutdownHook(new Thread() {
    public void run() { database.close(); }
});

这个成语很好,只要你永远不需要取消钩子,在这种情况下你需要在创建它时保存对钩子的引用。