springboot整合mybatis-plus代码生成器的配置解析

package mybatis_plus;

import com.baomidou.mybatisplus.annotation.DbType;

import com.baomidou.mybatisplus.annotation.FieldFill;

import com.baomidou.mybatisplus.annotation.IdType;

import com.baomidou.mybatisplus.core.exceptions.MybatisPlusException;

import com.baomidou.mybatisplus.core.toolkit.StringPool;

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.TableFill;

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 org.apache测试数据mons.lang3.StringUtils;

import java.util.ArrayList;

import java.util.List;

import java.util.Scanner;

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.isNotBlank(ipt)) {

         return ipt;

       }

     }

     throw new MybatisPlusException( "请输入正确的" + tip + "！" );

   }

   public static void main(String[] args) {

     // 构建代码生成器

     AutoGenerator mpg = new AutoGenerator();

     //配置策略

     //1.全局配置

     GlobalConfig gc = new GlobalConfig();

     String projectPath = System.getProperty( "user.dir" );

     gc.setOutputDir(projectPath + "/src/main/java" );

     gc.setAuthor( "heroMps" );

     gc.setOpen( false );

     gc.setFileOverride( false ); //是否覆盖

     gc.setServiceName( "%sService" ); //去除Service前面的I

     gc.setIdType(IdType.ID_WORKER);

     gc.setDateType(DateType.ONLY_DATE);

//    gc.setSwagger2(true); //实体属性 Swagger2 注解

     mpg.setGlobalConfig(gc);

     //2.设置数据源配置

     DataSourceConfig dsc = new DataSourceConfig();

     dsc.setUrl( "jdbc:mysql://127.0.0.1:3306/mybatis_plus?useUnicode=true&useSSL=false&characterEncoding=utf8" );

     // dsc.setSchemaName("public");

     dsc.setDriverName( "com.mysql.jdbc.Driver" );

     dsc.setUsername( "root" );

     dsc.setPassword( "root" );

     dsc.setDbType(DbType.MYSQL);

     mpg.setDataSource(dsc);

     //3.包配置

     PackageConfig pc = new PackageConfig();

//    pc.setModuleName("blog");

     pc.setParent( "mybatis_plus" );

     pc.setEntity( "entity" );

     pc.setMapper( "mapper" );

     pc.setService( "service" );

     pc.setController( "controller" );

     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<FileOutConfig> focList = new ArrayList<>();

     // 自定义配置会被优先输出

     focList.add( new FileOutConfig(templatePath) {

       @Override

       public String outputFile(TableInfo tableInfo) {

         // 自定义输出文件名 ， 如果你 Entity 设置了前后缀、此处注意 xml 的名称会跟着发生变化！！

         return projectPath + "/src/main/resources/mapper/"

             + "/" + 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();

     templateConfig.setXml( null );

     mpg.setTemplate(templateConfig);

     //4.策略配置

     StrategyConfig strategy = new StrategyConfig();

     strategy.setInclude(scanner( "表名，多个英文逗号分割" ).split( "," ));

     strategy.setNaming(NamingStrategy.underline_to_camel);

     strategy.setColumnNaming(NamingStrategy.underline_to_camel);

     strategy.setEntityLombokModel( true );

     strategy.setLogicDeleteFieldName( "deleted" );

     //自动填充设置

     TableFill create_time = new TableFill( "create_time" , FieldFill.INSERT);

     TableFill update_time = new TableFill( "update_time" , FieldFill.INSERT_UPDATE);

     ArrayList<TableFill> list = new ArrayList<>();

     list.add(create_time);

     list.add(update_time);

     strategy.setTableFillList(list);

     //5.乐观锁配置

     strategy.setVersionFieldName( "version" );

     strategy.setRestControllerStyle( true );

     strategy.setControllerMappingHyphenStyle( true );

     mpg.setStrategy(strategy);

     mpg.execute();

   }

}