From 1ff3df2a091e99540d0bd7d2619d571d6d7e3ed8 Mon Sep 17 00:00:00 2001
From: Henning Baldersheim <balder@yahoo-inc.com>
Date: Wed, 13 May 2020 10:32:45 +0000
Subject: Use computeIfAbsent

---
 .../com/yahoo/documentapi/messagebus/protocol/LoadBalancer.java    | 7 +------
 1 file changed, 1 insertion(+), 6 deletions(-)

(limited to 'documentapi/src')

diff --git a/documentapi/src/main/java/com/yahoo/documentapi/messagebus/protocol/LoadBalancer.java b/documentapi/src/main/java/com/yahoo/documentapi/messagebus/protocol/LoadBalancer.java
index 6d08dc2cee3..a9b0632a767 100644
--- a/documentapi/src/main/java/com/yahoo/documentapi/messagebus/protocol/LoadBalancer.java
+++ b/documentapi/src/main/java/com/yahoo/documentapi/messagebus/protocol/LoadBalancer.java
@@ -56,12 +56,7 @@ public class LoadBalancer {
         }
     }
     private int getCachedIndex(String nodeName) {
-        Integer index = cachedIndex.get(nodeName);
-        if (index == null) {
-            index = getIndex(nodeName);
-            cachedIndex.put(nodeName, index);
-        }
-        return index;
+        return cachedIndex.computeIfAbsent(nodeName, key -> getIndex(key));
     }
 
     /**
-- 
cgit v1.2.3