测试DB2代码自动生成

mybatis-plus-generate/src/test/java/com/baomidou/mybatisplus/test/generator/DB2Generator.java

@@ -0,0 +1,62 @@
+package com.baomidou.mybatisplus.test.generator;
+
+import com.baomidou.mybatisplus.generator.AutoGenerator;
+import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
+import com.baomidou.mybatisplus.generator.config.GlobalConfig;
+import com.baomidou.mybatisplus.generator.config.PackageConfig;
+import com.baomidou.mybatisplus.generator.config.StrategyConfig;
+import com.baomidou.mybatisplus.generator.config.rules.DbType;
+import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
+
+public class DB2Generator {
+	
+	public static void main(String[] args) {
+		String packageName = "com.crea.gwms";
+        boolean serviceNameStartWithI = true;//user -> UserService, 设置成true: user -> IUserService
+        generateByTables(serviceNameStartWithI, packageName, "USER");
+	}
+	
+	public static void generateByTables(boolean serviceNameStartWithI, String packageName, String... tableNames) {
+        GlobalConfig config = new GlobalConfig();
+        String dbUrl = "jdbc:db2://192.168.0.227:50000/TRADECTR";
+        DataSourceConfig dataSourceConfig = new DataSourceConfig();
+        dataSourceConfig.setDbType(DbType.DB2)
+                .setUrl(dbUrl)
+                .setUsername("tc")
+                .setPassword("business")
+                .setDriverName("com.ibm.db2.jcc.DB2Driver");
+        StrategyConfig strategyConfig = new StrategyConfig();
+        strategyConfig
+                .setCapitalMode(true)
+                .setEntityLombokModel(false)
+                .setDbColumnUnderline(true)
+                .setNaming(NamingStrategy.underline_to_camel)
+                .entityTableFieldAnnotationEnable(true)//实体属性上添加表字段映射
+                .setInclude(tableNames);//修改替换成你需要的表名，多个表名传数组
+        config.setActiveRecord(true)
+                .setEnableCache(false)// XML 二级缓存
+                .setAuthor("zhanyao")
+                .setBaseResultMap(true)// XML ResultMap
+                .setBaseColumnList(false)// XML columList
+                .setOutputDir("/Users/zhanyao/Downloads/code")
+                .setFileOverride(true);
+        if (!serviceNameStartWithI) {
+            config.setServiceName("%sService");
+        }
+        new AutoGenerator().setGlobalConfig(config)
+                .setDataSource(dataSourceConfig)
+                .setStrategy(strategyConfig)
+                .setPackageInfo(
+                        new PackageConfig()
+                                .setParent(packageName)
+                                .setController("controller")
+                                .setMapper("dao")
+                                .setEntity("bean")
+                ).execute();
+    }
+
+    private void generateByTables(String packageName, String... tableNames) {
+        generateByTables(true, packageName, tableNames);
+    }
+    
+}