在Springboot应用内部关闭程序的方法

305 阅读2分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

前言

这段时间用SpringBoot写了一个使用web访问触发生成日志的无聊的小应用。写的时候遇到了一个问题,我想通过访问特定地址来关闭程序,但是找不到关闭程序的接口啊。 通过万能的前人经验,我主要找到了以下几种解决方案。

解决方案

spring-boot-starter-actuator

使用spring-boot-starter-actuator,不过该方法过于重量,在尝试成功后还是决定再找找。

保存程序context的静态变量

在程序入口执行SpringApplication.run(LogCreatorApplication.class, args)这段代码会返回一个ConfigurableApplicationContext对象,通过调用该对象的close方法可以关闭程序。如以下代码:

import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.context.ConfigurableApplicationContext;

@SpringBootApplication
public class Application {
    private static ConfigurableApplicationContext context;

    public static void main(String[] args) {
        context = SpringApplication.run(LogCreatorApplication.class, args);
    }

    public static void exit() {
        if (context != null)
            SpringApplication.exit(context, () -> 0);
    }
}

之后在关闭程序的时候使用Application.exit()就可以了。 不过这种方法保存的context和调用方法都是静态的,相当不SpringBoot,所以我有找了找有没有比较官方的解决办法。

实现ApplicationContextAware接口

最后我找到了这个解决方案,ApplicationContextAware接口需要类Override一个setApplicationContext方法,该方法传入了一个ApplicationContext对象,正是我们需要的程序的context。我们可以将这个context保存下来,然后通过@Component注解将我们实现的类注解为Bean,之后就可以自由的使用@Autowire注解引入这个类了。 以下是代码,setApplicationContext是SpringBoot自动帮我们获取的context的,无需我们手动调用。

import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.context.ConfigurableApplicationContext;
import org.springframework.stereotype.Component;

@Component
public class AppUtils implements ApplicationContextAware {

    private ConfigurableApplicationContext context;

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        context = (ConfigurableApplicationContext) applicationContext;
    }

    public void exit() {
        context.close();
    }
}

引入AppUtils,这里用了java自带的@Resource注解,效果和Autowired差不多

@Resource
AppUtils app;