update task (#194)

primihub · Oct 7, 2023 · 7b3ec5b · 7b3ec5b
1 parent fabdd98
commit 7b3ec5b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/primihub-service/biz/src/main/java/com/primihub/biz/service/data/DataTaskService.java b/primihub-service/biz/src/main/java/com/primihub/biz/service/data/DataTaskService.java
@@ -293,6 +293,7 @@ public BaseResultEntity cancelTask(Long taskId) {
         TaskParam taskParam = dataAsyncService.getTaskHelper().killTask(rawDataTask.getTaskIdName());
         if (taskParam.getSuccess()){
             rawDataTask.setTaskState(TaskStateEnum.CANCEL.getStateType());
+            rawDataTask.setTaskEndTime(System.currentTimeMillis());
             dataTaskPrRepository.updateDataTask(rawDataTask);
             if (rawDataTask.getTaskType().equals(TaskTypeEnum.PSI.getTaskType())){
                 DataPsiTask task = dataPsiRepository.selectPsiTaskById(taskId);