Use pure RMS to compute unbalance

CAFxX · Nov 14, 2017 · 917a869 · 917a869
1 parent c4fe3a5
commit 917a869
Showing 1 changed file with 1 addition and 9 deletions.
diff --git a/utils.go b/utils.go
@@ -132,25 +132,17 @@ func getBL(loads map[BrokerID]float64) []brokerLoad {
 
 func getUnbalanceBL(brokers []brokerLoad) float64 {
 	var sumBrokerLoad float64
-	var maxBrokerLoad float64
 
 	for _, broker := range brokers {
 		sumBrokerLoad += broker.Load
-		if maxBrokerLoad < broker.Load {
-			maxBrokerLoad = broker.Load
-		}
 	}
 
 	avgBrokerLoad := sumBrokerLoad / float64(len(brokers))
 
 	var brokerUnbalance float64
 	for _, broker := range brokers {
 		relBrokerLoad := broker.Load/avgBrokerLoad - 1.0
-		if relBrokerLoad > 0 {
-			brokerUnbalance += relBrokerLoad * relBrokerLoad
-		} else {
-			brokerUnbalance += relBrokerLoad * relBrokerLoad / 2
-		}
+		brokerUnbalance += relBrokerLoad * relBrokerLoad
 	}
 
 	return brokerUnbalance