Browse Source

mybatis自动生成工具调整

songxinlu 3 years ago
parent
commit
f53234517f
1 changed files with 15 additions and 1 deletions
  1. 15 1
      security-center/src/test/java/com/diagbot/CodeAutoGenerator.java

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

@@ -12,7 +12,9 @@ import com.baomidou.mybatisplus.generator.config.PackageConfig;
 import com.baomidou.mybatisplus.generator.config.StrategyConfig;
 import com.baomidou.mybatisplus.generator.config.TemplateConfig;
 import com.baomidou.mybatisplus.generator.config.builder.ConfigBuilder;
+import com.baomidou.mybatisplus.generator.config.converts.MySqlTypeConvert;
 import com.baomidou.mybatisplus.generator.config.po.TableInfo;
+import com.baomidou.mybatisplus.generator.config.rules.DbColumnType;
 import com.baomidou.mybatisplus.generator.config.rules.FileType;
 import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
 import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
@@ -222,7 +224,19 @@ public class CodeAutoGenerator {
                 // 用户名
                 .setUsername(username)
                 // 密码
-                .setPassword(password);
+                .setPassword(password)
+                // 自定义数据库表字段类型转换【可选】
+                .setTypeConvert(new MySqlTypeConvert() {
+                    @Override
+                    public DbColumnType processTypeConvert(GlobalConfig globalConfig, String fieldType) {
+                        System.out.println("转换类型:" + fieldType);
+                        //将数据库中datetime转换成date
+                        if ( fieldType.toLowerCase().contains( "datetime" ) ) {
+                            return DbColumnType.DATE;
+                        }
+                        return (DbColumnType) super.processTypeConvert(globalConfig, fieldType);
+                    }
+                });
     }
 
     /**