From 4e353f81eebf089b65a54cea553d3f5927a27aae Mon Sep 17 00:00:00 2001
From: morningman <yunyou@selectdb.com>
Date: Thu, 22 May 2025 21:28:08 +0800
Subject: [PATCH] [fix](export) remove task from map finally

---
 .../apache/doris/scheduler/manager/TransientTaskManager.java    | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/fe/fe-core/src/main/java/org/apache/doris/scheduler/manager/TransientTaskManager.java b/fe/fe-core/src/main/java/org/apache/doris/scheduler/manager/TransientTaskManager.java
index 5c62caede9c171..feaf3365a35017 100644
--- a/fe/fe-core/src/main/java/org/apache/doris/scheduler/manager/TransientTaskManager.java
+++ b/fe/fe-core/src/main/java/org/apache/doris/scheduler/manager/TransientTaskManager.java
@@ -63,7 +63,7 @@ public Long addMemoryTask(TransientTaskExecutor executor) throws JobException {
     }
 
     public void cancelMemoryTask(Long taskId) throws JobException {
-        TransientTaskExecutor transientTaskExecutor = taskExecutorMap.get(taskId);
+        TransientTaskExecutor transientTaskExecutor = taskExecutorMap.remove(taskId);
         if (transientTaskExecutor != null) {
             transientTaskExecutor.cancel();
         }