0
点赞
收藏
分享

微信扫一扫

Mybatis代码生成器

静鸡鸡的JC 2022-04-05 阅读 65
java

目录

添加依赖jar包

添加执行工具类


添加依赖jar包

            <dependency>
                <groupId>com.baomidou</groupId>
                <artifactId>mybatis-plus-generator</artifactId>
                <version>3.3.0</version>
            </dependency>
            <dependency>
                <groupId>mysql</groupId>
                <artifactId>mysql-connector-java</artifactId>
                <version>5.1.25</version>
            </dependency>
            <dependency>
                <groupId>org.apache.velocity</groupId>
                <artifactId>velocity-engine-core</artifactId>
                <version>2.0</version>
            </dependency>

添加执行工具类

        public class Generator {
              public static void main(String[] args) throws IOException {
                  AutoGenerator mpg = new AutoGenerator();// 代码生成器
                  GlobalConfig gc = new GlobalConfig(); // 全局配置
                  String projectPath = "D:\\parkson-platform-dev\\account"; //当前项目路径
                  gc.setOutputDir(projectPath + "/src/main/java");//输出路径
                  gc.setAuthor("parkson");//作者
                  gc.setOpen(false);
                  gc.setSwagger2(true);//实体属性 Swagger2 注解
                  mpg.setGlobalConfig(gc);
                  DataSourceConfig dsc = new DataSourceConfig();// 数据源配置
                  //设置数据库地址
                  dsc.setUrl("jdbc:mysql://200.86.3.85:8003/parkson_account_dev?serverTimezone=UTC&characterEncoding=UTF-8");
                  //dsc.setSchemaName("public");
                  dsc.setDriverName("com.mysql.jdbc.Driver");
                  dsc.setUsername("root");
                  dsc.setPassword("parkson1016");
                  mpg.setDataSource(dsc);
                  PackageConfig pc = new PackageConfig();// 包配置
                  // pc.setModuleName(scanner("模块名"));
                  pc.setModuleName("");
                  pc.setParent("com.parkson.platform.account");//生成的代码路径
                  mpg.setPackageInfo(pc);

                  // 自定义配置
                  InjectionConfig cfg = new InjectionConfig() {
                      @Override
                      public void initMap() {
                          // to do nothing
                      }
                  };

                  // 配置自定义输出模板
                  String templatePath = "/templates/mapper.xml.vm";
                  List<FileOutConfig> focList = new ArrayList<>();
                  focList.add(new FileOutConfig(templatePath) {
                      @Override
                      public String outputFile(TableInfo tableInfo) {
                          // 自定义输出文件名
                          return projectPath + "/src/main/resources/mapper/"
                                  + "/" + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;
                      }
                  });
                  cfg.setFileOutConfigList(focList);
                  mpg.setCfg(cfg);

                  // 配置模板
                  TemplateConfig templateConfig = new TemplateConfig();
                  templateConfig.setXml(null);
                  mpg.setTemplate(templateConfig);

                  // 策略配置
                  StrategyConfig strategy = new StrategyConfig();
                  strategy.setNaming(NamingStrategy.underline_to_camel);
                  strategy.setColumnNaming(NamingStrategy.underline_to_camel);;
                  strategy.setEntityLombokModel(true);
                  strategy.setRestControllerStyle(true);
                  strategy.setControllerMappingHyphenStyle(true);
                  strategy.setTablePrefix(pc.getModuleName() + "_");
                  strategy.setInclude("t_vcard");
                  mpg.setStrategy(strategy);
                  //mpg.setTemplateEngine(new VelocityTemplateEngine());
                  mpg.execute();
              }
      }

示例2

    <dependency>
        <groupId>com.oracle</groupId>
        <artifactId>ojdbc7</artifactId>
        <version>12.1.0.2</version>
    </dependency>
  /**
   * Oracle数据库
   * mybatis-plus代码生成器
   */
  public class Generator {

      public static void main(String[] args) {
          AutoGenerator mpg = new AutoGenerator();

          // 全局配置
          GlobalConfig gc = new GlobalConfig();
          gc.setOutputDir("E:\\Mybatis_Plus");// 生产文件所在的目录
          gc.setFileOverride(true); // 默认 false ,是否覆盖已生成文件
          gc.setOpen(false); //默认true ,是否打开文件生产所在的目录
          gc.setIdType(IdType.ASSIGN_UUID); // 指定生成的主键类型
          gc.setActiveRecord(true);
          gc.setEnableCache(false);// XML 二级缓存
          gc.setBaseResultMap(true);// XML ResultMap
          gc.setBaseColumnList(false);// XML columList
          gc.setAuthor("jian.xing@getech.cn");
          // 生成文件命名('%s'会自动填充表名)
          gc.setMapperName("%sMapper");
          gc.setXmlName("%sMapper");
          gc.setServiceName("%sService");
          gc.setServiceImplName("%sServiceImpl");
          gc.setControllerName("%sController");
          gc.setSwagger2(true);
          mpg.setGlobalConfig(gc);

          // 数据源配置
          DataSourceConfig dsc = new DataSourceConfig();
          dsc.setDbType(DbType.ORACLE);
          dsc.setDriverName("oracle.jdbc.driver.OracleDriver");
          dsc.setUsername("qms");
          dsc.setPassword("csotqms1129");
          dsc.setUrl("jdbc:oracle:thin:@10.108.234.130:1521:d2pidb1");
          mpg.setDataSource(dsc);

          // 策略配置
          StrategyConfig strategy = new StrategyConfig();
          //strategy.setTablePrefix(new String[]{"D_"});// 设置表前缀
          strategy.setNaming(NamingStrategy.underline_to_camel);// 表名生成策略
          strategy.setInclude(new String[]{"QMS_LAB_APP_RESULTS"}); // 需要生成的表
          mpg.setStrategy(strategy);

          // 文件生产所在目录配置
          PackageConfig pc = new PackageConfig();
          pc.setParent("com.csot.qms");
          pc.setEntity("entity");
          pc.setMapper("mapper");
          pc.setXml("mapper");
          pc.setService("service");
          pc.setServiceImpl("service.impl");
          pc.setController("controller");
          mpg.setPackageInfo(pc);

          // 注入自定义配置,可以在VM中使用cfg.abc设置的值
          InjectionConfig cfg = new InjectionConfig() {
              @Override
              public void initMap() {
              }
          };
          // 配置模板
          TemplateConfig tc = new TemplateConfig();
          // templates/entity.java 模板路径配置,默认在templates目录下,.vm 后缀不用加
          tc.setEntity("templates/entity.java");//使用自定义模板生成实体类
          tc.setXml("");
          //mpg.setTemplate(tc);
          mpg.setCfg(cfg);
          // 执行生成
          mpg.execute();
          // 打印注入设置
          //System.err.println(mpg.getCfg().getMap().get("abc"));
      }
  }
举报

相关推荐

0 条评论