Do not override thetas if sum is less or equals to 0

Closes #4390
elastic · Dec 10, 2013 · a3c93ec · a3c93ec
1 parent dade4a2
commit a3c93ec
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.../java/org/elasticsearch/cluster/routing/allocation/allocator/BalancedShardsAllocator.java b/.../java/org/elasticsearch/cluster/routing/allocation/allocator/BalancedShardsAllocator.java
@@ -203,8 +203,9 @@ public WeightFunction(float indexBalance, float shardBalance, float primaryBalan
                         sum = indexBalance + shardBalance;
                         if (sum <= 0.0f) {
                             thetaMap.put(operation, defaultTheta);
+                        } else {
+                            thetaMap.put(operation, new float[]{shardBalance / sum, indexBalance / sum, 0});
                         }
-                        thetaMap.put(operation, new float[]{shardBalance / sum, indexBalance / sum, 0});
                         break;
                     case BALANCE:
                     case ALLOCATE: