assignments

Documentation

Overview

Package assignments implements helper functions for partition assignment operations.

Index

• func AddPartitions(assignments [][]int32, targetPartitions int, targetRepFactor int, ...) [][]int32
• func AlterReplicationFactor(assignments [][]int32, targetRepFactor int, clusterReplicaCounts map[int32]int, ...) [][]int32
• func Copy(assignments [][]int32) [][]int32
• func Rebalance(assignments [][]int32, clusterReplicaCounts map[int32]int, brokers []int32) [][]int32
• func RebalanceWithRackConstraints(assignments [][]int32, rackConstraints [][]string, ...) [][]int32
• func SyncRackConstraints(assignments [][]int32, rackConstraints [][]string, ...) [][]int32

Functions

func AddPartitions

func AddPartitions(
	assignments [][]int32,
	targetPartitions int,
	targetRepFactor int,
	clusterReplicaCounts map[int32]int,
	brokers []int32,
) [][]int32

AddPartitions adds partitions and returns assignments for the new partitions.

func AlterReplicationFactor

func AlterReplicationFactor(
	assignments [][]int32,
	targetRepFactor int,
	clusterReplicaCounts map[int32]int,
	brokers []int32,
) [][]int32

AlterReplicationFactor alters the replication factor and returns the new assignments.

func Copy

func Copy(assignments [][]int32) [][]int32

Copy makes a copy of partition assignments.

func Rebalance

func Rebalance(
	assignments [][]int32,
	clusterReplicaCounts map[int32]int,
	brokers []int32,
) [][]int32

func RebalanceWithRackConstraints

func RebalanceWithRackConstraints(
	assignments [][]int32,
	rackConstraints [][]string,
	clusterReplicaCounts map[int32]int,
	brokersByRack map[string][]int32,
) [][]int32

func SyncRackConstraints

func SyncRackConstraints(
	assignments [][]int32,
	rackConstraints [][]string,
	brokersByRack map[string][]int32,
	clusterReplicaCounts map[int32]int,
) [][]int32

SyncRackConstraints checks partition assignments are in sync with rack constraints, updating if necessary.