自定义配置文件脱敏工具

Author: zhifu.xin

springboot-jasypt/src/main/java/com/xiaofu/JasyptApplication.java

@@ -1,14 +1,20 @@
 package com.xiaofu;
 
+import com.xiaofu.process.MyPropertiesBeanFactoryPostProcessor;
+import com.xiaofu.process.conf;
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.context.ApplicationContext;
+import org.springframework.context.ConfigurableApplicationContext;
 import org.springframework.context.annotation.ComponentScan;
+import org.springframework.context.annotation.Import;
 
+@Import({MyPropertiesBeanFactoryPostProcessor.class, conf.class})
 @ComponentScan("com.xiaofu")
 @SpringBootApplication
 public class JasyptApplication {
 
-	public static void main(String[] args) {
-		SpringApplication.run(JasyptApplication.class, args);
-	}
+    public static void main(String[] args) {
+        ConfigurableApplicationContext run = SpringApplication.run(JasyptApplication.class, args);
+    }
 }

springboot-jasypt/src/main/java/com/xiaofu/controller/Encryptor.java

@@ -3,14 +3,12 @@
 import com.alibaba.fastjson.JSON;
 import com.xiaofu.annotation.EncryptField;
 import com.xiaofu.annotation.EncryptMethod;
-import lombok.Data;
+import com.xiaofu.model.UserVo;
 import lombok.extern.slf4j.Slf4j;
 import org.jasypt.encryption.StringEncryptor;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.*;
 
-import java.io.Serializable;
-
 @Slf4j
 @RestController
 @RequestMapping("/encryptor/")
@@ -36,18 +34,4 @@ private UserVo insertUser(UserVo user, String name) {
         System.out.println("加密后的数据：user" + JSON.toJSONString(user));
         return user;
     }
-
-    @Data
-    public class UserVo implements Serializable {
-
-        private Long userId;
-
-        @EncryptField
-        private String mobile;
-
-        @EncryptField
-        private String address;
-
-        private String age;
-    }
 }

springboot-jasypt/src/main/java/com/xiaofu/model/UserVo.java

@@ -10,16 +10,16 @@
  * @Date: 2021/7/26 15:10
  * @Description:
  */
-//@Data
-//public class UserVo implements Serializable {
-//
-//    private Long userId;
-//
-//    @EncryptField
-//    private String mobile;
-//
-//    @EncryptField
-//    private String address;
-//
-//    private String age;
-//}
+@Data
+public class UserVo implements Serializable {
+
+    private Long userId;
+
+    @EncryptField
+    private String mobile;
+
+    @EncryptField
+    private String address;
+
+    private String age;
+}

springboot-jasypt/src/main/java/com/xiaofu/process/EnvCopy1.java

@@ -0,0 +1,34 @@
+package com.xiaofu.process;
+
+import com.ulisesbocchio.jasyptspringboot.EncryptablePropertySource;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.core.env.ConfigurableEnvironment;
+import org.springframework.core.env.PropertySource;
+import org.springframework.core.env.StandardEnvironment;
+import org.springframework.stereotype.Component;
+
+import java.util.Optional;
+
+/**
+ * @Auther: [email protected]
+ * @Date: 2021/7/30 18:35
+ * @Description:
+ */
+public class EnvCopy1 {
+
+    StandardEnvironment copy1;
+
+    public EnvCopy1(final ConfigurableEnvironment environment) {
+        copy1 = new StandardEnvironment();
+        Optional.ofNullable(environment.getPropertySources()).ifPresent(sources -> sources.forEach(ps -> {
+            final PropertySource<?> original = ps instanceof EncryptablePropertySource
+                    ? ((EncryptablePropertySource) ps).getDelegate()
+                    : ps;
+            copy1.getPropertySources().addLast(original);
+        }));
+    }
+
+    public ConfigurableEnvironment get() {
+        return copy1;
+    }
+}

springboot-jasypt/src/main/java/com/xiaofu/process/MyPropertiesBeanFactoryPostProcessor.java

@@ -0,0 +1,39 @@
+package com.xiaofu.process;
+
+import lombok.NoArgsConstructor;
+import org.springframework.beans.BeansException;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.config.BeanFactoryPostProcessor;
+import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;
+import org.springframework.context.annotation.Import;
+import org.springframework.core.Ordered;
+import org.springframework.core.env.*;
+
+import java.util.Iterator;
+
+@NoArgsConstructor
+public class MyPropertiesBeanFactoryPostProcessor implements BeanFactoryPostProcessor, Ordered {
+
+    EnvCopy1 envCopy1 ;
+
+    @Override
+    public void postProcessBeanFactory(ConfigurableListableBeanFactory configurableListableBeanFactory) throws BeansException {
+        MutablePropertySources propertySources = envCopy1.get().getPropertySources();
+        Iterator<PropertySource<?>> iterator = propertySources.iterator();
+        while (iterator.hasNext()) {
+            PropertySource<?> propertySource = iterator.next();
+
+            if (propertySource instanceof PropertiesPropertySource) {
+                System.out.println(propertySource.getProperty("propertyKey"));
+                // 用propertySource.getSource()  可以获取全部配置
+            }
+        }
+    }
+
+    @Override
+    public int getOrder() {
+        return 0;
+    }
+
+
+}

springboot-jasypt/src/main/java/com/xiaofu/process/conf.java

@@ -0,0 +1,19 @@
+package com.xiaofu.process;
+
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.core.env.ConfigurableEnvironment;
+
+/**
+ * @Auther: [email protected]
+ * @Date: 2021/7/30 18:41
+ * @Description:
+ */
+@Configuration
+public class conf {
+
+    @Bean
+    public EnvCopy1 envCopy1(final ConfigurableEnvironment environment) {
+        return new EnvCopy1(environment);
+    }
+}