Update IndicesRequestCache.java

Signed-off-by: Kiran Prakash <awskiran@amazon.com>
opensearch-project · Jun 17, 2024 · 8a949d4 · 8a949d4
1 parent 5a56ec1
commit 8a949d4
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/server/src/main/java/org/opensearch/indices/IndicesRequestCache.java b/server/src/main/java/org/opensearch/indices/IndicesRequestCache.java
@@ -573,14 +573,8 @@ private void updateStaleCountOnCacheInsert(CleanupKey cleanupKey) {
 
         // pkg-private for testing
         void addToCleanupKeyToCountMap(ShardId shardId, String readerCacheKeyId) {
-            cleanupKeyToCountMap.compute(shardId, (currentShardId, readerCacheKeyMap) -> {
-                final ConcurrentHashMap<String, Integer> updatedReaderCacheKeyMap = Objects.requireNonNullElseGet(
-                    readerCacheKeyMap,
-                    ConcurrentHashMap::new
-                );
-                updatedReaderCacheKeyMap.compute(readerCacheKeyId, (currentReaderCacheKeyId, count) -> (count == null) ? 1 : count + 1);
-                return updatedReaderCacheKeyMap;
-            });
+            cleanupKeyToCountMap.computeIfAbsent(shardId, k -> new ConcurrentHashMap<>())
+                .merge(readerCacheKeyId, 1, Integer::sum);
         }
 
         /**