|
|
|
|
@ -0,0 +1,222 @@
|
|
|
|
|
package cn.iocoder.dashboard.modules.tool.service.codegen.impl;
|
|
|
|
|
|
|
|
|
|
import cn.hutool.core.collection.CollUtil;
|
|
|
|
|
import cn.hutool.core.map.MapUtil;
|
|
|
|
|
import cn.hutool.core.util.StrUtil;
|
|
|
|
|
import cn.iocoder.dashboard.framework.mybatis.core.dataobject.BaseDO;
|
|
|
|
|
import cn.iocoder.dashboard.modules.tool.convert.codegen.CodegenConvert;
|
|
|
|
|
import cn.iocoder.dashboard.modules.tool.dal.mysql.dao.coegen.ToolCodegenColumnMapper;
|
|
|
|
|
import cn.iocoder.dashboard.modules.tool.dal.mysql.dao.coegen.ToolCodegenTableMapper;
|
|
|
|
|
import cn.iocoder.dashboard.modules.tool.dal.mysql.dao.coegen.ToolInformationSchemaColumnMapper;
|
|
|
|
|
import cn.iocoder.dashboard.modules.tool.dal.mysql.dao.coegen.ToolInformationSchemaTableMapper;
|
|
|
|
|
import cn.iocoder.dashboard.modules.tool.dal.mysql.dataobject.codegen.ToolCodegenColumnDO;
|
|
|
|
|
import cn.iocoder.dashboard.modules.tool.dal.mysql.dataobject.codegen.ToolCodegenTableDO;
|
|
|
|
|
import cn.iocoder.dashboard.modules.tool.dal.mysql.dataobject.codegen.ToolInformationSchemaColumnDO;
|
|
|
|
|
import cn.iocoder.dashboard.modules.tool.dal.mysql.dataobject.codegen.ToolInformationSchemaTableDO;
|
|
|
|
|
import cn.iocoder.dashboard.modules.tool.enums.codegen.ToolCodegenColumnHtmlTypeEnum;
|
|
|
|
|
import cn.iocoder.dashboard.modules.tool.enums.codegen.ToolCodegenColumnListConditionEnum;
|
|
|
|
|
import cn.iocoder.dashboard.modules.tool.service.codegen.ToolCodegenService;
|
|
|
|
|
import com.google.common.collect.Sets;
|
|
|
|
|
import org.springframework.stereotype.Service;
|
|
|
|
|
import org.springframework.transaction.annotation.Transactional;
|
|
|
|
|
|
|
|
|
|
import javax.annotation.Resource;
|
|
|
|
|
import java.math.BigDecimal;
|
|
|
|
|
import java.util.*;
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* 代码生成 Service 实现类
|
|
|
|
|
*
|
|
|
|
|
* @author 芋道源码
|
|
|
|
|
*/
|
|
|
|
|
@Service
|
|
|
|
|
public class ToolCodegenServiceImpl implements ToolCodegenService {
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* 字段名与 {@link ToolCodegenColumnListConditionEnum} 的默认映射
|
|
|
|
|
* 注意，字段的匹配以后缀的方式
|
|
|
|
|
*/
|
|
|
|
|
private static final Map<String, ToolCodegenColumnListConditionEnum> columnListOperationConditionMappings =
|
|
|
|
|
MapUtil.<String, ToolCodegenColumnListConditionEnum>builder()
|
|
|
|
|
.put("name", ToolCodegenColumnListConditionEnum.LIKE)
|
|
|
|
|
.put("time", ToolCodegenColumnListConditionEnum.BETWEEN)
|
|
|
|
|
.put("date", ToolCodegenColumnListConditionEnum.BETWEEN)
|
|
|
|
|
.build();
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* 字段名与 {@link ToolCodegenColumnHtmlTypeEnum} 的默认映射
|
|
|
|
|
* 注意，字段的匹配以后缀的方式
|
|
|
|
|
*/
|
|
|
|
|
private static final Map<String, ToolCodegenColumnHtmlTypeEnum> columnHtmlTypeMappings =
|
|
|
|
|
MapUtil.<String, ToolCodegenColumnHtmlTypeEnum>builder()
|
|
|
|
|
.put("status", ToolCodegenColumnHtmlTypeEnum.RADIO)
|
|
|
|
|
.put("sex", ToolCodegenColumnHtmlTypeEnum.RADIO)
|
|
|
|
|
.put("type", ToolCodegenColumnHtmlTypeEnum.SELECT)
|
|
|
|
|
.put("image", ToolCodegenColumnHtmlTypeEnum.UPLOAD_IMAGE)
|
|
|
|
|
.put("file", ToolCodegenColumnHtmlTypeEnum.UPLOAD_FILE)
|
|
|
|
|
.put("content", ToolCodegenColumnHtmlTypeEnum.EDITOR)
|
|
|
|
|
.build();
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* 新增操作，不需要传递的字段
|
|
|
|
|
*/
|
|
|
|
|
private static final Set<String> CREATE_OPERATION_EXCLUDE_COLUMN = Sets.newHashSet(
|
|
|
|
|
"id");
|
|
|
|
|
/**
|
|
|
|
|
* 修改操作，不需要传递的字段
|
|
|
|
|
*/
|
|
|
|
|
private static final Set<String> UPDATE_OPERATION_EXCLUDE_COLUMN = Sets.newHashSet();
|
|
|
|
|
/**
|
|
|
|
|
* 列表操作的条件，不需要传递的字段
|
|
|
|
|
*/
|
|
|
|
|
private static final Set<String> LIST_OPERATION_CONDITION_COLUMN = Sets.newHashSet();
|
|
|
|
|
/**
|
|
|
|
|
* 列表操作的结果，不需要返回的字段
|
|
|
|
|
*/
|
|
|
|
|
private static final Set<String> LIST_OPERATION_RESULT_COLUMN = Sets.newHashSet();
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Java 类型与 MySQL 类型的映射关系
|
|
|
|
|
*/
|
|
|
|
|
private static final Map<String, Set<String>> javaTypeMappings = MapUtil.<String, Set<String>>builder()
|
|
|
|
|
.put(Boolean.class.getSimpleName(), Sets.newHashSet("bit"))
|
|
|
|
|
.put(Integer.class.getSimpleName(), Sets.newHashSet("tinyint", "smallint", "mediumint", "int"))
|
|
|
|
|
.put(Long.class.getSimpleName(), Collections.singleton("bigint"))
|
|
|
|
|
.put(Double.class.getSimpleName(), Sets.newHashSet("float", "double"))
|
|
|
|
|
.put(BigDecimal.class.getSimpleName(), Sets.newHashSet("decimal", "numeric"))
|
|
|
|
|
.put(String.class.getSimpleName(), Sets.newHashSet("tinytext", "text", "mediumtext", "longtext", // 长文本
|
|
|
|
|
"char", "varchar", "nvarchar", "varchar2")) // 短文本
|
|
|
|
|
.put(Date.class.getSimpleName(), Sets.newHashSet("datetime", "time", "date", "timestamp"))
|
|
|
|
|
.build();
|
|
|
|
|
|
|
|
|
|
static {
|
|
|
|
|
Arrays.stream(BaseDO.class.getDeclaredFields()).forEach(field -> {
|
|
|
|
|
CREATE_OPERATION_EXCLUDE_COLUMN.add(field.getName());
|
|
|
|
|
UPDATE_OPERATION_EXCLUDE_COLUMN.add(field.getName());
|
|
|
|
|
LIST_OPERATION_CONDITION_COLUMN.add(field.getName());
|
|
|
|
|
LIST_OPERATION_RESULT_COLUMN.add(field.getName());
|
|
|
|
|
});
|
|
|
|
|
LIST_OPERATION_RESULT_COLUMN.remove("create_time"); // 创建时间，还是需要返回的
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Resource
|
|
|
|
|
private ToolInformationSchemaTableMapper informationSchemaTableMapper;
|
|
|
|
|
@Resource
|
|
|
|
|
private ToolInformationSchemaColumnMapper informationSchemaColumnMapper;
|
|
|
|
|
@Resource
|
|
|
|
|
private ToolCodegenTableMapper codegenTableMapper;
|
|
|
|
|
@Resource
|
|
|
|
|
private ToolCodegenColumnMapper codegenColumnMapper;
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
@Transactional
|
|
|
|
|
public Long createCodegenTable(String tableName) {
|
|
|
|
|
// 从数据库中，获得数据库表结构
|
|
|
|
|
ToolInformationSchemaTableDO schemaTable = informationSchemaTableMapper.selectByTableName(tableName);
|
|
|
|
|
if (schemaTable == null) {
|
|
|
|
|
throw new RuntimeException(""); // TODO
|
|
|
|
|
}
|
|
|
|
|
List<ToolInformationSchemaColumnDO> schemaColumns = informationSchemaColumnMapper.selectListByTableName(tableName);
|
|
|
|
|
if (CollUtil.isEmpty(schemaColumns)) {
|
|
|
|
|
throw new RuntimeException(""); // TODO
|
|
|
|
|
}
|
|
|
|
|
// 校验是否已经存在
|
|
|
|
|
if (codegenTableMapper.selectByTableName(tableName) != null) {
|
|
|
|
|
throw new RuntimeException(""); // TODO
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// 将 table 插入到数据库
|
|
|
|
|
ToolCodegenTableDO table = CodegenConvert.INSTANCE.convert(schemaTable);
|
|
|
|
|
initTableDefault(table);
|
|
|
|
|
codegenTableMapper.insert(table);
|
|
|
|
|
// 将 column 插入到数据库
|
|
|
|
|
List<ToolCodegenColumnDO> columns = CodegenConvert.INSTANCE.convertList(schemaColumns);
|
|
|
|
|
columns.forEach(column -> {
|
|
|
|
|
initColumnDefault(column);
|
|
|
|
|
column.setTableId(table.getId());
|
|
|
|
|
codegenColumnMapper.insert(column);
|
|
|
|
|
});
|
|
|
|
|
return table.getId();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* 初始化 Table 表的默认字段
|
|
|
|
|
*
|
|
|
|
|
* @param table 表定义
|
|
|
|
|
*/
|
|
|
|
|
private void initTableDefault(ToolCodegenTableDO table) {
|
|
|
|
|
table.setModuleName(StrUtil.subBefore(table.getTableName(),
|
|
|
|
|
'_', false)); // 第一个 _ 前缀的前面，作为 module 名字
|
|
|
|
|
table.setBusinessName(StrUtil.subAfter(table.getTableName(),
|
|
|
|
|
'_', false)); // 第一个 _ 前缀的后面，作为 module 名字
|
|
|
|
|
table.setBusinessName(StrUtil.toCamelCase(table.getBusinessName())); // 可能存在多个 _ 的情况，转换成驼峰
|
|
|
|
|
table.setClassName(StrUtil.toCamelCase(table.getClassName())); // 驼峰
|
|
|
|
|
table.setClassComment(StrUtil.subBefore(table.getClassComment(), // 去除结尾的表，作为类描述
|
|
|
|
|
'表', true));
|
|
|
|
|
table.setAuthor("芋艿"); // TODO 稍后改成创建人
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* 初始化 Column 列的默认字段
|
|
|
|
|
*
|
|
|
|
|
* @param column 列定义
|
|
|
|
|
*/
|
|
|
|
|
private void initColumnDefault(ToolCodegenColumnDO column) {
|
|
|
|
|
// 处理 Java 相关的字段的默认值
|
|
|
|
|
processColumnJava(column);
|
|
|
|
|
// 处理 CRUD 相关的字段的默认值
|
|
|
|
|
processColumnOperation(column);
|
|
|
|
|
// 处理 UI 相关的字段的默认值
|
|
|
|
|
processColumnUI(column);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private void processColumnJava(ToolCodegenColumnDO column) {
|
|
|
|
|
// 处理 javaField 字段
|
|
|
|
|
column.setJavaField(StrUtil.toCamelCase(column.getColumnName()));
|
|
|
|
|
// 处理 dictType 字段，暂无
|
|
|
|
|
// 处理 javaType 字段
|
|
|
|
|
String dbType = StrUtil.subBefore(column.getColumnName(), ')', false);
|
|
|
|
|
javaTypeMappings.entrySet().stream()
|
|
|
|
|
.filter(entry -> entry.getValue().contains(dbType))
|
|
|
|
|
.findFirst().ifPresent(entry -> column.setJavaType(entry.getKey()));
|
|
|
|
|
if (column.getJavaType() == null) {
|
|
|
|
|
throw new IllegalStateException(String.format("column(%s) 的数据库类型(%s) 找不到匹配的 Java 类型",
|
|
|
|
|
column.getColumnName(), column.getColumnType()));
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private void processColumnOperation(ToolCodegenColumnDO column) {
|
|
|
|
|
// 处理 createOperation 字段
|
|
|
|
|
column.setCreateOperation(!CREATE_OPERATION_EXCLUDE_COLUMN.contains(column.getColumnName())
|
|
|
|
|
&& !column.getPrimaryKey()); // 非主键
|
|
|
|
|
// 处理 updateOperation 字段
|
|
|
|
|
column.setUpdateOperation(!UPDATE_OPERATION_EXCLUDE_COLUMN.contains(column.getColumnName()));
|
|
|
|
|
// 处理 listOperationResult 字段
|
|
|
|
|
column.setListOperationResult(!LIST_OPERATION_RESULT_COLUMN.contains(column.getColumnName()));
|
|
|
|
|
// 处理 listOperationCondition 字段。默认设置为需要过滤的条件，手动进行取消
|
|
|
|
|
if (!LIST_OPERATION_CONDITION_COLUMN.contains(column.getColumnName())
|
|
|
|
|
&& !column.getPrimaryKey()) { // 非主键
|
|
|
|
|
column.setListOperationCondition(ToolCodegenColumnListConditionEnum.EQ.getCondition());
|
|
|
|
|
}
|
|
|
|
|
columnListOperationConditionMappings.entrySet().stream()
|
|
|
|
|
.filter(entry -> StrUtil.endWithIgnoreCase(column.getColumnName(), entry.getKey()))
|
|
|
|
|
.findFirst().ifPresent(entry -> column.setListOperationCondition(entry.getValue().getCondition()));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private void processColumnUI(ToolCodegenColumnDO column) {
|
|
|
|
|
// 基于后缀进行匹配
|
|
|
|
|
columnHtmlTypeMappings.entrySet().stream()
|
|
|
|
|
.filter(entry -> StrUtil.endWithIgnoreCase(column.getColumnName(), entry.getKey()))
|
|
|
|
|
.findFirst().ifPresent(entry -> column.setHtmlType(entry.getValue().getType()));
|
|
|
|
|
// 如果是 Boolean 类型时，设置为 radio 类型.
|
|
|
|
|
// 其它类型，因为字段名可以相对保障，所以不进行处理。例如说 date 对应 datetime 类型.
|
|
|
|
|
if (Boolean.class.getSimpleName().equals(column.getJavaType())) {
|
|
|
|
|
column.setHtmlType(ToolCodegenColumnHtmlTypeEnum.RADIO.getType());
|
|
|
|
|
}
|
|
|
|
|
// 兜底，设置默认为 input 类型
|
|
|
|
|
if (column.getHtmlType() == null) {
|
|
|
|
|
column.setHtmlType(ToolCodegenColumnHtmlTypeEnum.INPUT.getType());
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
}
|
YunaiV
5 years ago