feat: 初步实现子任务查看

taais-modules/taais-biz/src/main/java/com/taais/biz/controller/AlgorithmTaskTrackController.java

@@ -106,4 +106,9 @@ public class AlgorithmTaskTrackController extends BaseController {
         }
         return CommonResult.success();
     }
+
+    @GetMapping("/listSubTask/{id}")
+    public CommonResult listSubTask(@PathVariable Long id) {
+        return algorithmTaskTrackService.listSubTask(id);
+    }
 }

taais-modules/taais-biz/src/main/java/com/taais/biz/service/IAlgorithmTaskTrackService.java

@@ -5,6 +5,7 @@ import java.util.List;
 import com.taais.biz.domain.AlgorithmTaskTrack;
 import com.taais.biz.domain.vo.AlgorithmTaskTrackVo;
 import com.taais.biz.domain.bo.AlgorithmTaskTrackBo;
+import com.taais.common.core.core.domain.CommonResult;
 import com.taais.common.orm.core.service.IBaseService;
 import com.taais.common.core.core.page.PageResult;
 
@@ -63,4 +64,5 @@ public interface IAlgorithmTaskTrackService extends IBaseService<AlgorithmTaskTr
      */
     boolean deleteByIds(Long[] ids);
 
+    CommonResult listSubTask(Long id);
 }

taais-modules/taais-biz/src/main/java/com/taais/biz/service/impl/AlgorithmTaskTrackServiceImpl.java

@@ -14,6 +14,7 @@ import com.taais.biz.mapper.AlgorithmTaskTrackMapper;
 import com.taais.biz.service.IAlgorithmTaskTrackService;
 import com.taais.common.core.config.TaaisConfig;
 import com.taais.common.core.constant.Constants;
+import com.taais.common.core.core.domain.CommonResult;
 import com.taais.common.core.core.page.PageResult;
 import com.taais.common.core.utils.MapstructUtils;
 import com.taais.common.orm.core.page.PageQuery;
@@ -283,4 +284,13 @@ public class AlgorithmTaskTrackServiceImpl extends BaseServiceImpl<AlgorithmTask
         return this.removeByIds(Arrays.asList(ids));
     }
 
+    @Override
+    public CommonResult listSubTask(Long id) {
+        List<ToInfrared> toInfraredList = toInfraredService.getByParentTaskId(id);
+        List<TrackSequence> trackSequenceList = trackSequenceService.getByParentTaskId(id);
+        Map<String, Object> result = new HashMap<>();
+        result.put("toInfraredList", toInfraredList);
+        result.put("trackSequenceList", trackSequenceList);
+        return CommonResult.success(result);
+    }
 }

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

@@ -72,6 +72,12 @@ public class TargetDetectionServiceImpl extends BaseServiceImpl<TargetDetectionM
     @Autowired
     private AlgorithmModelTrackServiceImpl algorithmModelTrackService;
 
+    public List<TargetDetection> getByParentTaskId(Long parentTaskId) {
+        QueryWrapper queryWrapper = super.buildBaseQueryWrapper();
+        queryWrapper.and(TARGET_DETECTION.PARENT_TASK_ID.eq(parentTaskId));
+        return this.list(queryWrapper);
+    }
+
     @Override
     public QueryWrapper query() {
         return super.query().from(TARGET_DETECTION);

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

@@ -81,6 +81,12 @@ public class ToInfraredServiceImpl extends BaseServiceImpl<ToInfraredMapper, ToI
     @Autowired
     private AlgorithmModelTrackServiceImpl algorithmModelTrackService;
 
+    public List<ToInfrared> getByParentTaskId(Long parentTaskId) {
+        QueryWrapper queryWrapper = super.buildBaseQueryWrapper();
+        queryWrapper.and(TO_INFRARED.PARENT_TASK_ID.eq(parentTaskId));
+        return this.list(queryWrapper);
+    }
+
     @Override
     public QueryWrapper query() {
         return super.query().from(TO_INFRARED);

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

@@ -78,6 +78,12 @@ public class TrackSequenceServiceImpl extends BaseServiceImpl<TrackSequenceMappe
     @Resource
     private TrackSequenceMapper trackSequenceMapper;
 
+    public List<TrackSequence> getByParentTaskId(Long parentTaskId) {
+        QueryWrapper queryWrapper = super.buildBaseQueryWrapper();
+        queryWrapper.and(TRACK_SEQUENCE.PARENT_TASK_ID.eq(parentTaskId));
+        return this.list(queryWrapper);
+    }
+
     public String getTrainInputPath(SysOssVo ossEntity) {
         return getUnZipDirPath(ossEntity);
     }