好用的代码生成器工具类分享.
代码粘贴过去直接用, 方便, 不看就懂
直接上完整代码
/**
* freemarker模板引擎
* @author Huzz
*/
public class MybatisPlusCodeGenerator {
/** -------------------------------------------CONFIG------------------------------------------------------- */
private static final String PROJECT_PATH = System.getProperty("user.dir");
private static final String AUTHOR = "Huzz";
private static final String PACKAGE = "com.huzz";
private static final String DATABASE_URL = "jdbc:mysql://localhost:3306/db_2021?serverTimezone=Asia/Shanghai";
private static final String DATABASE_DRIVER = "com.mysql.cj.jdbc.Driver";
private static final String DATABASE_USERNAME = "root";
private static final String DATABASE_PASSWORD = "123456!";
/** ---------------------------------------------MAIN------------------------------------------------------- */
public static void main(String[] args) {
// 代码生成器
AutoGenerator autoGenerator = new AutoGenerator();
// 1. 全局配置
globalConfig(autoGenerator);
// 2. 包配置
PackageConfig pc = packageConfig(autoGenerator);
// 3. 数据库配置
dataSourceConfig(autoGenerator);
// 4. 自定义配置
injectionConfig(autoGenerator, pc);
// 5. 模板配置
templateConfig(autoGenerator);
// 6. 策略配置
strategyConfig(autoGenerator, pc);
// 7. 设置模板引擎
autoGenerator.setTemplateEngine(new FreemarkerTemplateEngine());
// 8. 执行
autoGenerator.execute();
}
/** -------------------------------------------METHOD------------------------------------------------------- */
/**
* 读取控制台内容
*/
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.isNotBlank(ipt)) {
return ipt;
}
}
throw new MybatisPlusException("请输入正确的" + tip + "!");
}
/**
* 全局配置
* @param autoGenerator 代码生成器
*/
private static void globalConfig(AutoGenerator autoGenerator) {
GlobalConfig gc = new GlobalConfig();
gc.setOutputDir(PROJECT_PATH + "/src/main/java");
gc.setAuthor(AUTHOR);
//重新生成时文件是否覆盖
gc.setFileOverride(false);
//生成后是否打开资源管理器
gc.setOpen(false);
// 定义生成的实体类中日期类型
gc.setDateType(DateType.ONLY_DATE);
// 实体属性 Swagger2 注解
gc.setSwagger2(true);
autoGenerator.setGlobalConfig(gc);
}
/**
* 包配置
* @param autoGenerator
* @return
*/
private static PackageConfig packageConfig(AutoGenerator autoGenerator) {
PackageConfig pc = new PackageConfig();
pc.setModuleName(scanner("设置系统模块名"));
pc.setParent(PACKAGE);
pc.setController("controller");
pc.setEntity("entity");
pc.setService("service");
pc.setMapper("mapper");
autoGenerator.setPackageInfo(pc);
return pc;
}
/**
* 数据库配置
* @param autoGenerator
*/
private static void dataSourceConfig(AutoGenerator autoGenerator) {
// 数据源配置
DataSourceConfig dsc = new DataSourceConfig();
dsc.setUrl(DATABASE_URL);
// dsc.setSchemaName("public");
dsc.setDriverName(DATABASE_DRIVER);
dsc.setUsername(DATABASE_USERNAME);
dsc.setPassword(DATABASE_PASSWORD);
// 数据库类型
dsc.setDbType(DbType.MYSQL);
autoGenerator.setDataSource(dsc);
}
/**
* 自定义配置
*/
private static void injectionConfig(AutoGenerator autoGenerator, PackageConfig 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 PROJECT_PATH + "/src/main/resources/mapper/" + pc.getModuleName()
+ "/" + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;
}
});
cfg.setFileOutConfigList(focList);
autoGenerator.setCfg(cfg);
}
/**
* 模板配置
* @param autoGenerator
*/
private static void templateConfig(AutoGenerator autoGenerator) {
TemplateConfig templateConfig = new TemplateConfig();
// 配置自定义输出模板
//指定自定义模板路径,注意不要带上.ftl/.vm, 会根据使用的模板引擎自动识别
// templateConfig.setEntity("templates/entity2.java");
// templateConfig.setService();
// templateConfig.setController();
templateConfig.setXml(null);
autoGenerator.setTemplate(templateConfig);
}
/**
* 策略配置
* @param autoGenerator
* @param pc
*/
private static void strategyConfig(AutoGenerator autoGenerator, PackageConfig pc) {
StrategyConfig strategy = new StrategyConfig();
// 数据库表映射到实体的命名策略
strategy.setNaming(NamingStrategy.underline_to_camel);
// 数据库表字段映射到实体的命名策略
strategy.setColumnNaming(NamingStrategy.underline_to_camel);
// strategy.setSuperEntityClass("你自己的父类实体,没有就不用设置!");
// lombok 模型 @Accessors(chain = true) setter链式操作
strategy.setEntityLombokModel(true);
// restful api风格控制器
strategy.setRestControllerStyle(true);
// 公共父类
// strategy.setSuperControllerClass("你自己的父类控制器,没有就不用设置!");
// 写于父类中的公共字段
strategy.setSuperEntityColumns("id");
// 对那些张表生成代码
strategy.setInclude(scanner("表名,多个英文逗号分割").split(","));
//url中驼峰转连字符
strategy.setControllerMappingHyphenStyle(true);
//生成实体时去掉表前缀
strategy.setTablePrefix(pc.getModuleName() + "_");
autoGenerator.setStrategy(strategy);
}
}
修改一下下面的配置就行,其他的不用管,也不用看,就是为了这样方便.
/** -------------------------------------------CONFIG------------------------------------------------------- */
private static final String PROJECT_PATH = System.getProperty("user.dir");
private static final String AUTHOR = "Huzz";
private static final String PACKAGE = "com.huzz";
private static final String DATABASE_URL = "jdbc:mysql://localhost:3306/db_2021?serverTimezone=Asia/Shanghai";
private static final String DATABASE_DRIVER = "com.mysql.cj.jdbc.Driver";
private static final String DATABASE_USERNAME = "root";
private static final String DATABASE_PASSWORD = "123456";
配置完成,启动main(),开始生成:
效果图:
开个玩笑😀, 下面才是啦