toolchaintechnologycenter/spdm-backend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

工况库绑定

Browse Source
This commit is contained in:
gulongcheng
2025-12-10 17:10:26 +08:00
parent 1d0fac1cc0
commit ddbcb9f4e1
4 changed files with 14 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
common/src/main/java/com/sdm/common/entity/pojo/task/TaskBaseInfo.java
View File

@@ -16,7 +16,7 @@ public class TaskBaseInfo {
public String version;
public Long poolId;
public Integer poolId;
public Long fileId;

2
common/src/main/java/com/sdm/common/entity/resp/data/FileSimulationMappingResp.java
View File

@@ -29,7 +29,7 @@ public class FileSimulationMappingResp implements Serializable {
private Long fileId;
private Long simulationPoolId;
private Integer simulationPoolId;
private String simulationPoolVersion;

2
common/src/main/java/com/sdm/common/entity/resp/data/PoolInfo.java
View File

@@ -7,7 +7,7 @@ import java.util.List;
@Data
public class PoolInfo {
private Long simulationPoolId;
private Integer simulationPoolId;
private String simulationPoolName;
private String simulationPoolVersion;
private List<String> simulationPoolTaskIds;

19
data/src/main/java/com/sdm/data/service/impl/MinioFileIDataFileServiceImpl.java
View File

@@ -2334,27 +2334,30 @@ public class MinioFileIDataFileServiceImpl implements IDataFileService {
if(CollectionUtils.isEmpty(fileSimulationMappingByFileId)){
return;
}
log.info("文件绑定知识库的查询结果 :{}",fileSimulationMappingByFileId);
fileSimulationMappingByFileId.forEach(fileSimulationMappingResp -> {
SdmResponse<Map<String, TaskBaseInfo>> taskPoolTaskMap = isSimuluationTaskPoolFeignClient.getTaskPoolTaskMap(fileSimulationMappingResp.getSimulationPoolId());
if(taskPoolTaskMap.isSuccess() && !Objects.isNull(taskPoolTaskMap.getData())) {
Map<String, TaskBaseInfo> data = taskPoolTaskMap.getData();
TaskBaseInfo taskBaseInfo = data.getOrDefault(fileSimulationMappingResp.getSimulationPoolTaskId(), null);
taskBaseInfo.setPoolId(fileSimulationMappingResp.getSimulationPoolId());
taskBaseInfo.setFileId(fileSimulationMappingResp.getFileId());
if(ObjectUtils.isNotEmpty(taskBaseInfo)) {
taskBaseInfo.setPoolId(fileSimulationMappingResp.getSimulationPoolId());
taskBaseInfo.setFileId(fileSimulationMappingResp.getFileId());
}
result.add(taskBaseInfo);
}
});
// 按 fileId 和 poolId 双重分组
Map<Long, Map<Long, List<TaskBaseInfo>>> fileTaskPoolMap = result.stream()
Map<Long, Map<Integer, List<TaskBaseInfo>>> fileTaskPoolMap = result.stream()
.collect(Collectors.groupingBy(
TaskBaseInfo::getFileId,
Collectors.groupingBy(TaskBaseInfo::getPoolId)
));
// 按 fileId 和 poolId 分组收集 simulationPoolTaskId
Map<Long, Map<Long, List<String>>> filePoolTaskIdsMap = fileSimulationMappingByFileId.stream()
Map<Long, Map<Integer, List<String>>> filePoolTaskIdsMap = fileSimulationMappingByFileId.stream()
.collect(Collectors.groupingBy(
FileSimulationMappingResp::getFileId,
Collectors.groupingBy(
@@ -2367,12 +2370,12 @@ public class MinioFileIDataFileServiceImpl implements IDataFileService {
for (FileMetadataInfo fileMetadataInfo : list) {
Long fileId = fileMetadataInfo.getId();
if (fileTaskPoolMap.containsKey(fileId)) {
Map<Long, List<TaskBaseInfo>> poolMap = fileTaskPoolMap.get(fileId);
Map<Long, List<String>> poolTaskIdsMap = filePoolTaskIdsMap.getOrDefault(fileId, new HashMap<>());
Map<Integer, List<TaskBaseInfo>> poolMap = fileTaskPoolMap.get(fileId);
Map<Integer, List<String>> poolTaskIdsMap = filePoolTaskIdsMap.getOrDefault(fileId, new HashMap<>());
List<PoolInfo> poolInfos = new ArrayList<>();
for (Map.Entry<Long, List<TaskBaseInfo>> poolEntry : poolMap.entrySet()) {
Long poolId = poolEntry.getKey();
for (Map.Entry<Integer, List<TaskBaseInfo>> poolEntry : poolMap.entrySet()) {
Integer poolId = poolEntry.getKey();
List<TaskBaseInfo> taskList = poolEntry.getValue();
List<String> taskIds = poolTaskIdsMap.getOrDefault(poolId, new ArrayList<>());