diff --git a/ambry-prioritization/src/main/java/com/github/ambry/replica/prioritization/ReplicationPrioritizationManager.java b/ambry-prioritization/src/main/java/com/github/ambry/replica/prioritization/ReplicationPrioritizationManager.java index 8eaffd8a67..60e57527b7 100644 --- a/ambry-prioritization/src/main/java/com/github/ambry/replica/prioritization/ReplicationPrioritizationManager.java +++ b/ambry-prioritization/src/main/java/com/github/ambry/replica/prioritization/ReplicationPrioritizationManager.java @@ -394,6 +394,7 @@ private void resetToNormalReplication() { private void processCompletedPartitions() { currentlyReplicatingPriorityPartitions.stream().filter(this::hasCompletedReplication).forEach(completedPartitions::add); + replicationPrioritizationMetrics.recordPartitionReplicationComplete(completedPartitions.size()); // Remove completed partitions from current set currentlyReplicatingPriorityPartitions.removeAll(completedPartitions); replicationPrioritizationMetrics.updateCurrentlyReplicatingPriorityPartitions(currentlyReplicatingPriorityPartitions); diff --git a/ambry-prioritization/src/main/java/com/github/ambry/replica/prioritization/ReplicationPrioritizationMetrics.java b/ambry-prioritization/src/main/java/com/github/ambry/replica/prioritization/ReplicationPrioritizationMetrics.java index 4dafaacc86..1ff8344a51 100644 --- a/ambry-prioritization/src/main/java/com/github/ambry/replica/prioritization/ReplicationPrioritizationMetrics.java +++ b/ambry-prioritization/src/main/java/com/github/ambry/replica/prioritization/ReplicationPrioritizationMetrics.java @@ -242,6 +242,13 @@ public void recordPartitionReplicationComplete() { partitionsCompletedCount.inc(); } + /** + * Record completion of replication for a partition. + */ + public void recordPartitionReplicationComplete(int numPartitions) { + partitionsCompletedCount.inc(numPartitions); + } + /** * Record addition of partitions to high priority replication. * @param newHighPriorityPartitions The partitions added to high priority.