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

Signed-off-by: Sagar Upadhyaya <[email protected]>
opensearch-project · Sep 12, 2024 · 4223fab · 4223fab
1 parent 330b249
commit 4223fab
Showing 1 changed file with 1 addition and 1 deletion.
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());
     }
 
     /**