新增数据标记接口。

application-webadmin/src/main/java/com/yice/webadmin/app/controller/DatasetDataController.java

@@ -68,6 +68,35 @@ public class DatasetDataController {
         return ResponseResult.success();
     }
 
+    /**
+     * 更新数据集版本数据。
+     *
+     * @param datasetMongoDto 更新对象。
+     * @return 应答结果对象。
+     */
+    @OperationLog(type = SysOperationLogType.UPDATE)
+    @PostMapping("/add")
+    public ResponseResult<Void> add(@MyRequestBody DatasetDataDto datasetMongoDto) throws Exception{
+        String errorMessage = MyCommonUtil.getModelValidationError(datasetMongoDto, true);
+        if (errorMessage != null) {
+            return ResponseResult.error(ErrorCodeEnum.DATA_VALIDATED_FAILED, errorMessage);
+        }
+        ObjectMapper objectMapper = new ObjectMapper();
+        JsonNode rootNode = objectMapper.readTree(datasetMongoDto.getData());
+        String output = rootNode.get(DatasetConstant.OUTPUT).textValue();
+        DatasetData datasetData = MyModelUtil.copyTo(datasetMongoDto, DatasetData.class);
+        datasetData.setMarkStatus(StringUtils.isNotBlank(output) ? DatasetConstant.MARK : DatasetConstant.UNMARK);
+        datasetDataService.update(datasetData);
+
+        DatasetVersion datasetVersion = datasetVersionService.getById(datasetMongoDto.getVersionId());
+        if(null != datasetVersion) {
+            datasetVersion.setUpdateTime(new Date());
+            datasetVersion.setDataVolume(datasetVersion.getDataVolume() + 1);
+            datasetVersionService.updateById(datasetVersion);
+        }
+        return ResponseResult.success();
+    }
+
 
     /**
      * 删除数据集版本数据。