6969import org .opensearch .index .store .Store ;
7070import org .opensearch .index .translog .Translog ;
7171import org .opensearch .index .translog .TranslogCorruptedException ;
72- import org .opensearch .indices .replication .common .ShardTargetCollection ;
73- import org .opensearch .indices .replication .common .ShardTargetCollection . ShardTargetRef ;
72+ import org .opensearch .indices .replication .common .ReplicationCollection ;
73+ import org .opensearch .indices .replication .common .ReplicationCollection . ReplicationRef ;
7474import org .opensearch .indices .replication .common .ReplicationLuceneIndex ;
7575import org .opensearch .indices .replication .common .ReplicationTimer ;
7676import org .opensearch .tasks .Task ;
@@ -125,7 +125,7 @@ public static class Actions {
125125 private final RecoverySettings recoverySettings ;
126126 private final ClusterService clusterService ;
127127
128- private final ShardTargetCollection <RecoveryTarget > onGoingRecoveries ;
128+ private final ReplicationCollection <RecoveryTarget > onGoingRecoveries ;
129129
130130 public PeerRecoveryTargetService (
131131 ThreadPool threadPool ,
@@ -137,7 +137,7 @@ public PeerRecoveryTargetService(
137137 this .transportService = transportService ;
138138 this .recoverySettings = recoverySettings ;
139139 this .clusterService = clusterService ;
140- this .onGoingRecoveries = new ShardTargetCollection <>(logger , threadPool );
140+ this .onGoingRecoveries = new ReplicationCollection <>(logger , threadPool );
141141
142142 transportService .registerRequestHandler (
143143 Actions .FILES_INFO ,
@@ -229,7 +229,7 @@ private void doRecovery(final long recoveryId, final StartRecoveryRequest preExi
229229 final TransportRequest requestToSend ;
230230 final StartRecoveryRequest startRequest ;
231231 final ReplicationTimer timer ;
232- try (ShardTargetCollection . ShardTargetRef <RecoveryTarget > recoveryRef = onGoingRecoveries .get (recoveryId )) {
232+ try (ReplicationRef <RecoveryTarget > recoveryRef = onGoingRecoveries .get (recoveryId )) {
233233 if (recoveryRef == null ) {
234234 logger .trace ("not running recovery with id [{}] - can not find it (probably finished)" , recoveryId );
235235 return ;
@@ -353,7 +353,7 @@ class PrepareForTranslogOperationsRequestHandler implements TransportRequestHand
353353
354354 @ Override
355355 public void messageReceived (RecoveryPrepareForTranslogOperationsRequest request , TransportChannel channel , Task task ) {
356- try (ShardTargetRef <RecoveryTarget > recoveryRef = onGoingRecoveries .getSafe (request .recoveryId (), request .shardId ())) {
356+ try (ReplicationRef <RecoveryTarget > recoveryRef = onGoingRecoveries .getSafe (request .recoveryId (), request .shardId ())) {
357357 final ActionListener <Void > listener = createOrFinishListener (recoveryRef , channel , Actions .PREPARE_TRANSLOG , request );
358358 if (listener == null ) {
359359 return ;
@@ -368,7 +368,7 @@ class FinalizeRecoveryRequestHandler implements TransportRequestHandler<Recovery
368368
369369 @ Override
370370 public void messageReceived (RecoveryFinalizeRecoveryRequest request , TransportChannel channel , Task task ) throws Exception {
371- try (ShardTargetRef <RecoveryTarget > recoveryRef = onGoingRecoveries .getSafe (request .recoveryId (), request .shardId ())) {
371+ try (ReplicationRef <RecoveryTarget > recoveryRef = onGoingRecoveries .getSafe (request .recoveryId (), request .shardId ())) {
372372 final ActionListener <Void > listener = createOrFinishListener (recoveryRef , channel , Actions .FINALIZE , request );
373373 if (listener == null ) {
374374 return ;
@@ -384,7 +384,7 @@ class HandoffPrimaryContextRequestHandler implements TransportRequestHandler<Rec
384384 @ Override
385385 public void messageReceived (final RecoveryHandoffPrimaryContextRequest request , final TransportChannel channel , Task task )
386386 throws Exception {
387- try (ShardTargetRef <RecoveryTarget > recoveryRef = onGoingRecoveries .getSafe (request .recoveryId (), request .shardId ())) {
387+ try (ReplicationRef <RecoveryTarget > recoveryRef = onGoingRecoveries .getSafe (request .recoveryId (), request .shardId ())) {
388388 recoveryRef .get ().handoffPrimaryContext (request .primaryContext ());
389389 }
390390 channel .sendResponse (TransportResponse .Empty .INSTANCE );
@@ -397,7 +397,7 @@ class TranslogOperationsRequestHandler implements TransportRequestHandler<Recove
397397 @ Override
398398 public void messageReceived (final RecoveryTranslogOperationsRequest request , final TransportChannel channel , Task task )
399399 throws IOException {
400- try (ShardTargetRef <RecoveryTarget > recoveryRef = onGoingRecoveries .getSafe (request .recoveryId (), request .shardId ())) {
400+ try (ReplicationRef <RecoveryTarget > recoveryRef = onGoingRecoveries .getSafe (request .recoveryId (), request .shardId ())) {
401401 final RecoveryTarget recoveryTarget = recoveryRef .get ();
402402 final ActionListener <Void > listener = createOrFinishListener (
403403 recoveryRef ,
@@ -417,7 +417,7 @@ public void messageReceived(final RecoveryTranslogOperationsRequest request, fin
417417 private void performTranslogOps (
418418 final RecoveryTranslogOperationsRequest request ,
419419 final ActionListener <Void > listener ,
420- final ShardTargetRef <RecoveryTarget > recoveryRef
420+ final ReplicationRef <RecoveryTarget > recoveryRef
421421 ) {
422422 final RecoveryTarget recoveryTarget = recoveryRef .get ();
423423
@@ -433,7 +433,7 @@ private void performTranslogOps(
433433 public void onNewClusterState (ClusterState state ) {
434434 threadPool .generic ().execute (ActionRunnable .wrap (listener , l -> {
435435 try (
436- ShardTargetRef <RecoveryTarget > recoveryRef = onGoingRecoveries .getSafe (
436+ ReplicationRef <RecoveryTarget > recoveryRef = onGoingRecoveries .getSafe (
437437 request .recoveryId (),
438438 request .shardId ()
439439 )
@@ -483,7 +483,7 @@ class FilesInfoRequestHandler implements TransportRequestHandler<RecoveryFilesIn
483483
484484 @ Override
485485 public void messageReceived (RecoveryFilesInfoRequest request , TransportChannel channel , Task task ) throws Exception {
486- try (ShardTargetRef <RecoveryTarget > recoveryRef = onGoingRecoveries .getSafe (request .recoveryId (), request .shardId ())) {
486+ try (ReplicationRef <RecoveryTarget > recoveryRef = onGoingRecoveries .getSafe (request .recoveryId (), request .shardId ())) {
487487 final ActionListener <Void > listener = createOrFinishListener (recoveryRef , channel , Actions .FILES_INFO , request );
488488 if (listener == null ) {
489489 return ;
@@ -506,7 +506,7 @@ class CleanFilesRequestHandler implements TransportRequestHandler<RecoveryCleanF
506506
507507 @ Override
508508 public void messageReceived (RecoveryCleanFilesRequest request , TransportChannel channel , Task task ) throws Exception {
509- try (ShardTargetRef <RecoveryTarget > recoveryRef = onGoingRecoveries .getSafe (request .recoveryId (), request .shardId ())) {
509+ try (ReplicationRef <RecoveryTarget > recoveryRef = onGoingRecoveries .getSafe (request .recoveryId (), request .shardId ())) {
510510 final ActionListener <Void > listener = createOrFinishListener (recoveryRef , channel , Actions .CLEAN_FILES , request );
511511 if (listener == null ) {
512512 return ;
@@ -525,7 +525,7 @@ class FileChunkTransportRequestHandler implements TransportRequestHandler<Recove
525525
526526 @ Override
527527 public void messageReceived (final RecoveryFileChunkRequest request , TransportChannel channel , Task task ) throws Exception {
528- try (ShardTargetRef <RecoveryTarget > recoveryRef = onGoingRecoveries .getSafe (request .recoveryId (), request .shardId ())) {
528+ try (ReplicationRef <RecoveryTarget > recoveryRef = onGoingRecoveries .getSafe (request .recoveryId (), request .shardId ())) {
529529 final RecoveryTarget recoveryTarget = recoveryRef .get ();
530530 final ActionListener <Void > listener = createOrFinishListener (recoveryRef , channel , Actions .FILE_CHUNK , request );
531531 if (listener == null ) {
@@ -561,7 +561,7 @@ public void messageReceived(final RecoveryFileChunkRequest request, TransportCha
561561 }
562562
563563 private ActionListener <Void > createOrFinishListener (
564- final ShardTargetRef <RecoveryTarget > recoveryRef ,
564+ final ReplicationRef <RecoveryTarget > recoveryRef ,
565565 final TransportChannel channel ,
566566 final String action ,
567567 final RecoveryTransportRequest request
@@ -570,7 +570,7 @@ private ActionListener<Void> createOrFinishListener(
570570 }
571571
572572 private ActionListener <Void > createOrFinishListener (
573- final ShardTargetRef <RecoveryTarget > recoveryRef ,
573+ final ReplicationRef <RecoveryTarget > recoveryRef ,
574574 final TransportChannel channel ,
575575 final String action ,
576576 final RecoveryTransportRequest request ,
@@ -607,7 +607,7 @@ class RecoveryRunner extends AbstractRunnable {
607607
608608 @ Override
609609 public void onFailure (Exception e ) {
610- try (ShardTargetRef <RecoveryTarget > recoveryRef = onGoingRecoveries .get (recoveryId )) {
610+ try (ReplicationRef <RecoveryTarget > recoveryRef = onGoingRecoveries .get (recoveryId )) {
611611 if (recoveryRef != null ) {
612612 logger .error (() -> new ParameterizedMessage ("unexpected error during recovery [{}], failing shard" , recoveryId ), e );
613613 onGoingRecoveries .fail (
0 commit comments