clean up completedReplications when SegmentReplicator.cancel is invoked (opensearch-project#18856)

guojialiang92 · jainankitk · commit e6d1fc99ee0d · 2025-09-22T15:49:05.000-07:00
Signed-off-by: guojialiang &lt;guojialiang.2012@bytedance.com&gt;
Signed-off-by: Ankit Jain &lt;jainankitk@apache.org&gt;
diff --git a/server/src/main/java/org/opensearch/indices/replication/SegmentReplicator.java b/server/src/main/java/org/opensearch/indices/replication/SegmentReplicator.java
@@ -399,6 +399,7 @@ void cancel(ShardId shardId, String reason) {
         onGoingMergedSegmentReplications.cancelForShard(shardId, reason);
         replicationCheckpointStats.remove(shardId);
         primaryCheckpoint.remove(shardId);
+        completedReplications.remove(shardId);
     }
 
     SegmentReplicationTarget get(ShardId shardId) {

Original file line number	Diff line number	Diff line change
`@@ -399,6 +399,7 @@ void cancel(ShardId shardId, String reason) {`
`399`	`399`	`onGoingMergedSegmentReplications.cancelForShard(shardId, reason);`
`400`	`400`	`replicationCheckpointStats.remove(shardId);`
`401`	`401`	`primaryCheckpoint.remove(shardId);`
	`402`	`+ completedReplications.remove(shardId);`
`402`	`403`	`}`
`403`	`404`
`404`	`405`	`SegmentReplicationTarget get(ShardId shardId) {`