瀏覽代碼

mybatis自动生成工具调整

songxinlu 4 年之前
父節點
當前提交
745588362e
共有 1 個文件被更改,包括 8 次插入8 次删除
  1. 8 8
      security-center/src/test/java/com/diagbot/CodeAutoGenerator.java

+ 8 - 8
security-center/src/test/java/com/diagbot/CodeAutoGenerator.java

@@ -19,6 +19,7 @@ import com.baomidou.mybatisplus.generator.config.rules.FileType;
 import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
 import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
 
+import java.awt.*;
 import java.io.File;
 import java.util.ArrayList;
 import java.util.List;
@@ -229,9 +230,8 @@ public class CodeAutoGenerator {
                 .setTypeConvert(new MySqlTypeConvert() {
                     @Override
                     public DbColumnType processTypeConvert(GlobalConfig globalConfig, String fieldType) {
-                        System.out.println("转换类型:" + fieldType);
                         //将数据库中datetime转换成date
-                        if ( fieldType.toLowerCase().contains( "datetime" ) ) {
+                        if (fieldType.toLowerCase().contains("datetime")) {
                             return DbColumnType.DATE;
                         }
                         return (DbColumnType) super.processTypeConvert(globalConfig, fieldType);
@@ -317,11 +317,11 @@ public class CodeAutoGenerator {
 
                         // 指定需要覆盖的文件
                         // 文件结尾名字参照 全局配置 中对各层文件的命名,未修改为默认值
-                        if (isExists(filePath) && (!filePath.endsWith("Mapper.xml")
-                                && !filePath.endsWith("Mapper.java"))
-                                && !filePath.contains(".service")
-                                && !filePath.endsWith("Facade.java")) {
-                            return false;
+                        if (isExists(filePath)) {
+                            if (filePath.endsWith("MapperParent.java")
+                                    || filePath.endsWith("Facade.java")) {
+                                return false;
+                            }
                         }
 
                         return true;
@@ -357,7 +357,7 @@ public class CodeAutoGenerator {
                 }
             });
         }
-        // facade文件输出
+        // facade文件输出 不会覆盖
         if (facadeFlag) {
             list.add(new FileOutConfig(FACADE_TEMPLATE) {
                 @Override