第一Maven中添加依赖
点击查看代码
<!-- MyBatis-Plus 扩展库 --><dependency><groupId>com.baomidou</groupId><artifactId>mybatis-plus-extension</artifactId><version>3.5.7</version></dependency><dependency><groupId>com.baomidou</groupId><artifactId>mybatis-plus-generator</artifactId><version>3.4.1</version></dependency><dependency><groupId>org.freemarker</groupId><artifactId>freemarker</artifactId><version>2.3.30</version></dependency><dependency><groupId>com.spring4all</groupId><artifactId>spring-boot-starter-swagger</artifactId><version>1.5.1.RELEASE</version></dependency>
第二添加代码生成器代码
文件格式：
具体代码如下
点击查看代码
package com.example.demo.common;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.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
public class CodeGenerator {/*** <p>*读取控制台内容* </p>*/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.isNotBlank(ipt)) {return ipt;}}throw new MybatisPlusException("请输⼊正确的" + tip + "！");}/***操作步骤：* 1.修改数据源包括地址密码信息，对应代码标记：⼀、 下同* 2.模块配置，可以修改包名* 3.修改模板（这步可忽略）* @param args*/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("wms");//作者名称gc.setOpen(false);gc.setSwagger2(true); //实体属性 Swagger2 注解gc.setBaseResultMap(true);// XML ResultMapgc.setBaseColumnList(true);// XML columList//去掉service接⼝⾸字⺟的I, 如DO为User则叫UserServicegc.setServiceName("%sService");mpg.setGlobalConfig(gc);//数据源配置//⼀、修改数据源DataSourceConfig dsc = new DataSourceConfig();//dsc.setUrl("jdbc:mysql://localhost:3306/wns?useUnicode=true&characterEncoding=UTF8&useSSL=false");// dsc.setSchemaName("public");dsc.setDriverName("com.mysql.cj.jdbc.Driver");dsc.setUsername("root");dsc.setPassword("1234");mpg.setDataSource(dsc);//包配置PackageConfig pc = new PackageConfig();//pc.setModuleName(scanner("模块名"));//⼆、模块配置pc.setParent("com.example.demo").setEntity("entity").setMapper("mapper").setService("service").setServiceImpl("service.impl").setController("controller");mpg.setPackageInfo(pc);//⾃定义配置InjectionConfig cfg = new InjectionConfig() {@Overridepublic void initMap() {// to do nothing}};//如果模板引擎是 freemarkerString templatePath = "templates/mapper.xml.ftl";//如果模板引擎是 velocity// String templatePath = "/templates/mapper.xml.vm";//⾃定义输出配置List<FileOutConfig> focList = new ArrayList<>();//⾃定义配置会被优先输出focList.add(new FileOutConfig(templatePath) {@Overridepublic String outputFile(TableInfo tableInfo) {//⾃定义输出⽂件名 ， 如果你 Entity 设置了前后缀、此处注意 xml 的名称会跟着发⽣变化！！return projectPath + "/src/main/resources/mapper/" + pc.getModuleName()+ "/" + tableInfo.getEntityName() + "Mapper" +StringPool.DOT_XML;}});/*cfg.setFileCreate(new IFileCreate() {@Overridepublic boolean isCreate(ConfigBuilder configBuilder, FileType fileType,
String filePath) {//
判断⾃定义⽂件夹是否需要创建checkDir("调⽤默认⽅法创建的⽬录，⾃定义⽬录⽤");if (fileType == FileType.MAPPER) {//
已经⽣成 mapper ⽂件判断存在，不想重新⽣成返回 falsereturn !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("templates/service2.java");templateConfig.setController("templates/controller2.java");templateConfig.setMapper("templates/mapper2.java");templateConfig.setServiceImpl("templates/serviceimpl2.java");*/templateConfig.setXml(null);mpg.setTemplate(templateConfig);//策略配置StrategyConfig strategy = new StrategyConfig();strategy.setNaming(NamingStrategy.underline_to_camel);strategy.setColumnNaming(NamingStrategy.underline_to_camel);// strategy.setSuperEntityClass("你⾃⼰的⽗类实体,没有就不⽤设置!");//strategy.setSuperEntityClass("BaseEntity");strategy.setEntityLombokModel(true);strategy.setRestControllerStyle(true);//公共⽗类//strategy.setSuperControllerClass("BaseController");// strategy.setSuperControllerClass("你⾃⼰的⽗类控制器,没有就不⽤设置!");//写于⽗类中的公共字段// strategy.setSuperEntityColumns("id");strategy.setInclude(scanner("表名，多个英⽂逗号分割").split(","));strategy.setControllerMappingHyphenStyle(true);//strategy.setTablePrefix(pc.getModuleName() + "_");//忽略表前缀tb_,⽐如说tb_user,直接映射成user对象//四、注意是否要去掉表前缀//strategy.setTablePrefix("tb_");mpg.setStrategy(strategy);mpg.setTemplateEngine(new FreemarkerTemplateEngine());mpg.execute();}
}
