Removing task Id from map using remove() (#15918)

sgup432 · web-flow · commit 4223fab40bed · 2024-09-13T04:01:46.000+05:30
Signed-off-by: Sagar Upadhyaya &lt;sagar.upadhyaya.121@gmail.com&gt;
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`	`/**`