mybatisplus之代码生成器


先创建一个项目:

 引入web依赖:

GeneratorApp类 

  内容:

import com.baomidou.mybatisplus.core.toolkit.StringPool;
import com.baomidou.mybatisplus.generator.InjectionConfig;
import com.baomidou.mybatisplus.generator.config.*;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.config.po.LikeTable;
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;

public class GeneratorApp {

public static void main(String[] args) {
//代码生成器
AutoGenerator mpg = new AutoGenerator();
//全局配置
GlobalConfig gc = new GlobalConfig();
//或者当前项目的路劲
String projectPath = System.getProperty("user.dir");
//设置生成路径
gc.setOutputDir(projectPath+"/src/main/java");
//设置作者
gc.setAuthor("jobob");
//代码生成是不是要打开所在文件夹
gc.setOpen(false);
//生成Swagger2注解
gc.setSwagger2(true);
//会在mapper.xml生成一个基础的映射所有的字段
gc.setBaseResultMap(true);
//同文件生成覆盖
gc.setFileOverride(true);
//生成时间
gc.setDateType(DateType.ONLY_DATE);
//实体类:直接用表名 %s=表名
gc.setEntityName("%s");
//mapper接口名
gc.setMapperName("%sMapper");
//mapper.xml文件名
gc.setXmlName("%sMapper");
//业务逻辑类名
gc.setServiceName("%sService");
//业务逻辑类实现类名
gc.setServiceName("%sImplService");
//将全局配置设置到AutoGenerator
mpg.setGlobalConfig(gc);


//数据源配置
DataSourceConfig dsc = new DataSourceConfig();
dsc.setUrl("jdbc:mysql://localhost:3306/qxgl?useUnicode=true&useSSL=false&characterEncoding=utf8");
dsc.setDriverName("com.mysql.jdbc.Driver");
dsc.setUsername("root");
dsc.setPassword("root");
mpg.setDataSource(dsc);

//包配置
PackageConfig pc = new PackageConfig();
//模块名
pc.setModuleName("sys");
//包名
pc.setParent("com.dmsc.springboot.generator");
//完整的报名com.dmsc.springboot.generator.sys
mpg.setPackageInfo(pc);



// 自定义配置
InjectionConfig cfg = new InjectionConfig() {
@Override
public void initMap() {
// to do nothing
}
};

// 如果模板引擎是 freemarker
//String templatePath = "/templates/mapper.xml.ftl";
// 如果模板引擎是 velocity
String templatePath = "/templates/mapper.xml.vm";

// 自定义输出配置
List focList = new ArrayList<>();
// 自定义配置会被优先输出
focList.add(new FileOutConfig(templatePath) {
@Override
public String outputFile(TableInfo tableInfo) {
// 自定义输出文件名 , 如果你 Entity 设置了前后缀、此处注意 xml 的名称会跟着发生变化!!
return projectPath + "/src/main/resources/mapper/" + pc.getModuleName()
+ "/" + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;
}
});

cfg.setFileOutConfigList(focList);
mpg.setCfg(cfg);

// 配置模板
TemplateConfig templateConfig = new TemplateConfig();


templateConfig.setXml(null);
mpg.setTemplate(templateConfig);

//策略配置
StrategyConfig strategy = new StrategyConfig();
//表名胡生成策略:下划线转驼峰 pms_product--PmsProoduct
strategy.setNaming(NamingStrategy.underline_to_camel);
//列名的生成策略:下划线转驼峰last_name -- lastName
strategy.setColumnNaming(NamingStrategy.underline_to_camel);

//strategy.setSuperEntityClass("你自己的父亲实体,没有就不用设置");
//strategy.setEntityLombokModel(true);
//在controller类上是否生成@RestController
strategy.setRestControllerStyle(true);
//公共父类
//strategy.setSuperControllerClass("你自己的父类控制器,没有就不用设置");
//写于父类中的公共字段
// strategy.setSuperEntityColumns("id");
//要生成的表名 多个用逗号分隔
//strategy.setInclude("");
//按前缀生成表
strategy.setLikeTable(new LikeTable("sys_"));
//驼峰转字符 比如 pms_product ->controller @RequestMapping("/pms/pmsProduct") 按前端生成表
strategy.setControllerMappingHyphenStyle(true);
//设置表替换前缀
strategy.setTablePrefix("sys_");
mpg.setStrategy(strategy);
//进行生成
mpg.execute();
}
}

其中生成Swagger2和Lombok时候需要在pom文件进行配置否者报错:
  
//生成Swagger2注解
gc.setSwagger2(true);
//生成Lombok注解
strategy.setEntityLombokModel(true);

pom依赖如下:
  

org.springframework.boot
spring-boot-starter-parent
2.3.12.RELEASE


1.8
2.9.2
1.6.0
1.6.0



org.springframework.boot
spring-boot-starter-web


mysql
mysql-connector-java



com.baomidou
mybatis-plus-generator
3.4.1




org.apache.velocity
velocity-engine-core
2.2



io.springfox
springfox-swagger-ui
${swagger2.version}


io.swagger
swagger-models
${swagger-model.version}


io.swagger
swagger-annotations
${swagger-annotations.version}



org.projectlombok
lombok



运行就可以使用了。