MyBatisPlus3.x中使用代码生成器(全注释)
场景
MyBaitsPlus3.x与2.x是不一样的。这里使用3.0.1版本。
官方文档
https://mp.baomidou.com/guide/generator.html
这里在IDEA上的SpringBoot项目中进行代码生成测试。
实现
添加依赖
添加代码生成器依赖
com.baomidou mybatis-plus-generator 3.2.0
注意：MyBatis-Plus从3.0.3之后移除了代码生成器与模板引擎的默认依赖，需要手动添加相关依赖
org.apache.velocity velocity-engine-core 2.0 com.baomidou mybatis-plus-boot-starter 3.0.1
模板引擎
MyBatis-Plus支持Velocity（默认）、Freemarker、Beetl，用户可以选择自己熟悉的模板引擎。
注意！如果您选择了非默认引擎，需要在AutoGenerator中设置模板引擎。
AutoGeneratorgenerator=newAutoGenerator(); //setfreemarkerengine generator.setTemplateEngine(newFreemarkerTemplateEngine()); //setbeetlengine generator.setTemplateEngine(newBeetlTemplateEngine()); //setcustomengine(referenceclassisyourcustomengineclass) generator.setTemplateEngine(newCustomTemplateEngine());
编写代码
在test下新建单元测试类Generatortest.java
全局配置
//全局配置
GlobalConfigconfig=newGlobalConfig();
//设置是否支持AR模式
config.setActiveRecord(true)
//设置生成代码的作者
.setAuthor("badaoliumangqizhi")
//设置输出代码的位置
.setOutputDir("f:output")
//.setEnableCache(false)//XML二级缓存
//.setBaseResultMap(true)//XMLResultMap
//.setBaseColumnList(true)//XMLcolumList
//.setKotlin(true)是否生成kotlin代码
//设置是否覆盖原来的代码
.setFileOverride(true);
数据源配置
//数据库连接url
StringdbUrl="jdbc:sqlserver://;DatabaseName=";
//数据源配置
DataSourceConfigdataSourceConfig=newDataSourceConfig();
//数据库类型枚举
dataSourceConfig.setDbType(DbType.SQL_SERVER)
//设置url
.setUrl(dbUrl)
//设置用户名
.setUsername("")
//设置密码
.setPassword("")
//设置数据库驱动
.setDriverName("com.microsoft.sqlserver.jdbc.SQLServerDriver")
//自定义数据库表字段类型转换【可选】
.setTypeConvert(newMySqlTypeConvert(){
@Override
publicDbColumnTypeprocessTypeConvert(GlobalConfigglobalConfig,StringfieldType){
System.out.println("转换类型："+fieldType);
//tinyint转换成Boolean
if(fieldType.toLowerCase().contains("tinyint")){
returnDbColumnType.BOOLEAN;
}
//将数据库中datetime转换成date
if(fieldType.toLowerCase().contains("datetime")){
returnDbColumnType.DATE;
}
return(DbColumnType)super.processTypeConvert(globalConfig,fieldType);
}
});
策略配置
//策略配置
StrategyConfigstrategyConfig=newStrategyConfig();
strategyConfig
//全局大写命名是否开启
.setCapitalMode(true)
//【实体】是否为lombok模型
.setEntityLombokModel(true)
//表名生成策略下划线转驼峰
.setNaming(NamingStrategy.underline_to_camel)
//自动填充设置
//.setTableFillList(tableFillList)
//修改替换成你需要的表名，多个表名传数组
.setInclude("wms_receive_order");
集成注入配置
//注入全局设置
newAutoGenerator().setGlobalConfig(config)
//注入数据源配置
.setDataSource(dataSourceConfig)
//注入策略配置
.setStrategy(strategyConfig)
//设置包名信息
.setPackageInfo(
newPackageConfig()
//提取公共父级包名
.setParent("com.badao.bus.sys")
//设置controller信息
.setController("controller")
//设置实体类信息
.setEntity("entity")
)
//设置自定义模板
.setTemplate(
newTemplateConfig()
//.setXml(null)//指定自定义模板路径,位置：/resources/templates/entity2.java.ftl(或者是.vm)
//注意不要带上.ftl(或者是.vm),会根据使用的模板引擎自动识别
//自定义模板配置，模板可以参考源码/mybatis-plus/src/main/resources/template使用copy
//至您项目src/main/resources/template目录下，模板名称也可自定义如下配置：
//.setController("...");
//.setEntity("...");
//.setMapper("...");
//.setXml("...");
//.setService("...");
.setServiceImpl("templates/serviceImpl.java")
)
//开始执行代码生成
.execute();
}
完整生成器代码
packagecom.ws.test.generator;
importcom.baomidou.mybatisplus.annotation.DbType;
importcom.baomidou.mybatisplus.annotation.FieldFill;
importcom.baomidou.mybatisplus.generator.config.*;
importcom.baomidou.mybatisplus.generator.config.converts.MySqlTypeConvert;
importcom.baomidou.mybatisplus.generator.config.po.TableFill;
importcom.baomidou.mybatisplus.generator.config.rules.DbColumnType;
importorg.junit.Test;
importcom.baomidou.mybatisplus.generator.AutoGenerator;
importcom.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
importjava.util.ArrayList;
importjava.util.List;
/**
*Createdbybadadoon2019/4/25.
*/
publicclassGeneratortest{
@Test
publicvoidgenerateTest(){
//全局配置
GlobalConfigconfig=newGlobalConfig();
//设置是否支持AR模式
config.setActiveRecord(true)
//设置生成代码的作者
.setAuthor("badaoliumangqizhi")
//设置输出代码的位置
.setOutputDir("f:output")
//.setEnableCache(false)//XML二级缓存
//.setBaseResultMap(true)//XMLResultMap
//.setBaseColumnList(true)//XMLcolumList
//.setKotlin(true)是否生成kotlin代码
//设置是否覆盖原来的代码
.setFileOverride(true);
//******************************数据源配置***************************************
//数据库连接url
StringdbUrl="jdbc:sqlserver://;DatabaseName=";
//数据源配置
DataSourceConfigdataSourceConfig=newDataSourceConfig();
//数据库类型枚举
dataSourceConfig.setDbType(DbType.SQL_SERVER)
//设置url
.setUrl(dbUrl)
//设置用户名
.setUsername("")
//设置密码
.setPassword("")
//设置数据库驱动
.setDriverName("com.microsoft.sqlserver.jdbc.SQLServerDriver")
//自定义数据库表字段类型转换【可选】
.setTypeConvert(newMySqlTypeConvert(){
@Override
publicDbColumnTypeprocessTypeConvert(GlobalConfigglobalConfig,StringfieldType){
System.out.println("转换类型："+fieldType);
//tinyint转换成Boolean
if(fieldType.toLowerCase().contains("tinyint")){
returnDbColumnType.BOOLEAN;
}
//将数据库中datetime转换成date
if(fieldType.toLowerCase().contains("datetime")){
returnDbColumnType.DATE;
}
return(DbColumnType)super.processTypeConvert(globalConfig,fieldType);
}
});
//******************************策略配置******************************************************
//自定义需要填充的字段数据库中的字段
ListtableFillList=newArrayList<>();
tableFillList.add(newTableFill("gmt_modified",FieldFill.INSERT_UPDATE));
tableFillList.add(newTableFill("modifier_id",FieldFill.INSERT_UPDATE));
tableFillList.add(newTableFill("creator_id",FieldFill.INSERT));
tableFillList.add(newTableFill("gmt_creat",FieldFill.INSERT));
tableFillList.add(newTableFill("available_flag",FieldFill.INSERT));
tableFillList.add(newTableFill("deleted_flag",FieldFill.INSERT));
tableFillList.add(newTableFill("sync_flag",FieldFill.INSERT));
//策略配置
StrategyConfigstrategyConfig=newStrategyConfig();
strategyConfig
//全局大写命名是否开启
.setCapitalMode(true)
//【实体】是否为lombok模型
.setEntityLombokModel(true)
//表名生成策略下划线转驼峰
.setNaming(NamingStrategy.underline_to_camel)
//自动填充设置
.setTableFillList(tableFillList)
//修改替换成你需要的表名，多个表名传数组
.setInclude("wms_receive_order");
//集成注入设置
//注入全局设置
newAutoGenerator().setGlobalConfig(config)
//注入数据源配置
.setDataSource(dataSourceConfig)
//注入策略配置
.setStrategy(strategyConfig)
//设置包名信息
.setPackageInfo(
newPackageConfig()
//提取公共父级包名
.setParent("com.badao.bus.sys")
//设置controller信息
.setController("controller")
//设置实体类信息
.setEntity("entity")
)
//设置自定义模板
.setTemplate(
newTemplateConfig()
//.setXml(null)//指定自定义模板路径,位置：/resources/templates/entity2.java.ftl(或者是.vm)
//注意不要带上.ftl(或者是.vm),会根据使用的模板引擎自动识别
//自定义模板配置，模板可以参考源码/mybatis-plus/src/main/resources/template使用copy
//至您项目src/main/resources/template目录下，模板名称也可自定义如下配置：
//.setController("...");
//.setEntity("...");
//.setMapper("...");
//.setXml("...");
//.setService("...");
.setServiceImpl("templates/serviceImpl.java")
)
//开始执行代码生成
.execute();
}
}
效果
运行测试
生成成功后会自动弹出生成代码的目录
声明：本文内容来源于网络，版权归原作者所有，内容由互联网用户自发贡献自行上传，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任。如果您发现有涉嫌版权的内容，欢迎发送邮件至：czq8825#qq.com（发邮件时，请将#更换为@）进行举报，并提供相关证据，一经查实，本站将立刻删除涉嫌侵权内容。