mybatis-plus 自动代码生成

 

 

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123

package com.lookcoder.perpayadmin.util;   import com.baomidou.mybatisplus.core.exceptions.MybatisPlusException; import com.baomidou.mybatisplus.core.toolkit.StringPool; import com.baomidou.mybatisplus.core.toolkit.StringUtils; 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.FreemarkerTemplateEngine;   import java.util.ArrayList; import java.util.List; import java.util.Scanner;   public class MPGeneratorUtils {       /**      * <p>      * 读取控制台内容      * </p>      */     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");         gc.setOutputDir(projectPath + "/src/main/java");         gc.setAuthor("lookcoder");         gc.setOpen(false);         // gc.setSwagger2(true); 实体属性 Swagger2 注解         mpg.setGlobalConfig(gc);           // 数据源配置         DataSourceConfig dsc = new DataSourceConfig();         dsc.setUrl("jdbc:mysql://localhost:3306/perpay?useUnicode=true&useSSL=false&characterEncoding=utf8");         // dsc.setSchemaName("public");         dsc.setDriverName("com.mysql.jdbc.Driver");         dsc.setUsername("root");         dsc.setPassword("root");         mpg.setDataSource(dsc);           // 包配置         String parent = "com.lookcoder.perpayadmin";         PackageConfig pc = new PackageConfig(); //        pc.setModuleName(scanner("模块名")); //        pc.setParent("com.lookcoder.perpayadmin.module");         pc.setParent(parent);         mpg.setPackageInfo(pc);           // 如果模板引擎是 freemarker //        String templatePath = "/templates/mapper.xml.ftl";         // 如果模板引擎是 velocity          String templatePath = "/templates/mapper.xml.vm";           // 自定义配置         InjectionConfig cfg = new InjectionConfig() {             @Override             public void initMap() {                 // to do nothing             }         };         // 自定义输出配置         List<FileOutConfig> focList = new ArrayList<>();         // 自定义配置会被优先输出         focList.add(new FileOutConfig(templatePath) {             @Override             public String outputFile(TableInfo tableInfo) {                 // 自定义输出文件名 ， 如果你 Entity 设置了前后缀、此处注意 xml 的名称会跟着发生变化！！                 String customFileName = projectPath + "/src/main/java/" + parent.replaceAll("\\.", "/") + "/mapper/"                         + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;                 System.out.println(customFileName);                 return customFileName;             }         });         cfg.setFileOutConfigList(focList);         mpg.setCfg(cfg);           // 配置模板         TemplateConfig templateConfig = new TemplateConfig();           // 配置自定义输出模板         //指定自定义模板路径，注意不要带上.ftl/.vm, 会根据使用的模板引擎自动识别         // templateConfig.setEntity("templates/entity2.java");         // templateConfig.setService();         // templateConfig.setController();           templateConfig.setXml(null);         mpg.setTemplate(templateConfig);           // 策略配置         StrategyConfig strategy = new StrategyConfig();         strategy.setNaming(NamingStrategy.underline_to_camel);         strategy.setColumnNaming(NamingStrategy.underline_to_camel);         strategy.setEntityLombokModel(true);         strategy.setRestControllerStyle(true);         // 写于父类中的公共字段         strategy.setSuperEntityColumns("id");         strategy.setInclude(scanner("表名，多个英文逗号分割").split(","));         strategy.setControllerMappingHyphenStyle(true);         strategy.setTablePrefix(pc.getModuleName() + "_");         strategy.setEntityLombokModel(false);         mpg.setStrategy(strategy);         mpg.execute();     } }