mybaity 程式碼自動生成器
阿新 • • 發佈：2021-08-20
package org.jeecg.util; import com.baomidou.mybatisplus.core.exceptions.MybatisPlusException; 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.po.TableInfo; import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy; import com.baomidou.mybatisplus.generator.engine.VelocityTemplateEngine; import org.apache.commons.lang3.StringUtils; import java.io.File; import java.util.ArrayList;import java.util.List; import java.util.Scanner; import java.util.regex.Matcher; import java.util.regex.Pattern; /** * @Author: Tring * @Date: 2021-08-05 * @ClassName: CodeGenerator * @Description: 程式碼生成器 */ public class CodeGenerator { public static String scanner(String tip) { Scanner scanner= new Scanner(System.in); StringBuilder help = new StringBuilder(); help.append("請輸入" + tip + "："); System.out.println(help.toString()); if (scanner.hasNext()) { String ipt = scanner.next(); if (StringUtils.isNotEmpty(ipt)) { return ipt; } } throw new MybatisPlusException("請輸入正確的" + tip + "！"); } public static void main(String[] args) { // 程式碼生成器 AutoGenerator mpg = new AutoGenerator(); // 全域性配置 GlobalConfig gc = new GlobalConfig(); String projectPath = System.getProperty("user.dir"); System.out.println(projectPath); String path = "/src/main/java"; gc.setOutputDir(projectPath + path); gc.setAuthor("Tring"); gc.setOpen(false); //實體屬性 Swagger2 註解 // gc.setSwagger2(true); mpg.setGlobalConfig(gc); // 資料來源配置 DataSourceConfig dsc = new DataSourceConfig(); // jdbc:mysql://127.0.0.1:3306/air dsc.setUrl("jdbc:mysql://183.61.0.54:3336/purtato?characterEncoding=UTF-8&useUnicode=true&useSSL=false&tinyInt1isBit=false&allowPublicKeyRetrieval=true&serverTimezone=Asia/Shanghai"); // dsc.setSchemaName("public"); dsc.setDriverName("com.mysql.cj.jdbc.Driver"); dsc.setUsername("purtato"); dsc.setPassword("purtato"); mpg.setDataSource(dsc); // 包配置 PackageConfig pc = new PackageConfig(); // pc.setModuleName(scanner("模組名")); String parent = "org.jeecg.modules.purtato"; pc.setParent(parent); 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/java/org/jeecg/modules/purtato/mapper/xml/" + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML; } }); /* cfg.setFileCreate(new IFileCreate() { @Override public boolean isCreate(ConfigBuilder configBuilder, FileType fileType, String filePath) { // 判斷自定義資料夾是否需要建立 checkDir("呼叫預設方法建立的目錄"); return false; } }); */ cfg.setFileOutConfigList(focList); mpg.setCfg(cfg); // 配置模板 TemplateConfig templateConfig = new TemplateConfig(); // 配置自定義輸出模板 //指定自定義模板路徑，注意不要帶上.ftl/.vm, 會根據使用的模板引擎自動識別 // templateConfig.setEntity("templates/entity2.java"); // templateConfig.setService(); // templateConfig.setController(); templateConfig.setXml(null); templateConfig.setController(""); mpg.setTemplate(templateConfig); // 策略配置 StrategyConfig strategy = new StrategyConfig(); strategy.setNaming(NamingStrategy.underline_to_camel); strategy.setColumnNaming(NamingStrategy.underline_to_camel); // strategy.setSuperEntityClass("com.syni.pacp.userservice.common.BaseEntity"); strategy.setEntityLombokModel(true); strategy.setRestControllerStyle(true); // 公共父類 // strategy.setSuperControllerClass("com.syni.pacp.userservice.common.BaseController"); // 寫於父類中的公共欄位 // strategy.setSuperEntityColumns("id"); String[] split = scanner("表名，多個英文逗號分割").split(","); strategy.setInclude(split); for (String s:split ) { String filePath = projectPath + path + File.separator + parent.replaceAll("\\.", Matcher.quoteReplacement(File.separator)) + File.separator + "entity" + File.separator + lineToHump(s) + ".java"; filePath = filePath.replaceAll("/", Matcher.quoteReplacement(File.separator)); File file = new File(filePath); if (file.exists()){ file.delete(); } } strategy.setControllerMappingHyphenStyle(true); strategy.setTablePrefix(pc.getModuleName() + "_"); mpg.setStrategy(strategy); mpg.setTemplateEngine(new VelocityTemplateEngine()); mpg.execute(); } private static Pattern linePattern = Pattern.compile("_(\\w)"); private static String lineToHump(String str) { str = str.toLowerCase(); Matcher matcher = linePattern.matcher(str); StringBuffer sb = new StringBuffer(); while (matcher.find()) { matcher.appendReplacement(sb, matcher.group(1).toUpperCase()); } matcher.appendTail(sb); return sb.substring(0,1).toUpperCase() + sb.substring(1); } }
maven匯入相關jar包
<properties>
<velocity.version>1.7</velocity.version>
<mybatis-plus-generator.version>3.3.1</mybatis-plus-generator.version>
</properties>
<!-- mybatis-plus 程式碼生成 -->
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-generator</artifactId>
<version>${mybatis-plus-generator.version}</version>
</dependency>
<!-- vm模版 -->
<dependency>
<groupId>org.apache.velocity</groupId>
<artifactId>velocity</artifactId>
<version>${velocity.version}</version>
</dependency>
直接執行那個檔案就ok了，會自動幫你生成entity、service、serviceImpl、mapper.java、mapper.xml、controller