package com.ruoyi.system.controller;
|
|
import com.baomidou.mybatisplus.generator.AutoGenerator;
|
import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
|
import com.baomidou.mybatisplus.generator.config.GlobalConfig;
|
import com.baomidou.mybatisplus.generator.config.PackageConfig;
|
import com.baomidou.mybatisplus.generator.config.StrategyConfig;
|
import com.baomidou.mybatisplus.generator.config.converts.MySqlTypeConvert;
|
import com.baomidou.mybatisplus.generator.config.rules.DbType;
|
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
|
|
public class GeneratorMybatisPlus {
|
|
private static final boolean RUN_GENERATOR = true;
|
|
public static void main(String[] args) {
|
if(!RUN_GENERATOR){
|
return;
|
}
|
// 代码生成器
|
AutoGenerator autoGenerator = new AutoGenerator();
|
// 全局配置
|
GlobalConfig globalConfig = new GlobalConfig();
|
//生成文件的输出目录
|
String path="D:/hrt";
|
globalConfig.setOutputDir(path);
|
// Author设置作者
|
globalConfig.setAuthor("mybatis-plus");
|
// 文件覆盖
|
globalConfig.setFileOverride(true);
|
// 生成后打开文件
|
globalConfig.setOpen(false);
|
// 自定义文件名风格，%s自动填充表实体属性
|
globalConfig.setMapperName("%sMapper");
|
globalConfig.setXmlName("%sMapper");
|
globalConfig.setServiceName("%sService");
|
globalConfig.setServiceImplName("%sServiceImpl");
|
globalConfig.setAuthor("jqs");
|
// globalConfig.setEntityName("%s");
|
globalConfig.setControllerName("%sController");
|
autoGenerator.setGlobalConfig(globalConfig);
|
|
// 数据源配置
|
DataSourceConfig dataSourceConfig = new DataSourceConfig();
|
dataSourceConfig.setDbType(DbType.MYSQL);
|
dataSourceConfig.setTypeConvert(new MySqlTypeConvert());
|
dataSourceConfig.setUrl("jdbc:mysql://47.109.78.184:10010/hrt_sys?tinyInt1isBit=false");
|
dataSourceConfig.setDriverName("com.mysql.cj.jdbc.Driver");
|
dataSourceConfig.setUsername("root");
|
dataSourceConfig.setPassword("hrt123456");
|
autoGenerator.setDataSource(dataSourceConfig);
|
|
// 包名配置
|
PackageConfig packageConfig = new PackageConfig();
|
// 父包和子包名分开处理
|
packageConfig.setParent("com.ruoyi.system");
|
packageConfig.setController("app");
|
packageConfig.setEntity("domain.pojo");
|
packageConfig.setMapper("mapper");
|
packageConfig.setService("service");
|
packageConfig.setServiceImpl("service.impl");
|
autoGenerator.setPackageInfo(packageConfig);
|
|
// 生成策略配置
|
StrategyConfig strategy = new StrategyConfig();
|
//设置命名格式
|
strategy.setNaming(NamingStrategy.underline_to_camel);
|
strategy.setColumnNaming(NamingStrategy.underline_to_camel);
|
// 实体是否为lombok模型，默认 false
|
strategy.setEntityLombokModel(true);
|
//生成 @RestController 控制器
|
strategy.setRestControllerStyle(true);
|
// 驼峰转连字符
|
strategy.setControllerMappingHyphenStyle(true);
|
//表和前缀处理
|
String[] table = {
|
"t_file_record"
|
};
|
strategy.setInclude(table);
|
String[] tablePre = new String[]{"t_"};
|
strategy.setTablePrefix(tablePre);
|
autoGenerator.setStrategy(strategy);
|
// 执行，以上相关参数可以基于动态输入获取
|
autoGenerator.execute();
|
}
|
}
|
