extenders

type BalancedExtender ¶

type BalancedExtender struct {
	// contains filtered or unexported fields
}

BalancedExtender adds extra partition assignments in a "balanced" way. The current algorithm is:

for each new partition:

set the leader rack to the next rack in the cycle
choose the leader using the picker
for each follower:
  set the rack to either the same one as the leader (if inRack true) or the next one in the
    cycle (if inRack false)
  pick the follower using the picker

func NewBalancedExtender(
	brokers []admin.BrokerInfo,
	inRack bool,
	picker pickers.Picker,
) *BalancedExtender

func (b *BalancedExtender) Extend(
	topic string,
	curr []admin.PartitionAssignment,
	extraPartitions int,
) ([]admin.PartitionAssignment, error)

type Extender ¶

type Extender interface {
	Extend(
		topic string,
		currAssignments []admin.PartitionAssignment,
		newPartitions int,
	) ([]admin.PartitionAssignment, error)
}

Extender is an interface for structs that determine how to add new partitions to an existing topic.

type StaticExtender ¶

type StaticExtender struct {
	Assignments []admin.PartitionAssignment
}

StaticExtender is an Extender that ignores the current state and assigns based on the value of the Assignments field. Generally intended for testing purposes.

func (s *StaticExtender) Extend(
	topic string,
	curr []admin.PartitionAssignment,
	newPartitions int,
) ([]admin.PartitionAssignment, error)