Browse Source

fix: 优化插入

WANGKANG 8 tháng trước cách đây
mục cha
commit
8c215ca8d7

+ 2 - 6
taais-modules/taais-biz/src/main/java/com/taais/biz/controller/TrackSequenceController.java

@@ -72,12 +72,8 @@ public class TrackSequenceController extends BaseController {
     @Log(title = "注视轨迹序列", businessType = BusinessType.INSERT)
     @RepeatSubmit()
     @PostMapping
-    public CommonResult<Void> add(@Validated @RequestBody TrackSequenceBo trackSequenceBo) {
-        boolean inserted = trackSequenceService.insert(trackSequenceBo);
-        if (!inserted) {
-            return CommonResult.fail("新增注视轨迹序列记录失败!");
-        }
-        return CommonResult.success();
+    public CommonResult add(@Validated @RequestBody TrackSequenceBo trackSequenceBo) {
+        return trackSequenceService.insert(trackSequenceBo);
     }
 
     /**

+ 1 - 1
taais-modules/taais-biz/src/main/java/com/taais/biz/service/ITrackSequenceService.java

@@ -49,7 +49,7 @@ public interface ITrackSequenceService extends IBaseService<TrackSequence> {
      * @param trackSequenceBo 注视轨迹序列Bo
      * @return 结果:true 操作成功,false 操作失败
      */
-    boolean insert(TrackSequenceBo trackSequenceBo);
+    CommonResult insert(TrackSequenceBo trackSequenceBo);
 
     /**
      * 修改注视轨迹序列

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

@@ -249,7 +249,7 @@ public class TargetDetectionServiceImpl extends BaseServiceImpl<TargetDetectionM
 
         Path path = Paths.get(resourcePath);
         Path inputPath = path.resolveSibling(fileName_without_suffix + BizConstant.UNZIP_SUFFIX);
-        Path outputPath = path.resolveSibling(entity.getId().toString() + BizConstant.TO_INFRARED_SUFFIX);
+        Path outputPath = path.resolveSibling(entity.getId().toString() + BizConstant.TARGET_DETECTION_SUFFIX);
 
 //        makeDir(inputPath.toString());
         makeDir(outputPath.toString());

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

@@ -163,15 +163,20 @@ public class TrackSequenceServiceImpl extends BaseServiceImpl<TrackSequenceMappe
      * @return 结果:true 操作成功,false 操作失败
      */
     @Override
-    public boolean insert(TrackSequenceBo trackSequenceBo) {
+    public CommonResult insert(TrackSequenceBo trackSequenceBo) {
+        // 检查任务名称
+        if(ObjectUtil.isEmpty(trackSequenceBo.getName()) || (!trackSequenceBo.getName().startsWith("MASC") && !trackSequenceBo.getName().startsWith("CAT"))){
+            return CommonResult.fail("任务命名错误,需以MASC或CAT开头!");
+        }
+
         // 检查input_oss_id是否存在
         if (ObjectUtil.isNull(trackSequenceBo.getInputOssId())) {
-            return false;
+            return  CommonResult.fail("请上传模型");
         }
 
         SysOssVo ossEntity = ossService.getById(trackSequenceBo.getInputOssId());
         if (ObjectUtil.isNull(ossEntity)) {
-            return false;
+            return  CommonResult.fail("找不到指定模型!");
         }
 
         TrackSequence trackSequence = new TrackSequence();
@@ -199,7 +204,13 @@ public class TrackSequenceServiceImpl extends BaseServiceImpl<TrackSequenceMappe
 
         trackSequence.setAlgorithmModelId(trackSequenceBo.getAlgorithmModelId());
 
-        return this.save(trackSequence);// 使用全局配置的雪花算法主键生成器生成ID值
+        boolean __  = this.save(trackSequence);// 使用全局配置的雪花算法主键生成器生成ID值
+        if(__) {
+            return CommonResult.success();
+        }
+        else {
+            return CommonResult.fail();
+        }
     }
 
     /**