TyleoDvDelaware3217/GoDoxy
1
0
Fork 0
mirror of https://github.com/yusing/godoxy.git synced 2025-05-20 12:42:34 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix loadbalancer panic on weight rebalance

Browse source
This commit is contained in:
yusing 2025-01-02 11:36:11 +08:00
parent cd74b76483
commit 320e29ba84
1 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
internal/net/http/loadbalancer/loadbalancer.go
View file

@ -65,7 +65,6 @@ func (lb *LoadBalancer) Start(parent task.Parent) E.Error {
lb.impl.OnRemoveServer(v) lb.impl.OnRemoveServer(v)
}) })
} }
lb.pool.Clear()
}) })
return nil return nil
} }
@ -171,12 +170,14 @@ func (lb *LoadBalancer) rebalance() {
if lb.sumWeight == maxWeight { if lb.sumWeight == maxWeight {
return return
} }
if lb.pool.Size() == 0 {
poolSize := lb.pool.Size()
if poolSize == 0 {
return return
} }
if lb.sumWeight == 0 { // distribute evenly if lb.sumWeight == 0 { // distribute evenly
weightEach := maxWeight / Weight(lb.pool.Size()) weightEach := maxWeight / Weight(poolSize)
remainder := maxWeight % Weight(lb.pool.Size()) remainder := maxWeight % Weight(poolSize)
lb.pool.RangeAll(func(_ string, s *Server) { lb.pool.RangeAll(func(_ string, s *Server) {
s.Weight = weightEach s.Weight = weightEach
lb.sumWeight += weightEach lb.sumWeight += weightEach