From 47943ade8cc662381fc7ce8f3abd043f3561c92b Mon Sep 17 00:00:00 2001
From: guojialiang <guojialiang.2012@bytedance.com>
Date: Tue, 29 Jul 2025 11:10:39 +0800
Subject: [PATCH] clean up completedReplications when SegmentReplicator.cancel
 is invoked

Signed-off-by: guojialiang <guojialiang.2012@bytedance.com>
---
 .../org/opensearch/indices/replication/SegmentReplicator.java    | 1 +
 1 file changed, 1 insertion(+)

diff --git a/server/src/main/java/org/opensearch/indices/replication/SegmentReplicator.java b/server/src/main/java/org/opensearch/indices/replication/SegmentReplicator.java
index 72abd4f33e465..342b83679861a 100644
--- a/server/src/main/java/org/opensearch/indices/replication/SegmentReplicator.java
+++ b/server/src/main/java/org/opensearch/indices/replication/SegmentReplicator.java
@@ -395,6 +395,7 @@ void cancel(ShardId shardId, String reason) {
         onGoingMergedSegmentReplications.cancelForShard(shardId, reason);
         replicationCheckpointStats.remove(shardId);
         primaryCheckpoint.remove(shardId);
+        completedReplications.remove(shardId);
     }
 
     SegmentReplicationTarget get(ShardId shardId) {