pkuwm commented on a change in pull request #362: The WAGED rebalancer cluster model implementation URL: https://github.com/apache/helix/pull/362#discussion_r307935935
########## File path: helix-core/src/main/java/org/apache/helix/controller/rebalancer/waged/model/AssignableReplica.java ########## @@ -19,9 +19,124 @@ * under the License. */ +import org.apache.helix.model.ResourceConfig; +import org.apache.helix.model.StateModelDefinition; + +import java.io.IOException; +import java.util.Map; + /** - * A placeholder before we have the implementation. - * * This class represents a partition replication that needs to be allocated. */ -public class AssignableReplica { } +public class AssignableReplica implements Comparable<AssignableReplica> { + private final String _partitionName; + private final String _resourceName; + private final String _resourceInstanceGroupTag; + private final int _resourceMaxPartitionsPerInstance; + private final Map<String, Integer> _capacityUsage; + // The priority of the replica's state + private final int _statePriority; + // The state of the replica + private final String _replicaState; + + /** + * @param resourceConfig The resource config for the resource which contains the replication. + * @param partitionName The replication's partition name. + * @param replicaState The state of the replication. + * @param statePriority The priority of the replication's state. + */ + AssignableReplica(ResourceConfig resourceConfig, String partitionName, String replicaState, + int statePriority) { + _partitionName = partitionName; + _replicaState = replicaState; + _statePriority = statePriority; + _resourceName = resourceConfig.getResourceName(); + _capacityUsage = fetchCapacityUsage(partitionName, resourceConfig); + _resourceInstanceGroupTag = resourceConfig.getInstanceGroupTag(); + _resourceMaxPartitionsPerInstance = resourceConfig.getMaxPartitionsPerInstance(); + } + + public Map<String, Integer> getCapacity() { + return _capacityUsage; + } + + public String getPartitionName() { + return _partitionName; + } + + public String getReplicaState() { + return _replicaState; + } + + public boolean isReplicaTopState() { + return _statePriority == StateModelDefinition.TOP_STATE_PRIORITY; + } + + public int getStatePriority() { + return _statePriority; + } + + public String getResourceName() { + return _resourceName; + } + + public String getResourceInstanceGroupTag() { + return _resourceInstanceGroupTag; + } + + public Integer getResourceMaxPartitionsPerInstance() { Review comment: Checked callers and assignments, it seems there is no need to return an `Integer`? ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services