pickers

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

type LowestIndexPicker struct{}

type Picker interface {
	// PickNew is the primary method used for assignment and extension. It chooses a new
	// replica for the given partition and index and directly modifies the argument assignments.
	PickNew(
		topic string,
		brokerChoices []int,
		curr []admin.PartitionAssignment,
		partition int,
		index int,
	) error

	// SortRemovals is a helper for choosing which replica in a set of partitions to replace.
	// Because the actual replacement logic is somewhat complex at the moment, the interface
	// is a little different than the PickNew function above. In particular, the choices are
	// a slice of partitions and these are sorted in place (without any replacement in curr).
	SortRemovals(
		topic string,
		partitionChoices []int,
		curr []admin.PartitionAssignment,
		index int,
	) error

	// ScoreBroker is a helper for generating a static "score" for a broker. It's used in
	// rebalancing and other applications where we're doing swaps as opposed to a single
	// addition or subtraction. A higher score should correspond to higher frequency, i.e.
	// more likely to be removed.
	ScoreBroker(
		topic string,
		brokerID int,
		partition int,
		index int,
	) int
}

type RandomizedPicker struct{}