add weighted_lb feature

weibocom · Apr 3, 2024 · cae619f · cae619f
1 parent 7af8144
commit cae619f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lb/weightRoundRobinLb_test.go b/lb/weightRoundRobinLb_test.go
@@ -178,6 +178,7 @@ func TestRoundRobinSelector(t *testing.T) {
 func checkRR(t *testing.T, lb *WeightRoundRobinLB, size int, initialMaxWeight int64,
 	round int, expectMaxDelta float64, expectAvgDelta float64, unavailableSize int, url *core.URL) {
 	eps := buildTestDynamicEpsWithUnavailable(size, true, initialMaxWeight, true, unavailableSize, url)
+	rand.Seed(time.Now().UnixNano())
 	eps = core.EndpointShuffle(eps)
 	lb.OnRefresh(eps)
 	_, ok := lb.getSelector().(*roundRobinSelector)
@@ -213,6 +214,7 @@ func TestWeightRingSelector(t *testing.T) {
 func checkKWR(t *testing.T, lb *WeightRoundRobinLB, size int, initialMaxWeight int64,
 	round int, expectMaxDelta float64, expectAvgDelta float64, unavailableSize int, url *core.URL) {
 	eps := buildTestDynamicEpsWithUnavailable(size, false, initialMaxWeight, true, unavailableSize, url)
+	rand.Seed(time.Now().UnixNano())
 	eps = core.EndpointShuffle(eps)
 	lb.OnRefresh(eps)
 	_, ok := lb.getSelector().(*weightedRingSelector)
@@ -256,6 +258,7 @@ func TestSlidingWindowWeightedRoundRobinSelector(t *testing.T) {
 func checkSWWRR(t *testing.T, lb *WeightRoundRobinLB, size int, initialMaxWeight int64,
 	round int, expectMaxDelta float64, expectAvgDelta float64, unavailableSize int, url *core.URL) {
 	eps := buildTestDynamicEpsWithUnavailable(size, false, initialMaxWeight, true, unavailableSize, url)
+	rand.Seed(time.Now().UnixNano())
 	eps = core.EndpointShuffle(eps)
 	lb.OnRefresh(eps)
 	_, ok := lb.getSelector().(*slidingWindowWeightedRoundRobinSelector)