修改mybatis plus 代码生成 2023年3月23日11:34:02

src/main/java/com/zhangmeng/tools/controller/MybatisPlusGenerator.java

@@ -2,8 +2,10 @@ package com.zhangmeng.tools.controller;
 
 import cn.hutool.core.text.StrFormatter;
 import com.alibaba.druid.pool.DruidDataSource;
+import com.baomidou.mybatisplus.annotation.IdType;
 import com.baomidou.mybatisplus.generator.FastAutoGenerator;
 import com.baomidou.mybatisplus.generator.config.OutputFile;
+import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
 import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
 import com.zhangmeng.tools.dto.BeanField;
 import com.zhangmeng.tools.dto.GenerateDetail;
@@ -223,7 +225,17 @@ public class MybatisPlusGenerator {
                             .pathInfo(Collections.singletonMap(OutputFile.mapperXml, dataSourceInfo.getMapper_xml_path())); // 设置mapperXml生成路径
                 }).strategyConfig(builder -> {
                     builder.addInclude(dataSourceInfo.getTables()) // 设置需要生成的表名
-                            .addTablePrefix(dataSourceInfo.getTable_prefixes()); // 设置过滤表前缀
+                            .addTablePrefix(dataSourceInfo.getTable_prefixes())
+                            .entityBuilder()
+                            .enableChainModel()
+                            .enableLombok()
+                            .enableRemoveIsPrefix()
+                            .enableTableFieldAnnotation()
+                            .enableActiveRecord()
+                            .naming(NamingStrategy.no_change)
+                            .columnNaming(NamingStrategy.underline_to_camel)
+                            .idType(IdType.AUTO)
+                            .build();
                 }).templateEngine(new FreemarkerTemplateEngine()) // 使用Freemarker引擎模板，默认的是Velocity引擎模板
                 .execute();
     }