Merge pull request #1884 from Codeprh/OptimizedLoadBalancingAlgorithm

feat: Optimized load balancing algorithm
apache · May 13, 2022 · a463b2a · a463b2a
2 parents c7bab46 + 6e2c290
commit a463b2a
Showing 1 changed file with 16 additions and 13 deletions.
diff --git a/cluster/loadbalance/leastactive/loadbalance.go b/cluster/loadbalance/leastactive/loadbalance.go
@@ -57,34 +57,37 @@ func (lb *leastActiveLoadBalance) Select(invokers []protocol.Invoker, invocation
 	}
 
 	var (
-		leastActive  int32                = -1 // The least active value of all invokers
-		totalWeight  int64                     // The number of invokers having the same least active value (LEAST_ACTIVE)
-		firstWeight  int64                     // Initial value, used for comparison
-		leastCount   int                       // The number of invokers having the same least active value (LEAST_ACTIVE)
-		leastIndexes = make([]int, count)      // The index of invokers having the same least active value (LEAST_ACTIVE)
-		sameWeight   = true                    // Every invoker has the same weight value?
+		leastActive  int32                  = -1 // The least active value of all invokers
+		totalWeight  int64                       // The number of invokers having the same least active value (LEAST_ACTIVE)
+		firstWeight  int64                       // Initial value, used for comparison
+		leastCount   int                         // The number of invokers having the same least active value (LEAST_ACTIVE)
+		leastIndexes = make([]int, count)        // The index of invokers having the same least active value (LEAST_ACTIVE)
+		sameWeight   = true                      // Every invoker has the same weight value?
+		weights      = make([]int64, count)      // The weight of every invokers
 	)
 
 	for i := 0; i < count; i++ {
 		invoker := invokers[i]
 		// Active number
 		active := protocol.GetMethodStatus(invoker.GetURL(), invocation.MethodName()).GetActive()
 		// current weight (maybe in warmUp)
-		weight := loadbalance.GetWeight(invoker, invocation)
+		afterWarmup := loadbalance.GetWeight(invoker, invocation)
+		// save for later use
+		weights[i] = afterWarmup
 		// There are smaller active services
 		if leastActive == -1 || active < leastActive {
 			leastActive = active
 			leastIndexes[0] = i
 			leastCount = 1 // next available leastIndex offset
-			totalWeight = weight
-			firstWeight = weight
+			totalWeight = afterWarmup
+			firstWeight = afterWarmup
 			sameWeight = true
 		} else if active == leastActive {
 			leastIndexes[leastCount] = i
-			totalWeight += weight
+			totalWeight += afterWarmup
 			leastCount++
 
-			if sameWeight && (i > 0) && weight != firstWeight {
+			if sameWeight && afterWarmup != firstWeight {
 				sameWeight = false
 			}
 		}
@@ -98,8 +101,8 @@ func (lb *leastActiveLoadBalance) Select(invokers []protocol.Invoker, invocation
 		offsetWeight := rand.Int63n(totalWeight) + 1
 		for i := 0; i < leastCount; i++ {
 			leastIndex := leastIndexes[i]
-			offsetWeight -= loadbalance.GetWeight(invokers[i], invocation)
-			if offsetWeight <= 0 {
+			offsetWeight -= weights[leastIndex]
+			if offsetWeight < 0 {
 				return invokers[leastIndex]
 			}
 		}