add: 数据扩增代码修改

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

@@ -10,6 +10,7 @@ import com.mybatisflex.core.paginate.Page;
 import com.mybatisflex.core.query.QueryWrapper;
 import com.taais.biz.constant.BizConstant;
 import com.taais.biz.domain.Data;
+import com.taais.biz.domain.DataAmplificationTask;
 import com.taais.biz.domain.bo.DataAmplificationTaskBo;
 import com.taais.biz.domain.bo.DataBo;
 import com.taais.biz.domain.vo.BatchDataResult;
@@ -308,16 +309,15 @@ public class DataServiceImpl extends BaseServiceImpl<DataMapper, Data> implement
     @Override
     @Transactional
     public CommonResult<Boolean> dataAmplify(DataAmplifyDto dataAmplifyDto) {
-        DataAmplifyDto amplifyDto = new DataAmplifyDto();
         DataAmplificationTaskBo dataAmplificationTaskBo = new DataAmplificationTaskBo();
-        dataAmplificationTaskBo.setName(amplifyDto.getTaskName());
+        dataAmplificationTaskBo.setName(dataAmplifyDto.getTaskName());
         dataAmplificationTaskBo.setStatus(BizConstant.TASK_STATUS_PENDING);
-        dataAmplificationTaskBo.setDataBatchNums(amplifyDto.getBatchNum());
-        dataAmplificationTaskBo.setAugmentationType(amplifyDto.getAugmentationType());
-        dataAmplificationTaskBo.setParameters(JsonUtils.toJsonString(amplifyDto.getOtherParams()));
+        dataAmplificationTaskBo.setDataBatchNums(dataAmplifyDto.getBatchNum());
+        dataAmplificationTaskBo.setAugmentationType(dataAmplifyDto.getAugmentationType());
+        dataAmplificationTaskBo.setParameters(JsonUtils.toJsonString(dataAmplifyDto.getOtherParams()));
         dataAmplificationTaskBo.setDelFlag(0);
-        dataAmplificationTaskService.insert(dataAmplificationTaskBo);
-        return this.amplifyForData(dataAmplificationTaskBo.getId().toString());
+        DataAmplificationTask insertedTask = dataAmplificationTaskService.insertTask(dataAmplificationTaskBo);
+        return this.amplifyForData(insertedTask.getId().toString());
     }
 
     private void initFileInfo(String dest, List<File> extractedImagesFileList, boolean directory, String fileName) {
@@ -474,8 +474,6 @@ public class DataServiceImpl extends BaseServiceImpl<DataMapper, Data> implement
             for (Map<String, String> param : dataAmplifyDto.getOtherParams()) {
                 otherParams.put(param.get("agName"), param.get("defaultValue"));
             }
-            Map<String, String> param = dataAmplifyDto.getOtherParams()
-                .stream().collect(Collectors.toMap(item -> item.get("agName"), item -> item.get("defaultValue")));
             dataListInfo.forEach(dataInfo -> {
                 try {
                     boolean success = true;