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

Class TaskPartitionEqualSharingRebalancer

java.lang.Object

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

All Implemented Interfaces:

TaskPartitionRebalancer

public class TaskPartitionEqualSharingRebalancer
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 distributes partitions equally between tasks.

Field Summary

Fields

Modifier and Type	Field	Description
private static final org.slf4j.Logger	LOGGER

Constructor Summary

Constructors

Constructor	Description
TaskPartitionEqualSharingRebalancer()

Method Summary

Modifier and Type	Method	Description
private Set<String>	collectOwnedPartitionTokens(Collection<TaskState>... taskStates)
private Set<String>	collectPartitionTokens(Collection<TaskState>... taskStates)
private TaskState	distributePartitionsFromObsoleteTasks(TaskState leaderTaskState, Map<String,TaskState> survivedTasks, Map<String,TaskState> obsoleteTasks)
private Map<String,TaskState>	excludeLeader(String leaderTaskUid, Map<String,TaskState> tasks)
private List<PartitionState>	filterDuplications(List<PartitionState> partitionStates)
private String	findCandidateToSharePartition(TaskState leaderTaskState, Map<String,TaskState> survivedTasks)
private TaskState	moveFinishedPartitionsFromObsoleteTasks(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)
private TaskState	takeSharedPartitionsToObsoleteTask(TaskState leaderTaskState, Map<String,TaskState> survivedTasks)

Methods inherited from class java.lang.Object

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

Field Details

LOGGER

private static final org.slf4j.Logger LOGGER

Constructor Details

TaskPartitionEqualSharingRebalancer

public TaskPartitionEqualSharingRebalancer()

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)

moveFinishedPartitionsFromObsoleteTasks

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

distributePartitionsFromObsoleteTasks

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

findCandidateToSharePartition

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

moveSharedPartitionsFromObsoleteTasks

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

takeSharedPartitionsFromSurvivedTasks

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

takeSharedPartitionsToObsoleteTask

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

filterDuplications

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

collectPartitionTokens

private Set<String> collectPartitionTokens(Collection<TaskState>... taskStates)

collectOwnedPartitionTokens

private Set<String> collectOwnedPartitionTokens(Collection<TaskState>... taskStates)