diff options
-rw-r--r-- | documentapi/src/main/java/com/yahoo/documentapi/messagebus/protocol/LoadBalancer.java | 7 |
1 files changed, 1 insertions, 6 deletions
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)); } /** |