Skip to content

Commit

Permalink
feat(generator): 代码生成同步最新数据表结构支持同步排序
Browse files Browse the repository at this point in the history
  • Loading branch information
Charles7c committed Mar 7, 2024
1 parent b40d872 commit 89546de
Show file tree
Hide file tree
Showing 5 changed files with 32 additions and 28 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,6 @@ public interface FieldConfigMapper extends BaseMapper<FieldConfigDO> {
* @param tableName 表名称
* @return 字段配置信息
*/
@Select("SELECT * FROM gen_field_config WHERE table_name = #{tableName}")
@Select("SELECT * FROM gen_field_config WHERE table_name = #{tableName} ORDER BY field_sort ASC")
List<FieldConfigDO> selectListByTableName(@Param("tableName") String tableName);
}
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@
import com.baomidou.mybatisplus.annotation.TableName;
import io.swagger.v3.oas.annotations.media.Schema;
import jakarta.validation.constraints.NotBlank;
import jakarta.validation.constraints.NotNull;
import lombok.Data;
import lombok.NoArgsConstructor;
import lombok.NonNull;
Expand Down Expand Up @@ -93,6 +94,13 @@ public class FieldConfigDO implements Serializable {
@NotBlank(message = "字段类型不能为空")
private String fieldType;

/**
* 字段排序
*/
@Schema(description = "字段排序", example = "字段排序")
@NotNull(message = "字段排序不能为空")
private Integer fieldSort;

/**
* 注释
*/
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -130,36 +130,29 @@ public GenConfigDO getGenConfig(String tableName) throws SQLException {
@Override
public List<FieldConfigDO> listFieldConfig(String tableName, Boolean requireSync) {
List<FieldConfigDO> fieldConfigList = fieldConfigMapper.selectListByTableName(tableName);
if (CollUtil.isEmpty(fieldConfigList)) {
Collection<Column> columnList = MetaUtils.getColumns(dataSource, tableName);
return columnList.stream().map(FieldConfigDO::new).toList();
if (CollUtil.isNotEmpty(fieldConfigList) && Boolean.FALSE.equals(requireSync)) {
return fieldConfigList;
}
// 同步最新数据表列信息
if (Boolean.TRUE.equals(requireSync)) {
Collection<Column> columnList = MetaUtils.getColumns(dataSource, tableName);
// 移除已不存在的字段配置
List<String> columnNameList = columnList.stream().map(Column::getName).toList();
fieldConfigList.removeIf(column -> !columnNameList.contains(column.getColumnName()));
// 新增或更新字段配置
Map<String, FieldConfigDO> fieldConfigMap = fieldConfigList.stream()
.collect(Collectors.toMap(FieldConfigDO::getColumnName, Function.identity(), (key1, key2) -> key2));
for (Column column : columnList) {
FieldConfigDO fieldConfig = fieldConfigMap.get(column.getName());
if (null != fieldConfig) {
// 更新已有字段配置
String columnType = StrUtil.splitToArray(column.getTypeName(), StringConstants.SPACE)[0]
.toLowerCase();
fieldConfig.setColumnType(columnType);
fieldConfig.setColumnSize(Convert.toStr(column.getSize()));
fieldConfig.setComment(column.getComment());
} else {
// 新增字段配置
fieldConfig = new FieldConfigDO(column);
fieldConfigList.add(fieldConfig);
}
List<FieldConfigDO> latestFieldConfigList = new ArrayList<>();
// 获取最新数据表列信息
Collection<Column> columnList = MetaUtils.getColumns(dataSource, tableName);
// 新增或更新字段配置
Map<String, FieldConfigDO> fieldConfigMap = fieldConfigList.stream()
.collect(Collectors.toMap(FieldConfigDO::getColumnName, Function.identity(), (key1, key2) -> key2));
int i = 1;
for (Column column : columnList) {
FieldConfigDO fieldConfig = Optional.ofNullable(fieldConfigMap.get(column.getName()))
.orElseGet(() -> new FieldConfigDO(column));
fieldConfig.setFieldSort(i++);
// 更新已有字段配置
if (null != fieldConfig.getCreateTime()) {
String columnType = StrUtil.splitToArray(column.getTypeName(), StringConstants.SPACE)[0].toLowerCase();
fieldConfig.setColumnType(columnType);
fieldConfig.setColumnSize(Convert.toStr(column.getSize()));
}
latestFieldConfigList.add(fieldConfig);
}
return fieldConfigList;
return latestFieldConfigList;
}

@Override
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -294,6 +294,7 @@ CREATE TABLE IF NOT EXISTS `gen_field_config` (
`column_size` bigint(20) DEFAULT NULL COMMENT '列大小',
`field_name` varchar(64) NOT NULL COMMENT '字段名称',
`field_type` varchar(25) NOT NULL COMMENT '字段类型',
`field_sort` int NOT NULL DEFAULT 999 COMMENT '字段排序',
`comment` varchar(512) DEFAULT NULL COMMENT '注释',
`is_required` bit(1) NOT NULL DEFAULT b'1' COMMENT '是否必填',
`show_in_list` bit(1) NOT NULL DEFAULT b'1' COMMENT '是否在列表中显示',
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -490,6 +490,7 @@ CREATE TABLE IF NOT EXISTS "gen_field_config" (
"column_size" int8 DEFAULT NULL,
"field_name" varchar(64) NOT NULL,
"field_type" varchar(25) NOT NULL,
"field_sort" int4 NOT NULL DEFAULT 999,
"comment" varchar(512) DEFAULT NULL,
"is_required" bool NOT NULL DEFAULT true,
"show_in_list" bool NOT NULL DEFAULT true,
Expand All @@ -506,6 +507,7 @@ COMMENT ON COLUMN "gen_field_config"."column_type" IS '列类型';
COMMENT ON COLUMN "gen_field_config"."column_size" IS '列大小';
COMMENT ON COLUMN "gen_field_config"."field_name" IS '字段名称';
COMMENT ON COLUMN "gen_field_config"."field_type" IS '字段类型';
COMMENT ON COLUMN "gen_field_config"."field_sort" IS '字段排序';
COMMENT ON COLUMN "gen_field_config"."comment" IS '注释';
COMMENT ON COLUMN "gen_field_config"."is_required" IS '是否必填';
COMMENT ON COLUMN "gen_field_config"."show_in_list" IS '是否在列表中显示';
Expand Down

0 comments on commit 89546de

Please sign in to comment.