diff --git a/collector-scheduling-management/src/main/java/com/docus/server/infrastructure/dao/impl/SchCollectorConfigDaoImpl.java b/collector-scheduling-management/src/main/java/com/docus/server/infrastructure/dao/impl/SchCollectorConfigDaoImpl.java index 1915c72..4a8a128 100644 --- a/collector-scheduling-management/src/main/java/com/docus/server/infrastructure/dao/impl/SchCollectorConfigDaoImpl.java +++ b/collector-scheduling-management/src/main/java/com/docus/server/infrastructure/dao/impl/SchCollectorConfigDaoImpl.java @@ -87,8 +87,8 @@ public class SchCollectorConfigDaoImpl extends BaseDaoImpl search(SearchDTO searchDTO) { - return SchCollectorConfigConvert.INSTANCE.convertVO(iSchCollectorConfigDao.search(searchDTO)); + + PageResult result = SchCollectorConfigConvert.INSTANCE.convertVO(iSchCollectorConfigDao.search(searchDTO)); + if (CollectionUtils.isEmpty(result.getList())) { + return new PageResult<>(); + } + + List schSystemParams = iSchSystemParamsDao.findBy("paramValue", ListUtils.distinctSelect(result.getList(), SchCollectorConfigVO::getCollectorId)); + Map taskConfigRetryLogMap = ListUtils.toMap(schSystemParams, SchSystemParams::getParamValue); + + Setters.instance().list(result.getList()).cycleSetProperties(p -> { + String collectorId = String.valueOf(p.getCollectorId()); + if (taskConfigRetryLogMap.containsKey(collectorId)) { + p.setCollectorName(taskConfigRetryLogMap.get(collectorId).getParamName()); + } else { + p.setCollectorName("公共"); + } + }); + + return result; } /** @@ -59,15 +87,18 @@ public class SchCollectorConfigServiceImpl implements ISchCollectorConfigService @Override public boolean add(AddSchCollectorConfigDTO addSchCollectorConfigDTO) { - String collectorCode = addSchCollectorConfigDTO.getCollectorCode(); + Long collectorId = addSchCollectorConfigDTO.getCollectorId(); - SchCollectorConfig schCollectorConfigDO = iSchCollectorConfigDao.findOneBy("collectorCode", collectorCode); + SchCollectorConfig schCollectorConfigDO = iSchCollectorConfigDao.findOneBy("collectorId", collectorId); if (null == schCollectorConfigDO) { + SchCollectorConfig schCollectorConfig = SchCollectorConfigConvert.INSTANCE.convertDO(addSchCollectorConfigDTO); schCollectorConfig.setId(idService.getDateSeq()); - + if (addSchCollectorConfigDTO.getConfigType().equals(ConfigTypeEnum.PUBLIC_CONFIG)) { + schCollectorConfig.setCollectorId(0L); //共有类型 + } return iSchCollectorConfigDao.add(schCollectorConfig); } else { @@ -84,7 +115,8 @@ public class SchCollectorConfigServiceImpl implements ISchCollectorConfigService */ @Override public boolean edit(EditSchCollectorConfigDTO editSchCollectorConfigDTO) { - SchCollectorConfig schCollectorConfig = SchCollectorConfigConvert.INSTANCE.convertDO(editSchCollectorConfigDTO); + SchCollectorConfig schCollectorConfig = iSchCollectorConfigDao.findById(editSchCollectorConfigDTO.getId()); + schCollectorConfig.setConfigJson(editSchCollectorConfigDTO.getConfigJson()); return iSchCollectorConfigDao.edit(schCollectorConfig); } diff --git a/collector-scheduling-management/src/main/resources/mapper/SchCollectorConfigMapper.xml b/collector-scheduling-management/src/main/resources/mapper/SchCollectorConfigMapper.xml index 3dcaff8..579ecab 100644 --- a/collector-scheduling-management/src/main/resources/mapper/SchCollectorConfigMapper.xml +++ b/collector-scheduling-management/src/main/resources/mapper/SchCollectorConfigMapper.xml @@ -6,7 +6,7 @@ - + @@ -15,7 +15,7 @@ - id, collector_code, config_type, config_json, create_time, update_time + id, collector_id, config_type, config_json, create_time, update_time diff --git a/docus-client-interface/src/main/java/com/docus/server/dto/scheduling.management/schcollectorconfig/AddSchCollectorConfigDTO.java b/docus-client-interface/src/main/java/com/docus/server/dto/scheduling.management/schcollectorconfig/AddSchCollectorConfigDTO.java index 31db221..04282ed 100644 --- a/docus-client-interface/src/main/java/com/docus/server/dto/scheduling.management/schcollectorconfig/AddSchCollectorConfigDTO.java +++ b/docus-client-interface/src/main/java/com/docus/server/dto/scheduling.management/schcollectorconfig/AddSchCollectorConfigDTO.java @@ -18,8 +18,8 @@ import java.io.Serializable; @ApiModel(value = "AddSchCollectorConfigDTO对象", description = "采集器配置") public class AddSchCollectorConfigDTO implements Serializable { - @ApiModelProperty(value = "采集器类型") - private String collectorCode; + @ApiModelProperty(value = "采集器id") + private Long collectorId; @NotNull @ApiModelProperty(value = "配置类型,0:采集器公共配置,1:采集器私有配置") diff --git a/docus-client-interface/src/main/java/com/docus/server/dto/scheduling.management/schcollectorconfig/EditSchCollectorConfigDTO.java b/docus-client-interface/src/main/java/com/docus/server/dto/scheduling.management/schcollectorconfig/EditSchCollectorConfigDTO.java index 9c01e0f..c8eb7c5 100644 --- a/docus-client-interface/src/main/java/com/docus/server/dto/scheduling.management/schcollectorconfig/EditSchCollectorConfigDTO.java +++ b/docus-client-interface/src/main/java/com/docus/server/dto/scheduling.management/schcollectorconfig/EditSchCollectorConfigDTO.java @@ -16,12 +16,9 @@ import java.io.Serializable; @ApiModel(value = "EditSchCollectorConfigDTO对象", description = "采集器配置") public class EditSchCollectorConfigDTO implements Serializable { - @ApiModelProperty(value = "主键") + @ApiModelProperty(value = "id") private Long id; - @ApiModelProperty(value = "采集器类型") - private String collectorCode; - @ApiModelProperty(value = "配置参数") private String configJson; diff --git a/docus-client-interface/src/main/java/com/docus/server/entity/scheduling.management/SchCollectorConfig.java b/docus-client-interface/src/main/java/com/docus/server/entity/scheduling.management/SchCollectorConfig.java index a31cca4..ac1894f 100644 --- a/docus-client-interface/src/main/java/com/docus/server/entity/scheduling.management/SchCollectorConfig.java +++ b/docus-client-interface/src/main/java/com/docus/server/entity/scheduling.management/SchCollectorConfig.java @@ -30,9 +30,9 @@ public class SchCollectorConfig implements Serializable { @TableId(value = "id", type = IdType.ASSIGN_UUID) private Long id; - @ApiModelProperty(value = "采集器类型") - @TableField("collector_code") - private String collectorCode; + @ApiModelProperty(value = "采集器id") + @TableField("collector_id") + private Long collectorId; @ApiModelProperty(value = "配置类型,0:采集器公共配置,1:采集器私有配置") @TableField("config_type") diff --git a/docus-client-interface/src/main/java/com/docus/server/vo/scheduling.management/schcollectorconfig/SchCollectorConfigVO.java b/docus-client-interface/src/main/java/com/docus/server/vo/scheduling.management/schcollectorconfig/SchCollectorConfigVO.java index ea51b50..7516b98 100644 --- a/docus-client-interface/src/main/java/com/docus/server/vo/scheduling.management/schcollectorconfig/SchCollectorConfigVO.java +++ b/docus-client-interface/src/main/java/com/docus/server/vo/scheduling.management/schcollectorconfig/SchCollectorConfigVO.java @@ -22,8 +22,11 @@ public class SchCollectorConfigVO implements Serializable { @ApiModelProperty(value = "主键") private Long id; + @ApiModelProperty(value = "采集器Id") + private Long collectorId; + @ApiModelProperty(value = "采集器类型") - private String collectorCode; + private String collectorName; @ApiModelProperty(value = "配置类型,0:采集器公共配置,1:采集器私有配置") private ConfigTypeEnum configType;