AutoGenerator 是 MyBatis-Plus 的代码生成器，通过 AutoGenerator 可以快速生成 Entity、Mapper、Mapper XML、Service、Controller 等各个模块的代码，极大的提升了开发效率。可以通过模版等一系列的方式来生成代码，⚠️这个比Mybatis-Generator的更加强大，纯java代码。。
java代码如下：
package com.zxh.generator;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.builder.ConfigBuilder;import com.baomidou.mybatisplus.generator.config.po.TableInfo;import com.baomidou.mybatisplus.generator.config.rules.FileType;import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;import java.io.File;import java.util.ArrayList;import java.util.List;import java.util.Scanner;// 演示例子，执行 main 方法控制台输入模块表名回车自动生成对应项目目录中public class CodeGenerator { /**
** 读取控制台内容
**/
public static String scanner(String tip) {
Scanner scanner = new Scanner(System.in);
StringBuilder help = new StringBuilder();
help.append("请输入" + tip + "：");
System.out.println(help.toString()); if (scanner.hasNext()) {
String ipt = scanner.next(); if (StringUtils.isNotEmpty(ipt)) { return ipt;
}
} throw new MybatisPlusException("请输入正确的" + tip + "！");
} private static void generatorFile(){ // 代码生成器
AutoGenerator mpg = new AutoGenerator(); // 全局配置
GlobalConfig gc = new GlobalConfig();
String projectPath = System.getProperty("user.dir")+"/zxh-core";
gc.setOutputDir(projectPath + "/src/main/java");
gc.setAuthor("马家立");
gc.setOpen(false);
gc.setBaseResultMap(true);
gc.setBaseColumnList(true); // 是否覆盖已有文件
gc.setFileOverride(false); // gc.setSwagger2(true); 实体属性 Swagger2 注解 mpg.setGlobalConfig(gc); // 数据源配置
DataSourceConfig dsc = new DataSourceConfig();
dsc.setUrl("jdbc:mysql://47.115.177.5:3306/finder?useUnicode=true&useSSL=false&characterEncoding=utf8"); // dsc.setSchemaName("public");
dsc.setDriverName("com.mysql.cj.jdbc.Driver");
dsc.setUsername("springclouduser");
dsc.setPassword("Djl^ZOXn2R");
mpg.setDataSource(dsc); // 包配置
PackageConfig pc = new PackageConfig();// pc.setModuleName(scanner("src内的模块名"));
pc.setParent("com.zxh");// pc.setEntity("entity");// pc.setService("service");// pc.setServiceImpl("service.impl");// pc.setMapper("mapper"); 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"; // 自定义输出配置
ListfocList = 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.setFileCreate(new IFileCreate() {// @Override// public boolean isCreate(ConfigBuilder configBuilder, FileType fileType, String filePath) {// // 判断自定义文件夹是否需要创建// checkDir("调用默认方法创建的目录，自定义目录用");// if (fileType == FileType.MAPPER) {// // 已经生成 mapper 文件判断存在，不想重新生成返回 false// return !new File(filePath).exists();// }// // 允许生成模板文件// return true;// }// }); cfg.setFileOutConfigList(focList);
mpg.setCfg(cfg); // 配置模板
TemplateConfig templateConfig = new TemplateConfig(); // 配置自定义输出模板 //指定自定义模板路径，注意不要带上.ftl/.vm, 会根据使用的模板引擎自动识别 // templateConfig.setEntity("templates/entity2.java"); // templateConfig.setService();
templateConfig.setController("");
mpg.setTemplate(templateConfig);// templateConfig.setXml(null);
mpg.setTemplateEngine(new FreemarkerTemplateEngine()); // 策略配置
StrategyConfig strategy = new StrategyConfig();
strategy.setEntityTableFieldAnnotationEnable(true);
strategy.setNaming(NamingStrategy.underline_to_camel);
strategy.setColumnNaming(NamingStrategy.underline_to_camel);// strategy.setSuperEntityClass("你自己的父类实体,没有就不用设置!");
strategy.setEntityLombokModel(true);// strategy.setRestControllerStyle(true); // 公共父类// strategy.setSuperControllerClass("你自己的父类控制器,没有就不用设置!"); // 写于父类中的公共字段// strategy.setSuperEntityColumns("id"); //要生成的数据库表
strategy.setInclude("t_user_audit");// strategy.setInclude(scanner("user,stock_code").split(","));// strategy.setControllerMappingHyphenStyle(true);// strategy.setTablePrefix(pc.getModuleName() + "_"); mpg.setStrategy(strategy);// mpg.setTemplateEngine(new FreemarkerTemplateEngine()); mpg.execute();
} public static void main(String[] args) {
generatorFile();
}
}
View Code
其中需要的maven依赖如下：
com.baomidoumybatis-plus-boot-starter3.0-RELEASEcom.baomidoumybatis-plus-generator3.0.7.1org.apache.velocityvelocity-engine-core2.0org.freemarkerfreemarker2.3.23com.ibeetlbeetl2.2.5
View Code
运行输入表名就可以了！！！！