Refactor code for AwarenessAllocationDecider

Signed-off-by: Anshu Agarwal <anshukag@amazon.com>
opensearch-project · Jun 14, 2022 · 3d5bff2 · 3d5bff2
1 parent 1705af0
commit 3d5bff2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...in/java/org/opensearch/cluster/routing/allocation/decider/AwarenessAllocationDecider.java b/...in/java/org/opensearch/cluster/routing/allocation/decider/AwarenessAllocationDecider.java
@@ -210,10 +210,10 @@ private Decision underCapacity(ShardRouting shardRouting, RoutingNode node, Rout
 
             int numberOfAttributes = nodesPerAttribute.size();
             List<String> fullValues = forcedAwarenessAttributes.get(awarenessAttribute);
-            Set<String> attributesSet = new HashSet<>();
+
             if (fullValues != null) {
                 // If forced awareness is enabled, numberOfAttributes = count(distinct((union(discovered_attributes, forced_attributes)))
-                attributesSet.addAll(forcedAwarenessAttributes.get(awarenessAttribute));
+                Set<String> attributesSet = new HashSet<>(fullValues);
                 for (ObjectCursor<String> stringObjectCursor : nodesPerAttribute.keys()) {
                     attributesSet.add(stringObjectCursor.value);
                 }