root 2 سال پیش
والد
کامیت
dcd876bf99

+ 7 - 3
simulation-resource-scheduler/src/main/java/com/css/simulation/resource/scheduler/consumer/ProjectConsumer.java

@@ -113,7 +113,7 @@ public class ProjectConsumer {
     public void createTaskAndFixData(ConsumerRecord<String, String> projectRecord) {
         //* -------------------------------- 0 读取消息,创建临时目录 --------------------------------
         String initialProjectJson = projectRecord.value();
-        log.info("ProjectConsumer.fixedData() 接收到项目开始消息为:" + initialProjectJson);
+        log.info("createTaskAndFixData() 接收到项目开始消息为:" + initialProjectJson);
         ProjectMessageDTO projectMessageDTO = JsonUtil.jsonToBean(initialProjectJson, ProjectMessageDTO.class);
         String projectId = projectMessageDTO.getProjectId();        // 手动执行项目 id 或 自动执行子项目 id
         String packageId = projectMessageDTO.getScenePackageId();   // 场景测试包 id
@@ -134,6 +134,7 @@ public class ProjectConsumer {
         indexMapper.deleteFirstTargetScoreByProjectId(projectId);
         indexMapper.deleteLastTargetScoreByProjectId(projectId);
         // -------------------------------- 1 查询场景 --------------------------------
+        log.info("createTaskAndFixData() 项目 " + projectId + " 开始查询场景。");
         //根据场景测试包 packageId,拿到场景集合(不包括重复场景),重复场景会在发送消息时根据叶子指标发送多次。
         List<ScenePO> scenePOList = projectService.getSceneList(projectId, packageId, projectPath);
         int taskTotal = scenePOList.size();
@@ -142,13 +143,16 @@ public class ProjectConsumer {
         //去重,之后发送消息的时候会补全指标,如果不去重的话会出现多个场景重复关联多个指标
         Set<ScenePO> scenePOSet = new HashSet<>(scenePOList);
         // -------------------------------- 2 查询模型 --------------------------------
+        log.info("createTaskAndFixData() 项目 " + projectId + " 开始查询模型。");
         //2-1 根据车辆配置id vehicleConfigId, 获取 模型信息和传感器信息
         VehiclePO vehiclePO = vehicleMapper.selectByVehicleConfigId(vehicleConfigId);   // 车辆
         List<CameraPO> cameraPOList = sensorCameraMapper.selectCameraByVehicleConfigId(vehicleConfigId);    // 摄像头
         List<OgtPO> ogtPOList = sensorOgtMapper.selectOgtByVehicleId(vehicleConfigId); // 完美传感器
         // -------------------------------- 3 算法导入 --------------------------------
+        log.info("createTaskAndFixData() 项目 " + projectId + " 开始算法导入。");
         String algorithmDockerImage = projectService.handleAlgorithm(projectId, algorithmId);
         // -------------------------------- 4 保存任务消息 --------------------------------
+        log.info("createTaskAndFixData() 项目 " + projectId + " 开始保存任务消息。");
         List<TaskPO> taskList = new ArrayList<>();
         for (ScenePO scenePO : scenePOSet) {
             String sceneId = scenePO.getId();
@@ -160,7 +164,7 @@ public class ProjectConsumer {
                 lastTargetIdList = indexMapper.selectLeafIndexIdByAutoSubProjectIdAndSceneId(projectId, "%" + sceneId + "%");
             }
             if (CollectionUtil.isEmpty(lastTargetIdList)) {
-                throw new RuntimeException("ProjectConsumer.createTaskAndFixData() -- 项目 " + projectId + " 使用的场景测试包 " + sceneId + " 不存在指标。");
+                throw new RuntimeException("createTaskAndFixData() -- 项目 " + projectId + " 使用的场景测试包 " + sceneId + " 不存在指标。");
             }
             for (String lastTargetId : lastTargetIdList) {
                 String taskId = StringUtil.getRandomUUID();
@@ -264,7 +268,7 @@ public class ProjectConsumer {
             }
         }
         taskManager.batchInsertTask(taskList);
-        log.info("ProjectConsumer.createTaskAndFixData() 项目 " + projectId + " 创建任务:" + taskList);
+        log.info("createTaskAndFixData() 项目 " + projectId + " 创建任务:" + taskList);
         //* -------------------------------- 4 开始排队 --------------------------------
         cacheProject(projectRecord);
 

+ 2 - 2
simulation-resource-scheduler/src/main/java/com/css/simulation/resource/scheduler/service/ProjectService.java

@@ -132,7 +132,7 @@ public class ProjectService {
         stringRedisTemplate.opsForValue().set(allIndexPrefix, JsonUtil.listToJson(allIndexList));
         //2 查询场景包叶子指标
         List<IndexTemplatePO> leafIndexList = allIndexList.stream().filter(index -> StringUtil.isNotEmpty(index.getRuleId())).collect(Collectors.toList());
-        log.info("ProjectService--handlePackage 项目 " + projectId + " 的叶子指标为:" + leafIndexList);
+        log.info("ProjectService--getSceneList 项目 " + projectId + " 的叶子指标为:" + leafIndexList);
 //        FileUtil.writeStringToLocalFile(JsonUtil.listToJson(allIndexList), projectPath + "leaf-index-list.json");
         stringRedisTemplate.opsForValue().set(leafIndexPrefix, JsonUtil.listToJson(leafIndexList));
         List<ScenePO> sceneList = new ArrayList<>();
@@ -158,7 +158,7 @@ public class ProjectService {
                 sceneList.addAll(sceneMapper.selectGeneralizationByIdList(generalizationIdList));
             }
         });
-        log.info("ProjectService--handlePackage 项目" + projectId + " 共有 " + sceneList.size() + " 个任务!");
+        log.info("ProjectService--getSceneList 项目" + projectId + " 共有 " + sceneList.size() + " 个任务!");
         return sceneList;
     }
 

+ 9 - 1
simulation-resource-scheduler/src/main/java/com/css/simulation/resource/scheduler/util/MinioUtil.java

@@ -6,6 +6,7 @@ import io.minio.http.Method;
 import io.minio.messages.Item;
 import jdk.nashorn.internal.runtime.regexp.joni.exception.InternalException;
 import lombok.SneakyThrows;
+import lombok.extern.slf4j.Slf4j;
 import org.springframework.web.multipart.MultipartFile;
 
 import java.io.File;
@@ -14,6 +15,7 @@ import java.io.InputStream;
 import java.security.InvalidKeyException;
 import java.security.NoSuchAlgorithmException;
 
+@Slf4j
 public class MinioUtil {
 
     /**
@@ -27,10 +29,15 @@ public class MinioUtil {
     public static void rmR(MinioClient minioClient, String bucketName, String prefix) {
         Iterable<Result<Item>> list = minioClient.listObjects(ListObjectsArgs.builder().bucket(bucketName)
                 .prefix(prefix).recursive(true).build());
+        if (list == null) {
+            log.info(prefix + " 不存在。");
+            return;
+        }
         for (Result<Item> object : list) {
             rm(minioClient, bucketName, object.get().objectName());
         }
     }
+
     /**
      * 删除文件
      *
@@ -64,6 +71,7 @@ public class MinioUtil {
             return false;
         }
     }
+
     /**
      * 创建 bucket
      */
@@ -143,7 +151,6 @@ public class MinioUtil {
     }
 
 
-
     /**
      * 下载文件
      */
@@ -164,6 +171,7 @@ public class MinioUtil {
                 .filename(targetFilePath)
                 .build());
     }
+
     /**
      * 下载文件
      */