root
/
mystyle-java-fx-tools
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

master
zhangmeng 2023-03-23 11:34:09 +08:00
parent 73b69d3b81
commit 811ce9e57e
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/main/java/com/zhangmeng/tools/controller/MybatisPlusGenerator.java
View File

@ -2,8 +2,10 @@ package com.zhangmeng.tools.controller;
import cn.hutool.core.text.StrFormatter; import cn.hutool.core.text.StrFormatter;
import com.alibaba.druid.pool.DruidDataSource; import com.alibaba.druid.pool.DruidDataSource;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.generator.FastAutoGenerator; import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.OutputFile; import com.baomidou.mybatisplus.generator.config.OutputFile;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine; import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
import com.zhangmeng.tools.dto.BeanField; import com.zhangmeng.tools.dto.BeanField;
import com.zhangmeng.tools.dto.GenerateDetail; import com.zhangmeng.tools.dto.GenerateDetail;
@ -223,7 +225,17 @@ public class MybatisPlusGenerator {
.pathInfo(Collections.singletonMap(OutputFile.mapperXml, dataSourceInfo.getMapper_xml_path())); // 设置mapperXml生成路径 .pathInfo(Collections.singletonMap(OutputFile.mapperXml, dataSourceInfo.getMapper_xml_path())); // 设置mapperXml生成路径
}).strategyConfig(builder -> { }).strategyConfig(builder -> {
builder.addInclude(dataSourceInfo.getTables()) // 设置需要生成的表名 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引擎模板 }).templateEngine(new FreemarkerTemplateEngine()) // 使用Freemarker引擎模板，默认的是Velocity引擎模板
.execute(); .execute();
} }