232 lines
8.2 KiB
Java
232 lines
8.2 KiB
Java
|
package com.aircraft.generate;
|
|||
|
|
|
|||
|
|
|
|||
|
|
import com.baomidou.mybatisplus.annotation.DbType;
|
|||
|
|
import com.baomidou.mybatisplus.annotation.FieldFill;
|
|||
|
|
import com.baomidou.mybatisplus.generator.AutoGenerator;
|
|||
|
|
import com.baomidou.mybatisplus.generator.InjectionConfig;
|
|||
|
|
import com.baomidou.mybatisplus.generator.config.*;
|
|||
|
|
import com.baomidou.mybatisplus.generator.config.converts.MySqlTypeConvert;
|
|||
|
|
import com.baomidou.mybatisplus.generator.config.po.TableFill;
|
|||
|
|
import com.baomidou.mybatisplus.generator.config.po.TableInfo;
|
|||
|
|
import com.baomidou.mybatisplus.generator.config.rules.DbColumnType;
|
|||
|
|
import com.baomidou.mybatisplus.generator.config.rules.IColumnType;
|
|||
|
|
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
import java.io.File;
|
|||
|
|
import java.util.*;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* <p>
|
|||
|
|
* 代码生成器父类
|
|||
|
|
* </p>
|
|||
|
|
*
|
|||
|
|
*/
|
|||
|
|
public class SuperGenerator {
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 获取TemplateConfig
|
|||
|
|
*
|
|||
|
|
* @return
|
|||
|
|
*/
|
|||
|
|
protected TemplateConfig getTemplateConfig() {
|
|||
|
|
return new TemplateConfig().setXml(null);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 获取InjectionConfig
|
|||
|
|
*
|
|||
|
|
* @return
|
|||
|
|
*/
|
|||
|
|
protected InjectionConfig getInjectionConfig() {
|
|||
|
|
return new InjectionConfig() {
|
|||
|
|
@Override
|
|||
|
|
public void initMap() {
|
|||
|
|
Map<String, Object> map = new HashMap<>();
|
|||
|
|
this.setMap(map);
|
|||
|
|
}
|
|||
|
|
}.setFileOutConfigList(Collections.<FileOutConfig>singletonList(new FileOutConfig(
|
|||
|
|
"/templates/mapper.xml.vm") {
|
|||
|
|
// 自定义输出文件目录
|
|||
|
|
@Override
|
|||
|
|
public String outputFile(TableInfo tableInfo) {
|
|||
|
|
return getResourcePath() + "/mapper/" + tableInfo.getEntityName() + "Mapper.xml";
|
|||
|
|
}
|
|||
|
|
}));
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 获取PackageConfig
|
|||
|
|
*
|
|||
|
|
* @return
|
|||
|
|
*/
|
|||
|
|
protected PackageConfig getPackageConfig() {
|
|||
|
|
return new PackageConfig()
|
|||
|
|
.setParent("com.aircraft.modules.system")
|
|||
|
|
.setController("controller")
|
|||
|
|
.setEntity("domain")
|
|||
|
|
.setMapper("mapper")
|
|||
|
|
.setService("service")
|
|||
|
|
.setServiceImpl("service.impl");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 获取StrategyConfig
|
|||
|
|
*
|
|||
|
|
* @param tableName
|
|||
|
|
* @return
|
|||
|
|
*/
|
|||
|
|
protected StrategyConfig getStrategyConfig(String tableName) {
|
|||
|
|
return new StrategyConfig()
|
|||
|
|
.setCapitalMode(false)// 全局大写命名
|
|||
|
|
// .setTablePrefix("sys_")// 去除前缀
|
|||
|
|
.setNaming(NamingStrategy.underline_to_camel)// 表名生成策略
|
|||
|
|
//.setInclude(new String[] { "user" }) // 需要生成的表
|
|||
|
|
//自定义实体父类
|
|||
|
|
.setSuperEntityClass("com.aircraft.base.BaseEntity")
|
|||
|
|
// 自定义实体，公共字段
|
|||
|
|
.setSuperEntityColumns("create_by")
|
|||
|
|
.setSuperEntityColumns("create_time")
|
|||
|
|
.setSuperEntityColumns("update_by")
|
|||
|
|
.setSuperEntityColumns("update_time")
|
|||
|
|
.setSuperEntityColumns("del_flag")
|
|||
|
|
|
|||
|
|
// 自定义 mapper 父类
|
|||
|
|
.setSuperMapperClass("com.baomidou.mybatisplus.core.mapper.BaseMapper")
|
|||
|
|
// 自定义 service 实现类父类
|
|||
|
|
.setSuperServiceImplClass("com.baomidou.mybatisplus.extension.service.impl.ServiceImpl")
|
|||
|
|
// 自定义 service 接口父类
|
|||
|
|
.setSuperServiceClass("com.baomidou.mybatisplus.extension.service.IService")
|
|||
|
|
// 【实体】是否生成字段常量（默认 false）
|
|||
|
|
// .setEntityColumnConstant(true)
|
|||
|
|
// 【实体】是否为构建者模型（默认 false）
|
|||
|
|
.setEntityBuilderModel(false)
|
|||
|
|
// 【实体】是否为lombok模型（默认 false）<a href="https://projectlombok.org/">document</a>
|
|||
|
|
.setEntityLombokModel(true)
|
|||
|
|
// Boolean类型字段是否移除is前缀处理
|
|||
|
|
.setEntityBooleanColumnRemoveIsPrefix(true)
|
|||
|
|
.setRestControllerStyle(false)
|
|||
|
|
.setRestControllerStyle(true)
|
|||
|
|
.setInclude(tableName);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 获取DataSourceConfig
|
|||
|
|
*
|
|||
|
|
* @return
|
|||
|
|
*/
|
|||
|
|
protected DataSourceConfig getDataSourceConfig() {
|
|||
|
|
return new DataSourceConfig()
|
|||
|
|
.setDbType(DbType.MYSQL)// 数据库类型
|
|||
|
|
.setTypeConvert(new MySqlTypeConvert() {
|
|||
|
|
@Override
|
|||
|
|
public IColumnType processTypeConvert(GlobalConfig globalConfig, String fieldType) {
|
|||
|
|
if (fieldType.toLowerCase().equals("bit")) {
|
|||
|
|
return DbColumnType.BOOLEAN;
|
|||
|
|
}
|
|||
|
|
if (fieldType.toLowerCase().equals("tinyint")) {
|
|||
|
|
return DbColumnType.BOOLEAN;
|
|||
|
|
}
|
|||
|
|
if (fieldType.toLowerCase().equals("date")) {
|
|||
|
|
return DbColumnType.LOCAL_DATE;
|
|||
|
|
}
|
|||
|
|
if (fieldType.toLowerCase().equals("time")) {
|
|||
|
|
return DbColumnType.LOCAL_TIME;
|
|||
|
|
}
|
|||
|
|
if (fieldType.toLowerCase().equals("datetime")) {
|
|||
|
|
return DbColumnType.DATE;
|
|||
|
|
}
|
|||
|
|
return super.processTypeConvert(globalConfig, fieldType);
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
.setDriverName("com.mysql.cj.jdbc.Driver")
|
|||
|
|
.setUsername("root")
|
|||
|
|
.setPassword("scutensave")
|
|||
|
|
.setUrl("jdbc:mysql://8.138.171.103:3306/aircraft-mp?useUnicode=true&characterEncoding=utf8&autoReconnect=true&serverTimezone=GMT%2B8&useSSL=false");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 获取GlobalConfig
|
|||
|
|
*
|
|||
|
|
* @return
|
|||
|
|
*/
|
|||
|
|
protected GlobalConfig getGlobalConfig() {
|
|||
|
|
return new GlobalConfig()
|
|||
|
|
.setOutputDir(getJavaPath())//输出目录
|
|||
|
|
.setFileOverride(false)// 是否覆盖文件
|
|||
|
|
.setActiveRecord(false)// 开启 activeRecord 模式
|
|||
|
|
.setEnableCache(false)// XML 二级缓存
|
|||
|
|
.setBaseResultMap(false)// XML ResultMap
|
|||
|
|
.setBaseColumnList(false)// XML columList
|
|||
|
|
.setKotlin(false) //是否生成 kotlin 代码
|
|||
|
|
.setOpen(false)
|
|||
|
|
.setAuthor("cli") //作者
|
|||
|
|
//自定义文件命名，注意 %s 会自动填充表实体属性！
|
|||
|
|
.setEntityName("%s")
|
|||
|
|
.setMapperName("%sMapper")
|
|||
|
|
.setXmlName("%sMapper")
|
|||
|
|
.setServiceName("%sService")
|
|||
|
|
.setServiceImplName("%sServiceImpl")
|
|||
|
|
.setControllerName("%sController");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 获取根目录
|
|||
|
|
*
|
|||
|
|
* @return
|
|||
|
|
*/
|
|||
|
|
private String getRootPath() {
|
|||
|
|
String file = Objects.requireNonNull(SuperGenerator.class.getClassLoader().getResource("")).getFile();
|
|||
|
|
return new File(file).getParentFile().getParent();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 获取JAVA目录
|
|||
|
|
*
|
|||
|
|
* @return
|
|||
|
|
*/
|
|||
|
|
protected String getJavaPath() {
|
|||
|
|
String javaPath = getRootPath() + "/src/main/java";
|
|||
|
|
System.err.println(" Generator Java Path:【 " + javaPath + " 】");
|
|||
|
|
return javaPath;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 获取Resource目录
|
|||
|
|
*
|
|||
|
|
* @return
|
|||
|
|
*/
|
|||
|
|
protected String getResourcePath() {
|
|||
|
|
String resourcePath = getRootPath() + "/src/main/resources";
|
|||
|
|
System.err.println(" Generator Resource Path:【 " + resourcePath + " 】");
|
|||
|
|
return resourcePath;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 获取AutoGenerator
|
|||
|
|
*
|
|||
|
|
* @param tableName
|
|||
|
|
* @return
|
|||
|
|
*/
|
|||
|
|
protected AutoGenerator getAutoGenerator(String tableName) {
|
|||
|
|
return new AutoGenerator()
|
|||
|
|
// 全局配置
|
|||
|
|
.setGlobalConfig(getGlobalConfig())
|
|||
|
|
// 数据源配置
|
|||
|
|
.setDataSource(getDataSourceConfig())
|
|||
|
|
// 策略配置
|
|||
|
|
.setStrategy(getStrategyConfig(tableName))
|
|||
|
|
// 包配置
|
|||
|
|
.setPackageInfo(getPackageConfig())
|
|||
|
|
// 注入自定义配置，可以在 VM 中使用 cfg.abc 设置的值
|
|||
|
|
.setCfg(getInjectionConfig())
|
|||
|
|
.setTemplate(getTemplateConfig());
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|