mybatis-plus的SafetyEncryptProcessor安全加密處理示例解析

更新時(shí)間：2023年08月28日 09:36:04 作者：codecraft

這篇文章主要為大家介紹了mybatis-plus的SafetyEncryptProcessor安全加密處理示例解析,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪

序

本文主要研究一下mybatis-plus的SafetyEncryptProcessor

SafetyEncryptProcessor

mybatis-plus-boot-starter/src/main/java/com/baomidou/mybatisplus/autoconfigure/SafetyEncryptProcessor.java

public class SafetyEncryptProcessor implements EnvironmentPostProcessor {
    @Override
    public void postProcessEnvironment(ConfigurableEnvironment environment, SpringApplication application) {
        /**
         * 命令行中獲取密鑰
         */
        String mpwKey = null;
        for (PropertySource<?> ps : environment.getPropertySources()) {
            if (ps instanceof SimpleCommandLinePropertySource) {
                SimpleCommandLinePropertySource source = (SimpleCommandLinePropertySource) ps;
                mpwKey = source.getProperty("mpw.key");
                break;
            }
        }
        /**
         * 處理加密內(nèi)容
         */
        if (StringUtils.isNotBlank(mpwKey)) {
            HashMap<String, Object> map = new HashMap<>();
            for (PropertySource<?> ps : environment.getPropertySources()) {
                if (ps instanceof OriginTrackedMapPropertySource) {
                    OriginTrackedMapPropertySource source = (OriginTrackedMapPropertySource) ps;
                    for (String name : source.getPropertyNames()) {
                        Object value = source.getProperty(name);
                        if (value instanceof String) {
                            String str = (String) value;
                            if (str.startsWith("mpw:")) {
                                map.put(name, AES.decrypt(str.substring(4), mpwKey));
                            }
                        }
                    }
                }
            }
            // 將解密的數(shù)據(jù)放入環(huán)境變量，并處于第一優(yōu)先級上
            if (CollectionUtils.isNotEmpty(map)) {
                environment.getPropertySources().addFirst(new MapPropertySource("custom-encrypt", map));
            }
        }
    }
}

SafetyEncryptProcessor實(shí)現(xiàn)了EnvironmentPostProcessor接口，在postProcessEnvironment方法中先是找到mpw.key，然后遍歷所有PropertySource的所有屬性，找到mpw:開頭的，然后進(jìn)行解密并替換到密文，最后放在environment的第一個(gè)PropertySource

spring.factories

mybatis-plus-boot-starter/src/main/resources/META-INF/spring.factories

# Auto Configure
org.springframework.boot.env.EnvironmentPostProcessor=\
  com.baomidou.mybatisplus.autoconfigure.SafetyEncryptProcessor

小結(jié)

之前的文章聊聊springboot的EnvironmentPostProcessor提到springboot提供了EnvironmentPostProcessor接口，該接口有postProcessEnvironment方法，其中envrionment參數(shù)類型為ConfigurableEnvironment，即應(yīng)用可以通過實(shí)現(xiàn)這個(gè)接口進(jìn)行env環(huán)境變量的操作。而mybatis-plus的SafetyEncryptProcessor正是一個(gè)實(shí)戰(zhàn)的好例子。

以上就是mybatis-plus的SafetyEncryptProcessor安全加密處理示例解析的詳細(xì)內(nèi)容，更多關(guān)于mybatis-plus SafetyEncryptProcessor的資料請關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: