MINOR: Use clear method to speed up removal (#18400)

Reviewers: Apoorv Mittal <apoorvmittal10@gmail.com>, Andrew Schofield <aschofield@confluent.io>
apache · Jan 6, 2025 · c116354 · c116354
1 parent 2521aee
commit c116354
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/core/src/main/java/kafka/server/share/SharePartition.java b/core/src/main/java/kafka/server/share/SharePartition.java
@@ -1755,10 +1755,7 @@ be removed once all the messages (0-99) are acknowledged (ACCEPT or REJECT).
             }
 
             if (lastKeyToRemove != -1) {
-                NavigableMap<Long, InFlightBatch> subMap = cachedState.subMap(firstKeyToRemove, true, lastKeyToRemove, true);
-                for (Long key : subMap.keySet()) {
-                    cachedState.remove(key);
-                }
+                cachedState.subMap(firstKeyToRemove, true, lastKeyToRemove, true).clear();
             }
         } finally {
             lock.writeLock().unlock();