#888 fix WRR selector

smallnest · Jan 28, 2025 · 0b3b436 · 0b3b436
1 parent da8106c
commit 0b3b436
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/client/selector.go b/client/selector.go
@@ -136,10 +136,10 @@ func (s *weightedRoundRobinSelector) Select(ctx context.Context, servicePath, se
 }
 
 func (s *weightedRoundRobinSelector) UpdateServer(servers map[string]string) {
-	ss := createWeighted(servers)
-	s.buildRing()
-	s.servers = ss
+	newServer := newWeightedRoundRobinSelector(servers).(*weightedRoundRobinSelector)
+	*s = *newServer
 }
+
 func (s *weightedRoundRobinSelector) buildRing() {
 	s.totalWeight = 0
 	for _, w := range s.servers {