diff options
Diffstat (limited to 'weed/mq/pub_balancer')
| -rw-r--r-- | weed/mq/pub_balancer/balancer.go | 3 | ||||
| -rw-r--r-- | weed/mq/pub_balancer/partition_list_broker.go | 10 |
2 files changed, 7 insertions, 6 deletions
diff --git a/weed/mq/pub_balancer/balancer.go b/weed/mq/pub_balancer/balancer.go index 988b971af..5e8c8275e 100644 --- a/weed/mq/pub_balancer/balancer.go +++ b/weed/mq/pub_balancer/balancer.go @@ -32,9 +32,10 @@ type Balancer struct { // Collected from all brokers when they connect to the broker leader TopicToBrokers cmap.ConcurrentMap[string, *PartitionSlotToBrokerList] // key: topic name } + func NewBalancer() *Balancer { return &Balancer{ - Brokers: cmap.New[*BrokerStats](), + Brokers: cmap.New[*BrokerStats](), TopicToBrokers: cmap.New[*PartitionSlotToBrokerList](), } } diff --git a/weed/mq/pub_balancer/partition_list_broker.go b/weed/mq/pub_balancer/partition_list_broker.go index 7ceb2a9fc..9dc6140b3 100644 --- a/weed/mq/pub_balancer/partition_list_broker.go +++ b/weed/mq/pub_balancer/partition_list_broker.go @@ -6,8 +6,8 @@ import ( ) type PartitionSlotToBroker struct { - RangeStart int32 - RangeStop int32 + RangeStart int32 + RangeStop int32 AssignedBroker string } @@ -36,12 +36,12 @@ func (ps *PartitionSlotToBrokerList) AddBroker(partition *mq_pb.Partition, broke } } ps.PartitionSlots = append(ps.PartitionSlots, &PartitionSlotToBroker{ - RangeStart: partition.RangeStart, - RangeStop: partition.RangeStop, + RangeStart: partition.RangeStart, + RangeStop: partition.RangeStop, AssignedBroker: broker, }) } -func (ps *PartitionSlotToBrokerList) RemoveBroker(broker string) { +func (ps *PartitionSlotToBrokerList) RemoveBroker(broker string) { for _, partitionSlot := range ps.PartitionSlots { if partitionSlot.AssignedBroker == broker { partitionSlot.AssignedBroker = "" |