mirror of
https://github.com/chrislusf/seaweedfs
synced 2024-05-19 18:01:03 +02:00
57 lines
1.6 KiB
Go
57 lines
1.6 KiB
Go
package pub_balancer
|
|
|
|
import (
|
|
"github.com/seaweedfs/seaweedfs/weed/glog"
|
|
"github.com/seaweedfs/seaweedfs/weed/pb/mq_pb"
|
|
)
|
|
|
|
type PartitionSlotToBroker struct {
|
|
RangeStart int32
|
|
RangeStop int32
|
|
UnixTimeNs int64
|
|
AssignedBroker string
|
|
}
|
|
|
|
type PartitionSlotToBrokerList struct {
|
|
PartitionSlots []*PartitionSlotToBroker
|
|
RingSize int32
|
|
}
|
|
|
|
func NewPartitionSlotToBrokerList(ringSize int32) *PartitionSlotToBrokerList {
|
|
return &PartitionSlotToBrokerList{
|
|
RingSize: ringSize,
|
|
}
|
|
}
|
|
|
|
func (ps *PartitionSlotToBrokerList) AddBroker(partition *mq_pb.Partition, broker string) {
|
|
for _, partitionSlot := range ps.PartitionSlots {
|
|
if partitionSlot.RangeStart == partition.RangeStart && partitionSlot.RangeStop == partition.RangeStop {
|
|
if partitionSlot.AssignedBroker == broker {
|
|
return
|
|
}
|
|
if partitionSlot.AssignedBroker != "" {
|
|
glog.V(0).Infof("partition %s broker change: %s => %s", partition, partitionSlot.AssignedBroker, broker)
|
|
}
|
|
partitionSlot.AssignedBroker = broker
|
|
return
|
|
}
|
|
}
|
|
ps.PartitionSlots = append(ps.PartitionSlots, &PartitionSlotToBroker{
|
|
RangeStart: partition.RangeStart,
|
|
RangeStop: partition.RangeStop,
|
|
UnixTimeNs: partition.UnixTimeNs,
|
|
AssignedBroker: broker,
|
|
})
|
|
}
|
|
func (ps *PartitionSlotToBrokerList) RemoveBroker(broker string) {
|
|
ps.ReplaceBroker(broker, "")
|
|
}
|
|
|
|
func (ps *PartitionSlotToBrokerList) ReplaceBroker(oldBroker string, newBroker string) {
|
|
for _, partitionSlot := range ps.PartitionSlots {
|
|
if partitionSlot.AssignedBroker == oldBroker {
|
|
partitionSlot.AssignedBroker = newBroker
|
|
}
|
|
}
|
|
}
|