MyBatis Plus Generator

  1. <!-- pom -->
  2. <dependency>
  3.     <groupId>com.baomidou</groupId>
  4.     <artifactId>mybatis-plus-generator</artifactId>
  5.     <version>3.4.0</version>
  6. </dependency>
  7. <dependency>
  8.     <groupId>org.apache.velocity</groupId>
  9.     <artifactId>velocity</artifactId>
  10.     <version>1.7</version>
  11. </dependency>
  1. import com.baomidou.mybatisplus.annotation.DbType;
  2. import com.baomidou.mybatisplus.core.toolkit.StringPool;
  3. import com.baomidou.mybatisplus.generator.AutoGenerator;
  4. import com.baomidou.mybatisplus.generator.InjectionConfig;
  5. import com.baomidou.mybatisplus.generator.config.*;
  6. import com.baomidou.mybatisplus.generator.config.po.TableInfo;
  7. import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
  9. import java.util.ArrayList;
  10. import java.util.List;
  12. /**
  13.  * @author fengyuhao
  14.  * @Description
  15.  */
  16. public class MpGenerator {
  18.     final static String dirPath = System.getProperty("user.dir") + "/src/main/java";
  19.     private static String[] tables = new String[]{"userdemo"}; // 配置生成指定的表
  21.     private static DbType dbType = DbType.POSTGRE_SQL;
  22.     private static String url = "jdbc:postgresql://localhost:5432/mybatisplus?useUnicode=true&" +
  23.             "characterEncoding=utf8&autoReconnect=true&zeroDateTimeBehavior=convertToNull&" +
  24.             "transformedBitIsBoolean=true&serverTimezone=GMT%2B8";
  25.     private static String UserName = "postgres";
  26.     private static String PassWord = "953598751";
  27.     private static String DriverName = "org.postgresql.Driver";
  29. //    private static DbType dbType = DbType.MYSQL;
  30. //    private static String UserName = "root";
  31. //    private static String PassWord = "123456";
  32. //    private static String DriverName = "com.mysql.cj.jdbc.Driver";
  33. //    private static String url ="jdbc:mysql://localhost:3306/map3d?characterEncoding=utf8&serverTimezone=UTC";
  35.     public static void main(String[] args) {
  36.         AutoGenerator mpg = new AutoGenerator();
  37.         // 全局配置
  38.         GlobalConfig gc = new GlobalConfig();
  39.         gc.setOutputDir(dirPath);
  40.         gc.setAuthor("fyh");
  41.         gc.setOpen(false); // 不打开文件夹
  42.         gc.setFileOverride(true); //是否覆盖
  43.         gc.setActiveRecord(true);// 不需要ActiveRecord特性的请改为false
  44.         gc.setEnableCache(false);// XML 二级缓存
  45.         gc.setBaseResultMap(true);// XML ResultMap
  46.         gc.setBaseColumnList(true);// XML columList
  48.         // 自定义文件命名，注意 %s 会自动填充表实体属性！
  49.         gc.setMapperName("%sDao");
  50. //        gc.setXmlName("%sMapper");
  51.         gc.setServiceName("%sService");
  52.         gc.setServiceImplName("%sServiceImpl");
  53.         gc.setControllerName("%sController");
  54.         mpg.setGlobalConfig(gc);
  56.         // 数据源配置
  57.         DataSourceConfig dsc = new DataSourceConfig();
  58.         dsc.setDbType(dbType);
  59.         dsc.setDriverName(DriverName);
  60.         dsc.setUsername(UserName);//?serverTimezone=GMT%2B8
  61.         dsc.setPassword(PassWord);
  62.         dsc.setUrl(url);
  63.         mpg.setDataSource(dsc);
  65.         // 策略配置
  66.         StrategyConfig strategy = new StrategyConfig();
  67.         // strategy.setCapitalMode(true);// 全局大写命名 ORACLE 注意
  68.         //strategy.setTablePrefix(new String[] { "tb_", "tsys_" });// 此处可以修改表前缀
  69.         strategy.setEntityLombokModel(true);// 自动添加 lombok 注解
  70.         strategy.setNaming(NamingStrategy.underline_to_camel);// 表名生成策略
  71.         strategy.setColumnNaming(NamingStrategy.underline_to_camel);
  72.         strategy.setInclude(tables); // 需要生成的表
  73.         // strategy.setExclude(new String[]{"test"}); // 排除生成的表
  74.         // 自定义实体父类
  75.         // strategy.setSuperEntityClass("com.baomidou.demo.TestEntity");
  76.         // 自定义实体，公共字段
  77.         // strategy.setSuperEntityColumns(new String[] { "test_id", "age" });
  78.         // 自定义 mapper 父类
  79.         // strategy.setSuperMapperClass("com.baomidou.demo.TestMapper");
  80.         // 自定义 service 父类
  81.         strategy.setSuperServiceClass("com.baomidou.mybatisplus.extension.service.IService");
  82.         // 自定义 service 实现类父类
  83.         strategy.setSuperServiceImplClass("com.baomidou.mybatisplus.extension.service.impl.ServiceImpl");
  84.         // 自定义 controller 父类
  85.         // strategy.setSuperControllerClass("com.baomidou.demo.TestController");
  86.         // 【实体】是否生成字段常量（默认 false）
  87.         // public static final String ID = "test_id";
  88.         // strategy.setEntityColumnConstant(true);
  89.         // 【实体】是否为构建者模型（默认 false）
  90.         // public User setName(String name) {this.name = name; return this;}
  91. //        strategy.setEntityBuilderModel(true);
  92.         mpg.setStrategy(strategy);
  94.         // 自定义配置
  95.         InjectionConfig cfg = new InjectionConfig() {
  96.             @Override
  97.             public void initMap() {
  98.                 // to do nothing
  99.             }
  100.         };
  101.         // 自定义输出配置
  102.         List<FileOutConfig> focList = new ArrayList<>();
  103.         // 自定义配置优先于默认配置生效
  104.         String mapperXmlPath = "/templates/mapper.xml.vm";
  105.         focList.add(new FileOutConfig(mapperXmlPath) {
  106.             @Override
  107.             public String outputFile(TableInfo tableInfo) {
  108.                 // 自定义xml 文件名和生成路径
  109.                 return System.getProperty("user.dir") + "/src/main/resources/mapper/" + tableInfo.getEntityName()+"Mapper" + StringPool.DOT_XML;
  110.             }
  111.         });
  112.         cfg.setFileOutConfigList(focList);
  113.         mpg.setCfg(cfg);
  114.         // 包配置
  115.         PackageConfig pc = new PackageConfig();
  116.         pc.setParent("com.cug.seckill")
  117. //      .setModuleName("module")
  118.         .setController("controller")
  119.         .setEntity("entity")
  120.         .setMapper("mapper")
  121.         .setService("service")
  122.         .setServiceImpl("service.impl");
  123. //        pc.setXml("mapper");
  124.         mpg.setPackageInfo(pc);
  126.         // 执行生成
  127.         mpg.execute();
  128.     }
  129. }