Mybatis-Plus-AutoGenerator 最详细使用方法!
作者:互联网
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
运行输入表名就可以了!!!!
标签:AutoGenerator,mybatisplus,baomidou,strategy,Plus,new,Mybatis,import,com 来源: https://blog.51cto.com/u_15222067/2820701