Removing task Id from map using remove() (opensearch-project#15918) (opensearch-project#15920)

opensearch-trigger-bot[bot] · github-actions[bot] · web-flow · commit 008b3b98b8d5 · 2024-09-13T06:26:59.000+05:30
(cherry picked from commit 4223fab) Signed-off-by: Sagar Upadhyaya <sagar.upadhyaya.121@gmail.com> Signed-off-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com> Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
diff --git a/server/src/main/java/org/opensearch/tasks/TaskCancellationMonitoringService.java b/server/src/main/java/org/opensearch/tasks/TaskCancellationMonitoringService.java
@@ -125,7 +125,7 @@ public void onTaskCompleted(Task task) {
         if (!TASKS_TO_TRACK.contains(task.getClass())) {
             return;
         }
-        this.cancelledTaskTracker.entrySet().removeIf(entry -> entry.getKey() == task.getId());
+        this.cancelledTaskTracker.remove(task.getId());
     }
 
     /**

Original file line number	Diff line number	Diff line change
`@@ -125,7 +125,7 @@ public void onTaskCompleted(Task task) {`
`125`	`125`	`if (!TASKS_TO_TRACK.contains(task.getClass())) {`
`126`	`126`	`return;`
`127`	`127`	`}`
`128`		`- this.cancelledTaskTracker.entrySet().removeIf(entry -> entry.getKey() == task.getId());`
	`128`	`+ this.cancelledTaskTracker.remove(task.getId());`
`129`	`129`	`}`
`130`	`130`
`131`	`131`	`/**`