public class ConfluentDataBalanceEngineContext extends Object implements DataBalanceEngineContext
Constructor and Description |
---|
ConfluentDataBalanceEngineContext(DataBalancerMetricsRegistry dataBalancerMetricsRegistry,
KafkaCruiseControl cruiseControl,
org.apache.kafka.common.utils.Time time,
Boolean incrementalBalancingEnabled) |
public ConfluentDataBalanceEngineContext(DataBalancerMetricsRegistry dataBalancerMetricsRegistry, KafkaCruiseControl cruiseControl, org.apache.kafka.common.utils.Time time, Boolean incrementalBalancingEnabled)
public Optional<KafkaCruiseControl> getCruiseControl()
public boolean isCruiseControlInitialized()
public void setCruiseControl(KafkaCruiseControl cruiseControl)
public org.apache.kafka.common.utils.Time getTime()
public DataBalancerMetricsRegistry getDataBalancerMetricsRegistry()
public ApiStatePersistenceStore getPersistenceStore()
DataBalanceEngineContext
getPersistenceStore
in interface DataBalanceEngineContext
public Map<ImmutableSet<Integer>,BrokerRemovalStateTracker> getBrokerRemovalsStateTrackers()
BrokerRemovalStateTracker
for the active broker removal operationsgetBrokerRemovalsStateTrackers
in interface DataBalanceEngineContext
public BalancerStatusTracker getBalancerStatusTracker()
getBalancerStatusTracker
in interface DataBalanceEngineContext
public void setBalancerStatusTracker(BalancerStatusTracker balancerStatusTracker)
public boolean isIncrementalBalancingEnabled()
public boolean shouldAutoHeal()
public void v2AdditionEnabled(boolean v2AdditionEnabled)
public void shouldAutoHeal(boolean shouldAutoHeal)
public EvenClusterLoadStateManager getEvenClusterLoadStateManager()
getEvenClusterLoadStateManager
in interface DataBalanceEngineContext
public BrokerAdditionContextContainer additionContext()
additionContext
in interface DataBalanceEngineContext
BrokerAdditionContextContainer
.public void init(ApiStatePersistenceStore persistenceStore)
public void putBrokerRemovalFuture(ImmutableSet<Integer> brokerIds, BrokerRemovalFuture future)
brokerIds
- the ids of the brokers being removedfuture
- a #BrokerRemovalFuture
for the full broker removal operation of broker #brokerId
public void removeBrokerRemovalFuture(ImmutableSet<Integer> brokerIds)
public BrokerRemovalFuture brokerRemovalFuture(ImmutableSet<Integer> brokerIds)
Future
for the full broker removal operation for brokers #brokerIds
-
the plan computation/shutdown operation and then the underlying reassignments executionpublic void closeAndClearState()
public BrokerAdditionContextContainer brokerAdditionContextContainer()