Package io.debezium.connector.spanner.task.leader.rebalancer

Class TaskPartitionGreedyLeaderRebalancer

java.lang.Object

io.debezium.connector.spanner.task.leader.rebalancer.TaskPartitionGreedyLeaderRebalancer

All Implemented Interfaces:

TaskPartitionRebalancer

public class TaskPartitionGreedyLeaderRebalancer
extends Object
implements TaskPartitionRebalancer

This task contains the functionality to rebalance change stream partitions from obsolete tasks to survived tasks after a rebalance event. Leader takes everything.

Constructor Summary

Constructors

Constructor	Description
TaskPartitionGreedyLeaderRebalancer()

Method Summary

Modifier and Type	Method	Description
private Set<String>	collectPartitionTokens(TaskState taskState)
private Map<String,TaskState>	excludeLeader(String leaderTaskUid, Map<String,TaskState> tasks)
private List<PartitionState>	filterDuplications(List<PartitionState> partitionStates)
private TaskState	moveFinishedPartitionsFromObsoleteTasks(TaskState leaderTaskState, Map<String,TaskState> obsoleteTasks)
private TaskState	movePartitionsFromObsoleteTasks(TaskState leaderTaskState, Map<String,TaskState> obsoleteTasks)
private TaskState	moveSharedPartitionsFromObsoleteTasks(TaskState leaderTaskState, Map<String,TaskState> survivedTasks, Map<String,TaskState> obsoleteTasks)
TaskState	rebalance(TaskState leaderTaskState, Map<String,TaskState> survivedTasks, Map<String,TaskState> obsoleteTaskStates)
private TaskState	takeSharedPartitionsFromSurvivedTasks(TaskState leaderTaskState, Map<String,TaskState> survivedTasks)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

TaskPartitionGreedyLeaderRebalancer

public TaskPartitionGreedyLeaderRebalancer()

Method Details

rebalance

public TaskState rebalance(TaskState leaderTaskState,
 Map<String,TaskState> survivedTasks,
 Map<String,TaskState> obsoleteTaskStates)

Specified by:

rebalance in interface TaskPartitionRebalancer

excludeLeader

private Map<String,TaskState> excludeLeader(String leaderTaskUid,
 Map<String,TaskState> tasks)

movePartitionsFromObsoleteTasks

private TaskState movePartitionsFromObsoleteTasks(TaskState leaderTaskState,
 Map<String,TaskState> obsoleteTasks)

moveSharedPartitionsFromObsoleteTasks

private TaskState moveSharedPartitionsFromObsoleteTasks(TaskState leaderTaskState,
 Map<String,TaskState> survivedTasks,
 Map<String,TaskState> obsoleteTasks)

takeSharedPartitionsFromSurvivedTasks

private TaskState takeSharedPartitionsFromSurvivedTasks(TaskState leaderTaskState,
 Map<String,TaskState> survivedTasks)

moveFinishedPartitionsFromObsoleteTasks

private TaskState moveFinishedPartitionsFromObsoleteTasks(TaskState leaderTaskState,
 Map<String,TaskState> obsoleteTasks)

filterDuplications

private List<PartitionState> filterDuplications(List<PartitionState> partitionStates)

collectPartitionTokens

private Set<String> collectPartitionTokens(TaskState taskState)