程式碼生成器 從entity到service mapper
阿新 • • 發佈：2021-01-30
技術標籤：java
/** * All rights Reserved, Designed By www.techsun.com.cn * @Title: CodeGenerator.java * @Package: com.techsun.platform.config.generator * @Description: (程式碼自動生成) * @author: sdj * @date: 2019年8月21日 下午2:48:54 * @version: V1.0 * @Copyright: 2019 www.techsun.com.cn Inc. All rights reserved. */ package com.techsun.client.config.generator; import java.util.ArrayList; import java.util.List; 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.DataSourceConfig; import com.baomidou.mybatisplus.generator.config.FileOutConfig; import com.baomidou.mybatisplus.generator.config.GlobalConfig; import com.baomidou.mybatisplus.generator.config.IFileCreate; import com.baomidou.mybatisplus.generator.config.PackageConfig; import com.baomidou.mybatisplus.generator.config.StrategyConfig; import com.baomidou.mybatisplus.generator.config.TemplateConfig; import com.baomidou.mybatisplus.generator.config.builder.ConfigBuilder; import com.baomidou.mybatisplus.generator.config.converts.MySqlTypeConvert; import com.baomidou.mybatisplus.generator.config.po.TableInfo; import com.baomidou.mybatisplus.generator.config.rules.DateType; import com.baomidou.mybatisplus.generator.config.rules.DbColumnType; import com.baomidou.mybatisplus.generator.config.rules.FileType; import com.baomidou.mybatisplus.generator.config.rules.IColumnType; import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy; import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine; /** * @ClassName: CodeGenerator * @Description: (程式碼自動生成) * @author: sdj * @date: 2019年8月21日 下午2:48:54 * @Copyright: 2019 www.techsun.com.cn Inc. All rights reserved. */ public class CodeGenerator { public static void main(String[] args) { // 程式碼生成器 AutoGenerator mpg = new AutoGenerator(); // 全域性配置 final GlobalConfig gc = new GlobalConfig(); final String projectPath = System.getProperty("user.dir"); gc.setFileOverride(true); gc.setIdType(IdType.AUTO); // 主鍵策略 gc.setOutputDir(projectPath + "/src/main/java"); gc.setAuthor("techsun"); gc.setOpen(false); gc.setDateType(DateType.ONLY_DATE); gc.setSwagger2(false); // 實體屬性 Swagger2 註解 mpg.setGlobalConfig(gc); // 資料來源配置 DataSourceConfig dsc = new DataSourceConfig(); dsc.setDbType(DbType.MYSQL); // dsc.setUrl("jdbc:mysql://rm-uf647ei49csnujzg7ho.mysql.rds.aliyuncs.com:9906/jinshatrace"); // dsc.setDriverName("com.mysql.jdbc.Driver"); // dsc.setUsername("jinshatrace"); // dsc.setPassword("@jinshatrace2019"); dsc.setUrl("jdbc:mysql://rm-uf647ei49csnujzg7ho.mysql.rds.aliyuncs.com:9906/jinshatrace"); dsc.setDriverName("com.mysql.jdbc.Driver"); dsc.setUsername("jinshatrace"); dsc.setPassword("@jinshatrace2019"); dsc.setTypeConvert(new MySqlTypeConvert() { // 自定義資料庫表字段型別轉換【可選】 @SuppressWarnings("unused") public IColumnType processTypeConvert(String fieldType) { // 將資料庫中datetime轉換成date if (fieldType.toLowerCase().contains("datetime")) { return DbColumnType.TIMESTAMP; } return (DbColumnType) super.processTypeConvert(gc, fieldType); } }); mpg.setDataSource(dsc); // 包配置 final PackageConfig pc = new PackageConfig(); pc.setModuleName("client"); pc.setParent("com.techsun"); mpg.setPackageInfo(pc); // 自定義配置 InjectionConfig cfg = new InjectionConfig() { @Override public void initMap() { // to do nothing } }; // 如果模板引擎是 freemarker String templatePath = "/templates/mapper.xml.ftl"; // 如果模板引擎是 velocity // String templatePath = "/templates/mapper.xml.vm"; // 自定義輸出配置 List<FileOutConfig> focList = new ArrayList<>(); // // 自定義配置會被優先輸出 focList.add(new FileOutConfig(templatePath) { @Override public String outputFile(TableInfo tableInfo) { // 自定義輸出檔名 ， 如果你 Entity 設定了前後綴、此處注意 xml 的名稱會跟著發生變化！！ return projectPath + "/src/main/resources/mapper/" + pc.getModuleName() + "/" + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML; } }); cfg.setFileCreate(new IFileCreate() { @Override public boolean isCreate(ConfigBuilder configBuilder, FileType fileType, String filePath) { return true; } }); cfg.setFileOutConfigList(focList); mpg.setCfg(cfg); // 配置模板templateConfig TemplateConfig templateConfig = new TemplateConfig(); // 配置自定義輸出模板 templateConfig.setEntity("/templates/entityself.java"); templateConfig.setMapper("/templates/mapperself.java"); templateConfig.setService("/templates/serviceself.java"); templateConfig.setServiceImpl("/templates/serviceImplself.java"); templateConfig.setXml(null); mpg.setTemplate(templateConfig); // 策略配置 StrategyConfig strategy = new StrategyConfig(); strategy.setNaming(NamingStrategy.underline_to_camel); strategy.setColumnNaming(NamingStrategy.underline_to_camel); //繼承 //strategy.setSuperEntityClass("com.techsun.client.entity.BaseEntity"); strategy.setEntityLombokModel(true); strategy.setRestControllerStyle(false); // 公共父類 // strategy.setSuperControllerClass("com.baomidou.ant.common.BaseController"); // 寫於父類中的公共欄位 // strategy.setSuperEntityColumns(new String[] { "updated_by", "updated_time", "created_by", "created_time" }); strategy.setInclude( ( // "warehouse_out_box_data,warehouse_out_his_bottle,warehouse_out_his_box,warehouse_out_sub_bill,warehouse_out_temp_bill,warehouse_out_temp_bottle,warehouse_out_temp_box,suite_box_info,suite_info,out_unusual_bottle,out_unusual_box,old_bottle_info,old_box_info,cloud_film_info,cloud_film_box_info,cloud_film_label_trace_data,old_cloud_film_label_trace_data,suite_label_trace_data" "no_trace_info_tag" ).split(",")); strategy.setControllerMappingHyphenStyle(false); strategy.setTablePrefix(pc.getModuleName() + "_"); mpg.setStrategy(strategy); mpg.setTemplateEngine(new FreemarkerTemplateEngine()); mpg.execute(); } }