Spring Boot 在启动时进行配置文件加解密

Spring Boot Application 事件和监听器

• 寻找到application.yml的读取的操作。
• 从spring.factories 中查看到

# Application Listeners
org.springframework.context.ApplicationListener=\
org.springframework.boot.context.config.ConfigFileApplicationListener,\

• ConfigFileApplicationListener 该对象对application.yml进行读取操作
• ConfigFileApplicationListener 事件的监听器， 继承了SmartApplicationListener接口
• SmartApplicationListener 接口继承了ApplicationListener 和 Ordered接口，能够实现有序监听。

一、SmartApplicationListener介绍

• Spring ApplicationEvent以及对应的Listener提供了一个事件监听、发布订阅的实现，内部实现方式是观察者模式，可以解耦业务系统之间的业务，提供系统的可拓展性、复用性以及可维护性。
• 在application.yml文件读取完会触发一个事件ConfigFileApplicationListener 该监听器实现文件的读取。
• SmartApplicationListener是高级监听器，是ApplicationListener的子类，能够实现有序监听
• SmartApplicationListener提供了两个方法：

/**
 *  指定支持哪些类型的事件
 */
boolean supportsEventType(Class<? extends ApplicationEvent> var1);

/**
 *  指定支持发生事件所在的类型
 */
boolean supportsSourceType(Class<?> var1);

如何在 SmartApplicationListener 实现监听解耦

• 1、我们只需在加载完成之后去加入一个监听器。
• 2、继承SmartApplicationListener接口
• 3、设置order属性决定监听器的顺序 ConfigFileApplicationListener.DEFAULT_ORDER + 1
• 4、将application.yml内容读取之后并修改

二、ConfigFileApplicationListener

• ConfigFileApplicationListener 是用来 读取配置文件的。 可以这样来粗劣的介绍一下
• 详情可以请看 springboot启动时是如何加载配置文件application.yml文件

三、最终结果：

• 新增一个监听器 既然我们要在配置文件加载之后搞事情那么我们直接复制ConfigFileApplicationListener 的实现方式
• 删除一下不需要处理的操作(大概就是以下代码) 并且order在ConfigFileApplicationListener 之后

public class AfterConfigListener implements SmartApplicationListener,Ordered {

    public boolean supportsEventType(Class<? extends ApplicationEvent> eventType) {
        return ApplicationEnvironmentPreparedEvent.class.isAssignableFrom(eventType) || ApplicationPreparedEvent.class.isAssignableFrom(eventType);
    }
    public void onApplicationEvent(ApplicationEvent event) {
        if (event instanceof ApplicationEnvironmentPreparedEvent) {
        }
        if (event instanceof ApplicationPreparedEvent) {
        }
    }
    @Override
    public int getOrder() {
        // 写在加载配置文件之后
        return ConfigFileApplicationListener.DEFAULT_ORDER + 1;
    }
}

• 这样子就完成了配置文件之后的代码监听。 SmartApplicationListener又是实现了ApplicationListener的监听的，那么我们可以在onApplicationEvent执行代码。
• 完善代码如下。 监听并且获取配置文件内容

public class AfterConfigListener implements SmartApplicationListener,Ordered {
    // 复制的ConfigFileApplicationListener文件内容
    public boolean supportsEventType(Class<? extends ApplicationEvent> eventType) {
        return ApplicationEnvironmentPreparedEvent.class.isAssignableFrom(eventType) || ApplicationPreparedEvent.class.isAssignableFrom(eventType);
    }

    public void onApplicationEvent(ApplicationEvent event) {
        // ApplicationEnvironmentPreparedEvent 是加载配置文件，初始化日志系统的事件。 
        if (event instanceof ApplicationEnvironmentPreparedEvent) {
            // 获得原来的password内容
            String password = ((ApplicationEnvironmentPreparedEvent) event).getEnvironment().getProperty("spring.datasource.password");
            
            // 进行密码的解密
              System.setProperty("spring.datasource.password", SM4Utils.encryptData_ECB(password));
        }

        if (event instanceof ApplicationPreparedEvent) {
        }	
    }
    @Override
    public int getOrder() {
        // 设置该监听器 在加载配置文件之后执行
        return ConfigFileApplicationListener.DEFAULT_ORDER + 1;
    }

}

• 并且在main方法中加入该监听器

public class Application {

    public static void main(String[] args) {
        SpringApplication springApplication = new SpringApplication(Application.class);
        springApplication.addListeners(new AfterConfigListener());
        springApplication.run(args);
    }

}