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
| package com.hqyj.util;
import com.baomidou.mybatisplus.generator.AutoGenerator; import com.baomidou.mybatisplus.generator.config.*; import com.baomidou.mybatisplus.generator.config.rules.DateType; import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy; import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
public class MybatisPlusUtil { public static void main(String[] args) { String userDir = System.getProperty("user.dir"); System.out.println("userDir=" + userDir); AutoGenerator mpg = new AutoGenerator();
GlobalConfig gc = new GlobalConfig(); String projectPath = System.getProperty("user.dir"); System.out.println(projectPath);
gc.setOutputDir(projectPath + "/src/main/java"); gc.setAuthor("rock"); gc.setDateType(DateType.ONLY_DATE); gc.setBaseResultMap(true); gc.setBaseColumnList(true); mpg.setGlobalConfig(gc);
DataSourceConfig dsc = new DataSourceConfig(); dsc.setUrl("jdbc:mysql://localhost:3306/demo01?useUnicode=true&characterEncoding=UTF-8&zeroDateTimeBehavior=convertToNull"); dsc.setDriverName("com.mysql.jdbc.Driver"); dsc.setUsername("root"); dsc.setPassword("1234"); mpg.setDataSource(dsc);
PackageConfig pc = new PackageConfig(); pc.setParent("com.hqyj"); pc.setController("controller"); pc.setEntity("pojo"); pc.setMapper("dao"); pc.setXml("mapper"); mpg.setPackageInfo(pc);
StrategyConfig strategy = new StrategyConfig(); strategy.setNaming(NamingStrategy.underline_to_camel); strategy.setColumnNaming(NamingStrategy.underline_to_camel); strategy.setEntityLombokModel(true); strategy.setRestControllerStyle(true); strategy.setInclude("user_info"); strategy.setControllerMappingHyphenStyle(true); strategy.setTablePrefix(pc.getModuleName() + "_"); mpg.setStrategy(strategy);
TemplateConfig tc = new TemplateConfig(); tc.setXml("/templates/mapper.xml"); mpg.setTemplate(tc);
mpg.execute(); }
}
|