diff --git a/src/main/java/com/zhangmeng/tools/controller/MybatisPlusGenerator.java b/src/main/java/com/zhangmeng/tools/controller/MybatisPlusGenerator.java index 4347b9d..7a15d2c 100644 --- a/src/main/java/com/zhangmeng/tools/controller/MybatisPlusGenerator.java +++ b/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(); }