[improve] move mybatis plus setting into MybatisConfig (#2099)

apache · Nov 26, 2022 · 0b100ce · 0b100ce
1 parent a2dbb6e
commit 0b100ce
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 12 deletions.
diff --git a/...onsole-service/src/main/java/org/apache/streampark/console/base/config/MybatisConfig.java b/...onsole-service/src/main/java/org/apache/streampark/console/base/config/MybatisConfig.java
@@ -20,6 +20,11 @@
 import org.apache.streampark.console.base.mybatis.interceptor.PostgreSQLPrepareInterceptor;
 import org.apache.streampark.console.base.mybatis.interceptor.PostgreSQLQueryInterceptor;
 
+import com.baomidou.mybatisplus.annotation.IdType;
+import com.baomidou.mybatisplus.autoconfigure.MybatisPlusPropertiesCustomizer;
+import com.baomidou.mybatisplus.core.MybatisConfiguration;
+import com.baomidou.mybatisplus.core.config.GlobalConfig;
+import com.baomidou.mybatisplus.core.toolkit.GlobalConfigUtils;
 import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
 import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
 import org.mybatis.spring.annotation.MapperScan;
@@ -63,4 +68,25 @@ public PostgreSQLPrepareInterceptor postgreSQLPrepareInterceptor() {
         return new PostgreSQLPrepareInterceptor();
     }
 
+    /**
+     * mybatis plus setting
+     *
+     * @return MybatisPlusPropertiesCustomizer
+     */
+    @Bean
+    public MybatisPlusPropertiesCustomizer mybatisPlusPropertiesCustomizer() {
+        return properties -> {
+            properties.setTypeAliasesPackage("org.apache.streampark.console.*.entity");
+            properties.setMapperLocations(new String[]{"classpath:mapper/*/*.xml"});
+            MybatisConfiguration mybatisConfiguration = new MybatisConfiguration();
+            mybatisConfiguration.setJdbcTypeForNull(null);
+            properties.setConfiguration(mybatisConfiguration);
+            GlobalConfig globalConfig = GlobalConfigUtils.getGlobalConfig(mybatisConfiguration);
+            GlobalConfig.DbConfig dbConfig = globalConfig.getDbConfig();
+            dbConfig.setIdType(IdType.AUTO);
+            // close mybatis-plus banner
+            globalConfig.setBanner(false);
+            properties.setGlobalConfig(globalConfig);
+        };
+    }
 }
diff --git a/streampark-console/streampark-console-service/src/main/resources/application.yml b/streampark-console/streampark-console-service/src/main/resources/application.yml
@@ -71,18 +71,6 @@ management:
       probes:
         enabled: true
 
-# mybatis plus setting
-mybatis-plus:
-  type-aliases-package: org.apache.streampark.console.*.entity
-  mapper-locations: classpath:mapper/*/*.xml
-  configuration:
-    jdbc-type-for-null: null
-  global-config:
-    db-config:
-      id-type: auto
-    # close mybatis-plus banner
-    banner: false
-
 streampark:
   proxy:
     # knox process address https://cdpsit02.example.cn:8443/gateway/cdp-proxy/yarn