Mybatis-Plus中的程式碼生成器
阿新 • • 發佈：2022-05-29
示例1：
import com.baomidou.mybatisplus.generator.AutoGenerator; import com.baomidou.mybatisplus.generator.InjectionConfig; import com.baomidou.mybatisplus.generator.config.*; import com.baomidou.mybatisplus.generator.config.rules.DbColumnType; import com.baomidou.mybatisplus.generator.config.rules.DbType;import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy; import org.junit.Test; import org.junit.runner.RunWith; import org.springframework.test.context.ContextConfiguration; import org.springframework.test.context.junit4.SpringRunner; import java.util.HashMap; import java.util.Map;/** * Created by Administrator on 2018/10/9. */ @RunWith(SpringRunner.class) @ContextConfiguration public class MpGenerator { @Test public void generateCode() { String packageName = "ame.biz.equipment"; /*generateByTables(new String[]{"ZD_", "ZR_", "Z_"}, packageName, "ACTIVITY", "ACTIVITY_PERM", "ACTIVITY_SETTING", "CUSTOMER", "CUSTOM_FIELDS","" "DC_GROUP", "DC_PARAMETER", "INVENTORY", "INVENTORY_ASSY_DATA", "ITEM", "ITEM_GROUP", "NC_CODE" );*/ generateByTables(new String[]{ "Z_"}, "ame.biz.equipment", "", "PG_MAN"); } /** *ORACLE 生成演示 * @param tablePerfix 表字首 * @param packageName 包名 * @param versionFieldName 樂觀鎖欄位名稱 * @param tableNames 表名 */ public void generateByTables(String[] tablePerfix, String packageName, String versionFieldName, String... tableNames) { AutoGenerator mpg = new AutoGenerator(); // 選擇 freemarker 引擎，預設 Veloctiy // mpg.setTemplateEngine(new FreemarkerTemplateEngine()); String basePath = "C://#tmp"; // 全域性配置 GlobalConfig gc = new GlobalConfig(); gc.setOutputDir(basePath); gc.setFileOverride(true); gc.setActiveRecord(true);// 不需要ActiveRecord特性的請改為false gc.setEnableCache(false);// XML 二級快取 gc.setBaseResultMap(true);// XML ResultMap gc.setBaseColumnList(true);// XML columList //gc.setKotlin(true);//是否生成 kotlin 程式碼 gc.setAuthor("Syngna"); gc.setServiceName("%sService"); // 自定義檔案命名，注意 %s 會自動填充表實體屬性！ // gc.setMapperName("%sDao"); // gc.setXmlName("%sDao"); // gc.setServiceName("MP%sService"); // gc.setServiceImplName("%sServiceDiy"); // gc.setControllerName("%sAction"); mpg.setGlobalConfig(gc); // 資料來源配置 DataSourceConfig dsc = new DataSourceConfig(); dsc.setDbType(DbType.ORACLE); /*dsc.setTypeConvert(new OracleTypeConvert() { // 自定義資料庫表字段型別轉換【可選】 @Override public DbColumnType processTypeConvert(String fieldType) { System.out.println("轉換型別：" + fieldType); // 注意！！processTypeConvert 存在預設型別轉換，如果不是你要的效果請自定義返回、非如下直接返回。 return super.processTypeConvert(fieldType); } });*/ // 自定義資料庫表字段型別轉換 dsc.setTypeConvert((fieldType) -> { String t = fieldType.toLowerCase(); if (t.contains("char") || t.contains("text")) { return DbColumnType.STRING; } else if (t.contains("bigint")) { return DbColumnType.LONG; } else if (t.contains("int")) { return DbColumnType.INTEGER; } else if (t.contains("date") || t.contains("time") || t.contains("year")) { return DbColumnType.DATE; } else if (t.contains("text")) { return DbColumnType.STRING; } else if (t.contains("bit")) { return DbColumnType.BOOLEAN; } else if (t.contains("decimal")) { return DbColumnType.BIG_DECIMAL; } else if (t.contains("clob")) { return DbColumnType.STRING; } else if (t.contains("blob")) { return DbColumnType.BLOB; } else if (t.contains("binary")) { return DbColumnType.BYTE_ARRAY; } else if (t.contains("float")) { return DbColumnType.FLOAT; } else if (t.contains("double")) { return DbColumnType.DOUBLE; } else if (t.contains("json") || t.contains("enum")) { return DbColumnType.STRING; } return DbColumnType.STRING; }); dsc.setDriverName("oracle.jdbc.OracleDriver"); dsc.setUsername("wip"); dsc.setPassword("wip"); dsc.setUrl("jdbc:oracle:thin:@10.30.1.89:1521:QAS"); mpg.setDataSource(dsc); // 策略配置 StrategyConfig strategy = new StrategyConfig(); // strategy.setCapitalMode(true);// 全域性大寫命名 ORACLE 注意 strategy.setTablePrefix(tablePerfix);// 此處可以修改為您的表字首 strategy.setNaming(NamingStrategy.underline_to_camel);// 表名生成策略 strategy.setInclude(tableNames); // 需要生成的表 strategy.setVersionFieldName(versionFieldName);//樂觀鎖註解 // strategy.setExclude(new String[]{"test"}); // 排除生成的表 // 自定義實體父類 // strategy.setSuperEntityClass("com.baomidou.demo.TestEntity"); // 自定義實體，公共欄位 // strategy.setSuperEntityColumns(new String[] { "test_id", "age" }); // 自定義 mapper 父類 // strategy.setSuperMapperClass("com.baomidou.demo.TestMapper"); // 自定義 service 父類 // strategy.setSuperServiceClass("com.baomidou.demo.TestService"); // 自定義 service 實現類父類 // strategy.setSuperServiceImplClass("com.baomidou.demo.TestServiceImpl"); // 自定義 controller 父類 // strategy.setSuperControllerClass("com.baomidou.demo.TestController"); // 【實體】是否生成欄位常量（預設 false） // public static final String ID = "test_id"; strategy.setEntityColumnConstant(true); // 【實體】是否為構建者模型（預設 false） // public User setName(String name) {this.name = name; return this;} //strategy.setEntityBuilderModel(true); mpg.setStrategy(strategy); // 包配置 PackageConfig pc = new PackageConfig(); pc.setParent(packageName); pc.setController("controller"); pc.setEntity("model"); mpg.setPackageInfo(pc); // 注入自定義配置，可以在 VM 中使用 cfg.abc 【可無】 ${cfg.abc} InjectionConfig cfg = new InjectionConfig() { @Override public void initMap() { Map<String, Object> map = new HashMap<String, Object>(); map.put("abc", this.getConfig().getGlobalConfig().getAuthor() + "-mp"); map.put("restControllerStyle", true); this.setMap(map); } }; /*// 自定義 xxListIndex.html 生成 List<FileOutConfig> focList = new ArrayList<FileOutConfig>(); focList.add(new FileOutConfig("/templatesMybatis/list.html.vm") { @Override public String outputFile(TableInfo tableInfo) { // 自定義輸入檔名稱 return basePath + "html//" + tableInfo.getEntityName() + "ListIndex.html"; } }); cfg.setFileOutConfigList(focList); mpg.setCfg(cfg); // 自定義 xxAdd.html 生成 focList.add(new FileOutConfig("/templatesMybatis/add.html.vm") { @Override public String outputFile(TableInfo tableInfo) { // 自定義輸入檔名稱 return basePath + "html//" + tableInfo.getEntityName() + "Add.html"; } }); cfg.setFileOutConfigList(focList); mpg.setCfg(cfg); // 自定義 xxUpdate.html生成 focList.add(new FileOutConfig("/templatesMybatis/update.html.vm") { @Override public String outputFile(TableInfo tableInfo) { // 自定義輸入檔名稱 return basePath + "html//" + tableInfo.getEntityName() + "Update.html"; } }); cfg.setFileOutConfigList(focList);*/ mpg.setCfg(cfg); // 關閉預設 xml 生成，調整生成 至 根目錄 /*TemplateConfig tc = new TemplateConfig(); tc.setXml(null); mpg.setTemplate(tc);*/ // 自定義模板配置，可以 copy 原始碼 mybatis-plus/src/main/resources/templates 下面內容修改， // 放置自己專案的 src/main/resources/templates 目錄下, 預設名稱一下可以不配置，也可以自定義模板名稱 TemplateConfig tc = new TemplateConfig(); tc.setController("/mybatis-templates/controller.java.vm"); tc.setService("/mybatis-templates/service.java.vm"); tc.setServiceImpl("/mybatis-templates/serviceImpl.java.vm"); tc.setEntity("/mybatis-templates/entity.java.vm"); tc.setMapper("/mybatis-templates/mapper.java.vm"); //tc.setXml("/templates/mapper.xml.vm"); // 如上任何一個模組如果設定 空 OR Null 將不生成該模組。 mpg.setTemplate(tc); // 執行生成 mpg.execute(); // 列印注入設定【可無】 System.err.println(mpg.getCfg().getMap().get("abc")); } }
示例2：
// 程式碼自動生成器 public class AutoGeneratorTest { 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("小爽帥到拖網速"); gc.setOpen(false); gc.setFileOverride(false); // 是否覆蓋 gc.setServiceName("%Serive"); // 服務介面，去Service的I字首 gc.setIdType(IdType.ID_WORKER); // 主鍵生成策略 gc.setDateType(DateType.ONLY_DATE); gc.setSwagger2(true); // 給程式碼自動生成器注入配置 mpg.setGlobalConfig(gc); // 2、 設定資料來源 DataSourceConfig dsc = new DataSourceConfig(); dsc.setUrl("jdbc:mysql://localhost:3306/mybatis_plus?useSSL=false&useUnicode=true&characterEncoding=utf-8&serverTimezone=GMT%2b8"); dsc.setDriverName("com.mysql.cj.jdbc.Driver"); dsc.setUsername("root"); dsc.setPassword("83821979Zs"); dsc.setDbType(DbType.MYSQL); mpg.setDataSource(dsc); // 3、包的配置 PackageConfig pc = new PackageConfig(); pc.setModuleName("blog"); pc.setParent("com.shuang"); pc.setEntity("entity"); pc.setMapper("mapper"); pc.setService("service"); pc.setController("controller"); mpg.setPackageInfo(pc); // 4、策略配置 StrategyConfig strategy = new StrategyConfig(); strategy.setInclude("user"); // 設定要對映的表名 strategy.setNaming(NamingStrategy.underline_to_camel); // 內建下劃線轉駝峰命名 strategy.setColumnNaming(NamingStrategy.underline_to_camel); strategy.setEntityLombokModel(true); // 自動Lombok strategy.setLogicDeleteFieldName("deleted"); // 邏輯刪除欄位 // 自動填充策略 TableFill gmtCreate = new TableFill("gmt_create", FieldFill.INSERT); TableFill gmtModifid = new TableFill("gmt_modifid", FieldFill.INSERT); ArrayList<TableFill> tableFills = new ArrayList<>(); tableFills.add(gmtCreate); tableFills.add(gmtModifid); strategy.setTableFillList(tableFills); // 樂觀鎖 strategy.setVersionFieldName("version"); strategy.setRestControllerStyle(true); strategy.setControllerMappingHyphenStyle(true); // Localhost:8080/hello_id_2 mpg.setStrategy(strategy); // 執行 mpg.execute(); } }