|
@@ -0,0 +1,47 @@
|
|
|
+package org.jeecg.config;
|
|
|
+
|
|
|
+import org.springframework.boot.SpringApplication;
|
|
|
+import org.springframework.boot.SpringApplicationRunListener;
|
|
|
+import org.springframework.context.ConfigurableApplicationContext;
|
|
|
+import org.springframework.core.env.ConfigurableEnvironment;
|
|
|
+import org.springframework.core.env.MapPropertySource;
|
|
|
+import org.springframework.core.env.MutablePropertySources;
|
|
|
+import org.springframework.core.env.PropertySource;
|
|
|
+
|
|
|
+import java.util.HashMap;
|
|
|
+import java.util.Map;
|
|
|
+
|
|
|
+/**
|
|
|
+ * @author eightmonth@qq.com
|
|
|
+ * @date 2024/4/8 11:37
|
|
|
+ */
|
|
|
+public class WallConfigRegister implements SpringApplicationRunListener {
|
|
|
+
|
|
|
+ public SpringApplication application;
|
|
|
+
|
|
|
+ private String[] args;
|
|
|
+
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 必备,否则启动报错
|
|
|
+ * @param application
|
|
|
+ * @param args
|
|
|
+ */
|
|
|
+ public WallConfigRegister(SpringApplication application, String[] args) {
|
|
|
+ this.application = application;
|
|
|
+ this.args = args;
|
|
|
+ }
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public void contextLoaded(ConfigurableApplicationContext context) {
|
|
|
+ ConfigurableEnvironment env = context.getEnvironment();
|
|
|
+ Map<String, Object> props = new HashMap<>();
|
|
|
+ props.put("spring.datasource.dynamic.druid.wall.selectWhereAlwayTrueCheck", false);
|
|
|
+
|
|
|
+ MutablePropertySources propertySources = env.getPropertySources();
|
|
|
+
|
|
|
+ PropertySource<Map<String, Object>> propertySource = new MapPropertySource("jeecg-datasource-config", props);
|
|
|
+
|
|
|
+ propertySources.addLast(propertySource);
|
|
|
+ }
|
|
|
+}
|