前言
这段时间用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;