clusterConfig.getStateTransitionThrottleConfigs();
clusterConfig.getStateTransitionThrottleConfigs();
private int getNumExtraReplicas(ClusterConfig clusterConfig) { int numExtraReplicas = StateTransitionThrottleConfig.DEFAULT_NUM_TRANSIT_REPLICAS; List<StateTransitionThrottleConfig> stateTransitionThrottleConfigs = clusterConfig.getStateTransitionThrottleConfigs(); for (StateTransitionThrottleConfig throttleConfig : stateTransitionThrottleConfigs) { if (StateTransitionThrottleConfig.ThrottleScope.PARTITION .equals(throttleConfig.getThrottleScope()) && StateTransitionThrottleConfig.RebalanceType.LOAD_BALANCE .equals(throttleConfig.getRebalanceType())) { numExtraReplicas = (int) Math.min(numExtraReplicas, throttleConfig.getMaxPartitionInTransition()); } } return numExtraReplicas; } }
private int getNumExtraReplicas(ClusterConfig clusterConfig) { int numExtraReplicas = StateTransitionThrottleConfig.DEFAULT_NUM_TRANSIT_REPLICAS; List<StateTransitionThrottleConfig> stateTransitionThrottleConfigs = clusterConfig.getStateTransitionThrottleConfigs(); for (StateTransitionThrottleConfig throttleConfig : stateTransitionThrottleConfigs) { if (StateTransitionThrottleConfig.ThrottleScope.PARTITION .equals(throttleConfig.getThrottleScope()) && StateTransitionThrottleConfig.RebalanceType.LOAD_BALANCE .equals(throttleConfig.getRebalanceType())) { numExtraReplicas = (int) Math.min(numExtraReplicas, throttleConfig.getMaxPartitionInTransition()); } } return numExtraReplicas; } }
private void setSingleThrottlingConfig(StateTransitionThrottleConfig.RebalanceType rebalanceType, StateTransitionThrottleConfig.ThrottleScope scope, int maxStateTransitions) { ClusterConfig clusterConfig = _configAccessor.getClusterConfig(CLUSTER_NAME); StateTransitionThrottleConfig anyTypeInstanceThrottle = new StateTransitionThrottleConfig(rebalanceType, scope, maxStateTransitions); List<StateTransitionThrottleConfig> currentThrottleConfigs = clusterConfig.getStateTransitionThrottleConfigs(); currentThrottleConfigs.add(anyTypeInstanceThrottle); clusterConfig.setStateTransitionThrottleConfigs(currentThrottleConfigs); _configAccessor.setClusterConfig(CLUSTER_NAME, clusterConfig); }
StateTransitionThrottleConfig.ThrottleScope.CLUSTER, 1); List<StateTransitionThrottleConfig> currentThrottleConfig = clusterConfig.getStateTransitionThrottleConfigs(); currentThrottleConfig.add(throttleConfigForLoadRebalance); clusterConfig.setStateTransitionThrottleConfigs(currentThrottleConfig);