无脑的代码,咱们能少写一行就少写一行,节约生命,从我做起!!!
CodeGenerator—SpringBoot整合MyBatisPlus代码生成器。
不想看过程的直接拉到最下面,完整代码我会放到最下面,开箱即用,各位猥大工程师自行CV!!!
先导入几个必要的依赖:
mybatis-plus的依赖
<!--mybatis-plus相关依赖-->
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-boot-starter</artifactId>
<version>3.4.2</version>
</dependency>
代码生成器依赖
<!-- mybatis-plus-generator代码生成器 -->
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-generator</artifactId>
<version>3.3.2</version>
</dependency>
选择一个模板引擎依赖
CodeGenerator支持多种模板引擎,比如:
1、Velocity (默认模板引擎)
2、Freemarker
3、Beetl
在这里我们就先用默认的,因为我这个人有选择恐惧症。
<!-- 模板引擎 -->
<dependency>
<groupId>org.apache.velocity</groupId>
<artifactId>velocity-engine-core</artifactId>
<version>2.0</version>
</dependency>
引入了上面的依赖之后,就能实例化出来一个AutoGenerator类,它是我们代码生成的核心,所有的配置都是围绕这个类进行的。
我们选中后,轻轻按压一下鼠标滚轮,进入到这个类中可以看到它有8个属性,并且每个属性都有自己的Get、Set方法,那么也就意味着,我们只要实例化这些对象,然过对其进行加工,最后再Set回去就能激活它的功能了!
就像这样:
为了能快速上手,我直接指出哪些地方需要修改成你的配置:
1、配置预编译类名:按照你自己的风格进行定义。
2、配置数据源:配置文件咋写你咋写。
3、配置自定义包名:我这个项目的父包路径是(“com.applet.mall.merchant.coding”),所以后面的包都是基于我的父包,来进行横向垂直往下生成的。
4、配置xml生成文件:你问我为什么要把mapper包放在dao包下面,而不是resource目录下面?别问,问就是有时候放在resource目录下面,Spring容器扫描不到,会导致项目起不来,放上面省事一些。
启动主类之后,在控制台输入表名(多表用逗号隔开)效果如下:
点击Enter之后,开始生成!
最后~~上才艺~:
package com.applet.mall.merchant.coding.utils;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.core.exceptions.MybatisPlusException;
import com.baomidou.mybatisplus.core.toolkit.StringPool;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.InjectionConfig;
import com.baomidou.mybatisplus.generator.config.*;
import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
/**
* 基于MybatisPlus的逆向工程
**/
public class MybatisPlusGeneratorUtil {
/**
* 读取控制台内容
*/
public static String scanner(String input) {
Scanner scanner = new Scanner(System.in);
System.out.println("请输入" + input + ":");
if (scanner.hasNext()) {
String ipt = scanner.next();
if (StringUtils.isNotBlank(ipt)) {
return ipt;
}
}
throw new MybatisPlusException("请输入正确的" + input + "!");
}
public static void main(String[] args) {
// 代码生成器
AutoGenerator mpg = new AutoGenerator();
// 全局配置
GlobalConfig gc = new GlobalConfig();
gc.setIdType(IdType.ASSIGN_UUID);
String projectPath = System.getProperty("user.dir");
gc.setOutputDir(projectPath + "/src/main/java").setAuthor("聪明的一休").setOpen(false).setFileOverride(true)
.setControllerName("%sController").setServiceName("I%sService").setServiceImplName("Default%sService").setMapperName("%sMapper").setXmlName("%sMapper");
//开启 BaseResultMap
gc.setBaseResultMap(true);
gc.setBaseColumnList(true);
//实体属性 Swagger2 注解(有人说都特么Swagger3了,还整这Swagger2!!!嘿嘿其实就代码而言没啥变化)
gc.setSwagger2(true);
//日期格式,采用Date
gc.setDateType(DateType.ONLY_DATE);
mpg.setGlobalConfig(gc);
// 数据源配置
DataSourceConfig dsc = new DataSourceConfig();
dsc.setUrl("jdbc:mysql://888.888.888.888:8888/mall_merchant?useUnicode=true&characterEncoding=utf8&useSSL=false&serverTimezone=GMT%2B8");
dsc.setDriverName("com.mysql.cj.jdbc.Driver");
dsc.setUsername("root");
dsc.setPassword("root");
mpg.setDataSource(dsc);
// 包配置
PackageConfig pc = new PackageConfig();
pc.setParent("com.applet.mall.merchant.coding").setController("controller").setService("service").setServiceImpl("service.impl").setEntity("bean.entity").setMapper("bean.dao");
mpg.setPackageInfo(pc);
// 修改xml文件路径
// 自定义配置
InjectionConfig cfg = new InjectionConfig() {
@Override
public void initMap() {
// to do nothing
}
};
// 如果模板引擎是 velocity
String templatePath = "/templates/mapper.xml.vm";
// 自定义输出配置
List<FileOutConfig> focList = new ArrayList<>();
// 自定义配置会被优先输出
// 这里自定义配置的是*Mapper.xml文件
// 所以templatePath = "/templates/mapper.xml.vm";
// 如果你想自定义配置其它 修改templatePath即可
focList.add(new FileOutConfig(templatePath) {
@Override
public String outputFile(TableInfo tableInfo) {
// 自定义输出文件名 如果你 Entity 设置了前后缀
String entityName = tableInfo.getEntityName();
return projectPath + "src/main/java/com/applet/mall/merchant/coding/dao/mapper/" +
entityName + "Mapper" + StringPool.DOT_XML;
}
});
cfg.setFileOutConfigList(focList);
mpg.setCfg(cfg);
// 配置模板
TemplateConfig templateConfig = new TemplateConfig()
// 配置自定义输出模板(这里就是我说的,根据模板引擎的不同,你需要写相对应的配置!!!)
//指定自定义模板路径,注意不要带上.ftl/.vm, 会根据使用的模板引擎自动识别
.setEntity("templates/entity.java")
.setMapper("templates/mapper.java")
.setService("templates/service.java")
.setServiceImpl("templates/serviceImpl.java")
.setController("templates/controller.java");
templateConfig.setXml(null);
mpg.setTemplate(templateConfig);
// 策略配置(这里你随意,不改也行)
StrategyConfig strategy = new StrategyConfig();
strategy.setEntitySerialVersionUID(true);
strategy.setNaming(NamingStrategy.underline_to_camel);
strategy.setColumnNaming(NamingStrategy.underline_to_camel);
strategy.setEntityLombokModel(true);
strategy.setRestControllerStyle(true);
strategy.setControllerMappingHyphenStyle(true);
strategy.setInclude(scanner("表名,多个英文逗号分割").split(","));
mpg.setStrategy(strategy);
mpg.execute();
}
}