瀏覽代碼

feat: start方法添加bizType参数

WANGKANG 9 月之前
父節點
當前提交
f0b3d3773b

+ 7 - 0
taais-modules/taais-biz/src/main/java/com/taais/biz/constant/BizConstant.java

@@ -7,6 +7,13 @@ public class BizConstant {
     // 视频文件存储目录
     public static final String UPLOAD_DIR = "uploads/";
 
+    public static class BizType {
+        public static final String VIDEO_STABLE = "video_stable";
+        public static final String TO_INFRARED = "to_infrared";
+        public static final String TARGET_DETECTION = "target_detection";
+        public static final String TRACK_SEQUENCE = "track_sequence";
+    }
+
     public static class SubSystem {
         public static final String  ToInfrared = "0";
         public static final String  TargetDetection = "1";

+ 1 - 0
taais-modules/taais-biz/src/main/java/com/taais/biz/domain/vo/StartToInfraredTask.java

@@ -17,6 +17,7 @@ import lombok.NoArgsConstructor;
 @AllArgsConstructor
 @NoArgsConstructor
 public class StartToInfraredTask {
+    private String bizType;
     private Long bizId;
     private String otherParams;
     private String model_path;

+ 2 - 0
taais-modules/taais-biz/src/main/java/com/taais/biz/service/impl/TargetDetectionServiceImpl.java

@@ -249,6 +249,8 @@ public class TargetDetectionServiceImpl extends BaseServiceImpl<TargetDetectionM
         startToInfraredTask.setSource_dir(targetDetection.getInputPath());
         startToInfraredTask.setResult_dir(targetDetection.getOutputPath());
 
+        startToInfraredTask.setBizType(BizConstant.BizType.TARGET_DETECTION);
+
         HttpResponseEntity responseEntity = sendPostMsg(algorithmConfigTrack.getAlgorithmAddress(), startToInfraredTask);
         if (ObjectUtil.isNotNull(responseEntity) && responseEntity.getStatus() == 200) {
             targetDetection.setStatus(BizConstant.VideoStatus.RUNNING);

+ 2 - 0
taais-modules/taais-biz/src/main/java/com/taais/biz/service/impl/ToInfraredServiceImpl.java

@@ -255,6 +255,8 @@ public class ToInfraredServiceImpl extends BaseServiceImpl<ToInfraredMapper, ToI
         startToInfraredTask.setSource_dir(toInfrared.getInputPath());
         startToInfraredTask.setResult_dir(toInfrared.getOutputPath());
 
+        startToInfraredTask.setBizType(BizConstant.BizType.TO_INFRARED);
+
         HttpResponseEntity responseEntity = sendPostMsg(algorithmConfigTrack.getAlgorithmAddress(), startToInfraredTask);
         if (ObjectUtil.isNotNull(responseEntity) && responseEntity.getStatus() == 200) {
             toInfrared.setStatus(BizConstant.VideoStatus.RUNNING);

+ 2 - 0
taais-modules/taais-biz/src/main/java/com/taais/biz/service/impl/TrackSequenceServiceImpl.java

@@ -252,6 +252,8 @@ public class TrackSequenceServiceImpl extends BaseServiceImpl<TrackSequenceMappe
         startToInfraredTask.setSource_dir(trackSequence.getInputPath());
         startToInfraredTask.setResult_dir(trackSequence.getOutputPath());
 
+        startToInfraredTask.setBizType(BizConstant.BizType.TRACK_SEQUENCE);
+
         HttpResponseEntity responseEntity = sendPostMsg(algorithmConfigTrack.getAlgorithmAddress(), startToInfraredTask);
         if (ObjectUtil.isNotNull(responseEntity) && responseEntity.getStatus() == 200) {
             trackSequence.setStatus(BizConstant.VideoStatus.RUNNING);