From 7b3ec5bd89f4017b08272d419d155850da9b3785 Mon Sep 17 00:00:00 2001
From: Like habits <30977529+likehabits@users.noreply.github.com>
Date: Sat, 7 Oct 2023 14:54:41 +0800
Subject: [PATCH] update task (#194)

---
 .../main/java/com/primihub/biz/service/data/DataTaskService.java | 1 +
 1 file changed, 1 insertion(+)

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
index f9973464a..3ae5a1310 100644
--- 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);