adoroszlai commented on code in PR #8057:
URL: https://github.com/apache/ozone/pull/8057#discussion_r1993392573
##
hadoop-hdds/server-scm/src/main/java/org/apache/hadoop/hdds/scm/container/states/ContainerStateMap.java:
##
@@ -214,30 +278,39 @@ public void updateState(ContainerID containerID,
LifeCycleState currentState,
currentInfo.setState(newState);
}
- public Set getAllContainerIDs() {
-return ImmutableSet.copyOf(containerMap.keySet());
+ public List getContainerInfos(ContainerID start, int count) {
+return containerMap.getInfos(start, count);
}
/**
- * Returns Containers in the System by the Type.
*
- * @param type - Replication type -- StandAlone, Ratis etc.
- * @return NavigableSet
+ * @param state the state of the {@link ContainerInfo}s
+ * @param start the start id
+ * @param count the maximum size of the returned list
+ * @return a list of {@link ContainerInfo}s sorted by {@link ContainerID}
*/
- public NavigableSet getContainerIDsByType(final ReplicationType
type) {
-Preconditions.checkNotNull(type);
-return typeMap.getCollection(type);
+ public List getContainerInfos(LifeCycleState state,
ContainerID start, int count) {
+Preconditions.assertTrue(count >= 0, "count < 0");
+return lifeCycleStateMap.tailSet(state, start).stream()
+.map(this::getContainerInfo)
+.limit(count)
+.collect(Collectors.toList());
}
- /**
- * Returns Containers by State.
- *
- * @param state - State - Open, Closed etc.
- * @return List of containers by state.
- */
- public NavigableSet getContainerIDsByState(
- final LifeCycleState state) {
-Preconditions.checkNotNull(state);
-return lifeCycleStateMap.getCollection(state);
+ public List getContainerInfos(LifeCycleState state) {
+return lifeCycleStateMap.getCollection(state).stream()
+.map(this::getContainerInfo)
+.collect(Collectors.toList());
+ }
+
+ public List getContainerInfos(ReplicationType state) {
+return typeMap.getCollection(state).stream()
Review Comment:
nit:
```suggestion
public List getContainerInfos(ReplicationType type) {
return typeMap.getCollection(type).stream()
```
##
hadoop-hdds/server-scm/src/main/java/org/apache/hadoop/hdds/scm/container/ContainerStateManager.java:
##
@@ -103,22 +104,33 @@ public interface ContainerStateManager {
boolean contains(ContainerID containerID);
/**
- * Returns the ID of all the managed containers.
+ * Get {@link ContainerInfo}s.
*
- * @return Set of {@link ContainerID}
+ * @param start the start {@link ContainerID} (inclusive)
+ * @param count the size limit
+ * @return a list of {@link ContainerInfo};
*/
- Set getContainerIDs();
+ List getContainerInfos(ContainerID start, int count);
/**
+ * Get {@link ContainerInfo}s for the given state.
*
+ * @param start the start {@link ContainerID} (inclusive)
+ * @param count the size limit
+ * @return a list of {@link ContainerInfo};
*/
- Set getContainerIDs(LifeCycleState state);
+ List getContainerInfos(LifeCycleState state, ContainerID
start, int count);
+ /** @return all {@link ContainerInfo}s for the given state. */
+ List getContainerInfos(LifeCycleState state);
/**
- * Returns the IDs of the Containers whose ReplicationType matches the given
type.
+ * @return all {@link ContainerID}s for the given state.
Review Comment:
nit:
```suggestion
* @return number of containers for the given state.
```
--
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.
To unsubscribe, e-mail: issues-unsubscr...@ozone.apache.org
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
-
To unsubscribe, e-mail: issues-unsubscr...@ozone.apache.org
For additional commands, e-mail: issues-h...@ozone.apache.org
