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<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(int brokerId, BrokerRemovalFuture future)
brokerId
- the id of the brokerfuture
- a #BrokerRemovalFuture
for the full broker removal operation of broker #brokerId
public void removeBrokerRemovalFuture(int brokerId)
public BrokerRemovalFuture brokerRemovalFuture(int brokerId)
Future
for the full broker removal operation of broker #brokerId
-
the plan computation/shutdown operation and then the underlying reassignments executionpublic void closeAndClearState()