package com.monkeylessey.gen.utils;
|
|
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
|
import com.baomidou.mybatisplus.generator.FastAutoGenerator;
|
import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
|
import com.baomidou.mybatisplus.generator.config.OutputFile;
|
import com.baomidou.mybatisplus.generator.config.TemplateConfig;
|
import com.monkeylessey.sys.domain.base.AbsEntity;
|
import com.monkeylessey.gen.domain.GenerateData;
|
import com.monkeylessey.gen.enums.GenFileTypeEnum;
|
import com.monkeylessey.gen.engine.MyVelocityTemplateEngine;
|
import com.monkeylessey.sys.domain.vo.TableColumnVO;
|
import com.monkeylessey.sys.mapper.SysMenuMapper;
|
import org.apache.commons.lang3.StringUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
import org.springframework.util.CollectionUtils;
|
|
import javax.sql.DataSource;
|
import java.io.File;
|
import java.util.Collections;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.function.Consumer;
|
import java.util.stream.Collectors;
|
|
/**
|
* 该类用于前端传递参数进行代码生成
|
*
|
* @author 29443
|
* @date 2022/4/7
|
*/
|
@Service
|
public class GenerateCodeUtil {
|
|
@Autowired
|
private DataSource dataSource;
|
@Autowired
|
private SysMenuMapper sysMenuMapper;
|
|
/**
|
* 代码生成方法
|
*
|
* @param data 所需数据对象
|
*/
|
public void generateCode(GenerateData data) {
|
String outDir = "";
|
String mapperDir = "";
|
if (StringUtils.isNotBlank(data.getWhichModule())) {
|
// 多模块应用
|
outDir = System.getProperty("user.dir") + "\\" + data.getWhichModule() + "\\src\\main\\java";
|
mapperDir = System.getProperty("user.dir") + "\\" + data.getWhichModule() + "\\src\\main\\resources\\mapper";
|
} else {
|
// 单模块应用
|
outDir = System.getProperty("user.dir") + "\\src\\main\\java";
|
mapperDir = System.getProperty("user.dir") + "\\src\\main\\resources\\mapper";
|
}
|
|
File outDirFile = new File(outDir);
|
if (!outDirFile.exists()) {
|
outDirFile.mkdirs();
|
}
|
File mapperDirFile = new File(mapperDir);
|
if (!mapperDirFile.exists()) {
|
mapperDirFile.mkdirs();
|
}
|
|
System.out.println(System.getProperty("user.dir"));
|
String finalOutDir = outDir;
|
String finalMapperDir = mapperDir;
|
|
// 自定义vo、form、vue文件的的模板路径
|
Map<String, String> customFileMap = new HashMap();
|
customFileMap.put(GenFileTypeEnum.FORM.getValue(), "/templates/form.java.vm");
|
customFileMap.put(GenFileTypeEnum.VO.getValue(), "/templates/vo.java.vm");
|
customFileMap.put(GenFileTypeEnum.QUERY.getValue(), "/templates/query.java.vm");
|
customFileMap.put(GenFileTypeEnum.VIEW.getValue(), "/templates/vueView.vue.vm");
|
customFileMap.put(GenFileTypeEnum.DIALOG.getValue(), "/templates/vueDialog.vue.vm");
|
customFileMap.put(GenFileTypeEnum.TABLE.getValue(), "/templates/vueTable.vue.vm");
|
customFileMap.put(GenFileTypeEnum.STORE.getValue(), "/templates/vueStore.js.vm");
|
customFileMap.put(GenFileTypeEnum.API.getValue(), "/templates/vueApi.js.vm");
|
|
// 自定义数据
|
Map<String, Object> customDataMap = new HashMap();
|
// 数据校验分组
|
customDataMap.put("addGroup", data.getAddGroupPath());
|
customDataMap.put("updateGroup", data.getUpdateGroupPath());
|
// 统一响应类
|
customDataMap.put("responseDataName", data.getResponseClassName());
|
customDataMap.put("responseDataPath", data.getResponseClassPath());
|
// 分页类名
|
customDataMap.put("pageClassName", data.getPageClassName());
|
// 分页类所在路径
|
customDataMap.put("pagePath", data.getPagePath());
|
|
List<TableColumnVO> refColumnList = data.getColumns()
|
.stream()
|
.filter(item -> org.springframework.util.StringUtils.hasText(item.getRefColumn()))
|
.collect(Collectors.toList());
|
customDataMap.put("refColumnList", refColumnList);
|
if (CollectionUtils.isEmpty(refColumnList)) {
|
customDataMap.put("hasRef", Boolean.FALSE);
|
} else {
|
customDataMap.put("hasRef", Boolean.TRUE);
|
}
|
FastAutoGenerator.create(new DataSourceConfig.Builder(dataSource))
|
.globalConfig(builder -> {
|
builder.author("向培") //作者
|
.outputDir(finalOutDir) //输出路径(写到java目录)
|
.enableSwagger() //开启swagger
|
.commentDate("yyyy-MM-dd");
|
//.fileOverride(); //开启覆盖之前生成的文件
|
|
})
|
.packageConfig(builder -> {
|
builder.parent("com.monkeylessey")
|
//.moduleName("onceCode")
|
.entity("domain.entity")
|
.service("service")
|
.serviceImpl("service.impl")
|
.controller("controller")
|
.mapper("mapper")
|
.other("domain")
|
.xml("mapper")
|
.pathInfo(Collections.singletonMap(OutputFile.other, finalOutDir + "\\domain"))
|
.pathInfo(Collections.singletonMap(OutputFile.xml, finalMapperDir));
|
})
|
.strategyConfig(builder -> {
|
builder.addInclude(data.getTableName())
|
.addTablePrefix(data.getTablePrefix())
|
.serviceBuilder()
|
.formatServiceFileName("%sService")
|
.formatServiceImplFileName("%sServiceImpl")
|
.entityBuilder()
|
// 需要说明的是，父类是需要自己创建的，下面的父类字段也是你自己创建才行。plus不会说发现父类并没有id而创建id字段
|
.superClass(AbsEntity.class)
|
// 指定哪些字段在父表中，实体类将不会生成这些字段了
|
.addSuperEntityColumns("id", "gmt_create", "gmt_update", "deleted")
|
.enableLombok()
|
// 开启序列化
|
//.disableSerialVersionUID()
|
//.logicDeleteColumnName("deleted")
|
.enableTableFieldAnnotation()
|
.controllerBuilder()
|
// 映射路径使用连字符格式，而不是驼峰
|
.enableHyphenStyle()
|
.formatFileName("%sController")
|
.enableRestStyle()
|
.mapperBuilder()
|
//生成通用的resultMap
|
.enableBaseResultMap()
|
.superClass(BaseMapper.class)
|
.formatMapperFileName("%sMapper")
|
.enableMapperAnnotation()
|
.formatXmlFileName("%sMapper");
|
})
|
.injectionConfig(builder -> {
|
builder.customMap(customDataMap)
|
.customFile(customFileMap);
|
})
|
.templateConfig(new Consumer<TemplateConfig.Builder>() {
|
// 使用自定义entity模板
|
@Override
|
public void accept(TemplateConfig.Builder builder) {
|
builder.entity("templates/entity.java");
|
}
|
})
|
// 默认的是Velocity引擎模板
|
.templateEngine(new MyVelocityTemplateEngine(data, sysMenuMapper))
|
.execute();
|
}
|
}
|
