diff --git a/llap-client/src/java/org/apache/hadoop/hive/registry/impl/ZkRegistryBase.java b/llap-client/src/java/org/apache/hadoop/hive/registry/impl/ZkRegistryBase.java index d28fd1778c..d52b3a69c7 100644 --- a/llap-client/src/java/org/apache/hadoop/hive/registry/impl/ZkRegistryBase.java +++ b/llap-client/src/java/org/apache/hadoop/hive/registry/impl/ZkRegistryBase.java @@ -446,9 +446,9 @@ private void putInNodeCache(String key, Map> cache, Set instanceSet = cache.get(key); if (instanceSet == null) { instanceSet = new HashSet<>(); - instanceSet.add(instance); + cache.put(key, instanceSet); } - cache.put(key, instanceSet); + instanceSet.add(instance); } protected final void populateCache(PathChildrenCache instancesCache, boolean doInvokeListeners) {