mirrors/seaweedfs
1
0
Fork 0
mirror of https://github.com/chrislusf/seaweedfs synced 2024-07-03 07:36:45 +02:00
Code Issues Projects Releases Wiki Activity

refactor

Browse source
This commit is contained in:
chrislu 2024-01-21 01:24:12 -08:00
parent a5b906c0dd
commit 9ce7c482b3
2 changed files with 12 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
weed/mq/pub_balancer/balancer.go
View file

@ -67,7 +67,12 @@ func (balancer *Balancer) RemoveBroker(broker string, stats *BrokerStats) {
if !found { if !found {
continue continue
} }
partitionSlotToBrokerList.RemoveBroker(broker) pickedBroker := pickBrokers(balancer.Brokers, 1)
if len(pickedBroker) == 0 {
partitionSlotToBrokerList.RemoveBroker(broker)
} else {
partitionSlotToBrokerList.ReplaceBroker(broker, pickedBroker[0])
}
} }
balancer.onPubRemoveBroker(broker, stats) balancer.onPubRemoveBroker(broker, stats)
balancer.OnRemoveBroker(broker, stats) balancer.OnRemoveBroker(broker, stats)

8
weed/mq/pub_balancer/partition_list_broker.go
View file

@ -44,9 +44,13 @@ func (ps *PartitionSlotToBrokerList) AddBroker(partition *mq_pb.Partition, broke
}) })
} }
func (ps *PartitionSlotToBrokerList) RemoveBroker(broker string) { func (ps *PartitionSlotToBrokerList) RemoveBroker(broker string) {
ps.ReplaceBroker(broker, "")
}
func (ps *PartitionSlotToBrokerList) ReplaceBroker(oldBroker string, newBroker string) {
for _, partitionSlot := range ps.PartitionSlots { for _, partitionSlot := range ps.PartitionSlots {
if partitionSlot.AssignedBroker == broker { if partitionSlot.AssignedBroker == oldBroker {
partitionSlot.AssignedBroker = "" partitionSlot.AssignedBroker = newBroker
} }
} }
} }