Bladeren bron

增加测试案例

steven.ma 4 jaren geleden
bovenliggende
commit
d20f87958e

+ 1 - 0
build.gradle

@@ -73,6 +73,7 @@ ext {
         "h2"                         : "com.h2database:h2:1.4.200",
         "mysql"                      : "mysql:mysql-connector-java:8.0.21",
         "sqlite"                     : "org.xerial:sqlite-jdbc:3.32.3.1",
+        "firebird"                   : "org.firebirdsql.jdbc:jaybird:4.0.1.java8",
         //code generator
         "velocity"                   : "org.apache.velocity:velocity-engine-core:2.2",
         "freemarker"                 : "org.freemarker:freemarker:2.3.30",

+ 1 - 0
mybatis-plus-generator/build.gradle

@@ -13,5 +13,6 @@ dependencies {
     testImplementation "${lib.h2}"
     testImplementation "${lib.mysql}"
     testImplementation "${lib.sqlite}"
+    testImplementation "${lib.firebird}"
     testImplementation "${lib.'swagger-annotations'}"
 }

+ 155 - 0
mybatis-plus-generator/src/test/java/com/baomidou/mybatisplus/test/generator/FirebirdGenerator.java

@@ -0,0 +1,155 @@
+/*
+ * Copyright (c) 2011-2020, baomidou (jobob@qq.com).
+ * <p>
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may not
+ * use this file except in compliance with the License. You may obtain a copy of
+ * the License at
+ * <p>
+ * https://www.apache.org/licenses/LICENSE-2.0
+ * <p>
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+ * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+ * License for the specific language governing permissions and limitations under
+ * the License.
+ */
+package com.baomidou.mybatisplus.test.generator;
+
+import com.baomidou.mybatisplus.annotation.DbType;
+import com.baomidou.mybatisplus.annotation.IdType;
+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.converts.OracleTypeConvert;
+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.IColumnType;
+import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
+import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * @author steven ma
+ * @since 2019/10/12
+ */
+public class FirebirdGenerator extends GeneratorTest {
+
+    /**
+     * RUN THIS
+     */
+    //生成文件所在项目路径
+    private static String baseProjectPath = "D:\\devhome2020";
+
+    //基础包名
+    private static String basePackage="cn.perfecton.datatransfer";
+    //设置作者
+    private static String authorName="steven";
+    //这里是要生成的表名(如果全部要生成的话,这里注释掉)
+    //private static String[] tables= {"t_role","t_resource","t_role_resource","t_user_role"};
+    //可以设置table前缀
+//    private static String prefix="t_";
+
+    //数据库配置四要素
+    private static String driverName = "org.firebirdsql.jdbc.FBDriver";
+    private static String url = "jdbc:firebirdsql://localhost:3050/C:\\XXX.GDB?lc_ctype=UTF8";
+    private static String username = "sysdba";
+    private static String password = "XXXX";
+
+
+    public static void main(String[] args) {
+        // 代码生成器
+        AutoGenerator mpg = new AutoGenerator();
+
+        // 全局配置
+        GlobalConfig gc = new GlobalConfig();
+        String projectPath = System.getProperty("user.dir");
+        gc.setOutputDir(baseProjectPath + "/src/main/java");
+        // TODO 设置用户名
+        gc.setAuthor(authorName);
+        gc.setOpen(true);
+        // service 命名方式
+        gc.setServiceName("%sService");
+        // service impl 命名方式
+        gc.setServiceImplName("%sServiceImpl");
+        // 自定义文件命名,注意 %s 会自动填充表实体属性!
+        gc.setMapperName("%sMapper");
+        gc.setXmlName("%sMapper");
+        gc.setFileOverride(true);
+        gc.setActiveRecord(true);
+        // XML 二级缓存
+        gc.setEnableCache(false);
+        // XML ResultMap
+        gc.setBaseResultMap(true);
+        // XML columList
+        gc.setBaseColumnList(false);
+
+        //日期用date类型
+        gc.setDateType(DateType.ONLY_DATE);
+
+        mpg.setGlobalConfig(gc);
+
+        // TODO 数据源配置
+        DataSourceConfig dsc = new DataSourceConfig();
+        dsc.setUrl(url);
+        dsc.setDriverName(driverName);
+        dsc.setUsername(username);
+        dsc.setPassword(password);
+        mpg.setDataSource(dsc);
+
+        // TODO 包配置
+        PackageConfig pc = new PackageConfig();
+        //pc.setModuleName(scanner("模块名"));
+        pc.setParent(basePackage);
+        pc.setEntity("entity");
+        pc.setService("service");
+        pc.setServiceImpl("service.impl");
+        mpg.setPackageInfo(pc);
+
+        // 自定义需要填充的字段
+        List<TableFill> tableFillList = new ArrayList<>();
+
+        // 自定义配置
+        InjectionConfig cfg = new InjectionConfig() {
+            @Override
+            public void initMap() {
+                // to do nothing
+            }
+        };
+        List<FileOutConfig> focList = new ArrayList<>();
+        focList.add(new FileOutConfig("/templates/mapper.xml.ftl") {
+            @Override
+            public String outputFile(TableInfo tableInfo) {
+                // 自定义输入文件名称
+                return projectPath + "/src/main/resources/mapper/"
+                    + "/" + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;
+            }
+        });
+        cfg.setFileOutConfigList(focList);
+        mpg.setCfg(cfg);
+        mpg.setTemplate(new TemplateConfig().setXml(null));
+
+        // 策略配置
+        StrategyConfig strategy = new StrategyConfig();
+        strategy.setNaming(NamingStrategy.underline_to_camel);
+        strategy.setColumnNaming(NamingStrategy.underline_to_camel);
+        strategy.setEntityLombokModel(true);
+        // 设置逻辑删除键(这个是逻辑删除的操作)
+        strategy.setLogicDeleteFieldName("deleted");
+        // TODO 指定生成的bean的数据库表名(如果全部生成,这里要注释掉)
+
+        //strategy.setSuperEntityColumns("id");
+        // 驼峰转连字符
+        strategy.setControllerMappingHyphenStyle(true);
+        mpg.setStrategy(strategy);
+        // 选择 freemarker 引擎需要指定如下加,注意 pom 依赖必须有!
+        mpg.setTemplateEngine(new FreemarkerTemplateEngine());
+        mpg.execute();
+    }
+
+}