kfaraz commented on code in PR #13197:
URL: https://github.com/apache/druid/pull/13197#discussion_r1232981597
##########
server/src/main/java/org/apache/druid/server/coordinator/SegmentReplicantLookup.java:
##########
@@ -125,49 +117,85 @@ public int getMovingReplicas(SegmentId segmentId, String
tier)
* Number of replicas of the segment which are safely loaded on the given
tier
* and are not being dropped.
*/
- public int getLoadedReplicas(SegmentId segmentId, String tier)
+ public int getLoadedNotDroppingReplicas(SegmentId segmentId, String tier)
+ {
+ ReplicaCount count = replicaCounts.get(segmentId, tier);
+ return (count == null) ? 0 : count.loadedNotDropping();
+ }
+
+ public int getLoadingReplicas(SegmentId segmentId, String tier)
{
ReplicaCount count = replicaCounts.get(segmentId, tier);
- return (count == null) ? 0 : count.safelyLoaded();
+ return count == null ? 0 : count.loading;
}
/**
- * Number of replicas of the segment which are safely loaded on the cluster
- * and are not being dropped.
+ * Number of replicas of the segment which are loaded on the cluster.
+ *
+ * @param includeDropping Whether segments which are being dropped should be
+ * included in the total count.
*/
- public int getTotalLoadedReplicas(SegmentId segmentId)
+ public int getLoadedReplicas(SegmentId segmentId, boolean includeDropping)
{
final Map<String, ReplicaCount> allTiers = replicaCounts.row(segmentId);
int totalLoaded = 0;
for (ReplicaCount count : allTiers.values()) {
- totalLoaded += count.safelyLoaded();
+ totalLoaded += includeDropping ? count.loaded :
count.loadedNotDropping();
}
return totalLoaded;
}
Review Comment:
Updated.
--
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: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
