public class ConfluentDataBalanceEngineContext extends Object implements DataBalanceEngineContext
Constructor and Description |
---|
ConfluentDataBalanceEngineContext(DataBalancerMetricsRegistry dataBalancerMetricsRegistry,
KafkaCruiseControl cruiseControl,
org.apache.kafka.common.utils.Time time) |
public ConfluentDataBalanceEngineContext(DataBalancerMetricsRegistry dataBalancerMetricsRegistry, KafkaCruiseControl cruiseControl, org.apache.kafka.common.utils.Time time)
public 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 Map<Integer,BrokerAdditionStateManager> getBrokerAdditionsStateManagers()
BrokerAdditionStateManager
s for the active broker addition operationsgetBrokerAdditionsStateManagers
in interface DataBalanceEngineContext
public void putBrokerStateManager(BrokerAdditionStateManager stateManager)
putBrokerStateManager
in interface DataBalanceEngineContext
public BalancerStatusTracker getBalancerStatusTracker()
getBalancerStatusTracker
in interface DataBalanceEngineContext
public void setBalancerStatusTracker(BalancerStatusTracker balancerStatusTracker)
public EvenClusterLoadStateManager getEvenClusterLoadStateManager()
getEvenClusterLoadStateManager
in interface DataBalanceEngineContext
public Set<Integer> brokersBeingAdded()
MultiBrokerAdditionOperation
brokersBeingAdded
in interface DataBalanceEngineContext
public void setPersistenceStore(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()