[fix](export) remove task from map finally(#51171) (#51206)

morningman · web-flow · commit 857a0eb4b412 · 2025-05-26T10:26:50.000+08:00
bp #51171
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
@@ -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();
         }

Original file line number	Diff line number	Diff line change
`@@ -63,7 +63,7 @@ public Long addMemoryTask(TransientTaskExecutor executor) throws JobException {`
`63`	`63`	`}`
`64`	`64`
`65`	`65`	`public void cancelMemoryTask(Long taskId) throws JobException {`
`66`		`- TransientTaskExecutor transientTaskExecutor = taskExecutorMap.get(taskId);`
	`66`	`+ TransientTaskExecutor transientTaskExecutor = taskExecutorMap.remove(taskId);`
`67`	`67`	`if (transientTaskExecutor != null) {`
`68`	`68`	`transientTaskExecutor.cancel();`
`69`	`69`	`}`