also replica

Author: tlrx

server/src/main/java/org/elasticsearch/action/support/replication/TransportReplicationAction.java

@@ -663,6 +663,8 @@ private final class AsyncReplicaAction extends AbstractRunnable implements Actio
         // something we want to avoid at all costs
         private final ClusterStateObserver observer = new ClusterStateObserver(clusterService, null, logger, threadPool.getThreadContext());
         private final ConcreteReplicaRequest<ReplicaRequest> replicaRequest;
+        private final Supplier<Long> localCheckpointSupplier;
+        private final Supplier<Long> globalCheckpointSupplier;
 
         AsyncReplicaAction(
             ConcreteReplicaRequest<ReplicaRequest> replicaRequest,
@@ -675,6 +677,9 @@ private final class AsyncReplicaAction extends AbstractRunnable implements Actio
             final ShardId shardId = replicaRequest.getRequest().shardId();
             assert shardId != null : "request shardId must be set";
             this.replica = getIndexShard(shardId);
+            this.localCheckpointSupplier = replica.getLocalCheckpointSupplier();
+            this.globalCheckpointSupplier = replica.getLastSyncedGlobalCheckpointSupplier();
+
         }
 
         @Override
@@ -685,10 +690,7 @@ public void onResponse(Releasable releasable) {
                     replicaRequest.getRequest(),
                     replica,
                     ActionListener.wrap((replicaResult) -> replicaResult.runPostReplicaActions(ActionListener.wrap(r -> {
-                        final ReplicaResponse response = new ReplicaResponse(
-                            replica.getLocalCheckpoint(),
-                            replica.getLastSyncedGlobalCheckpoint()
-                        );
+                        final ReplicaResponse response = new ReplicaResponse(localCheckpointSupplier.get(), globalCheckpointSupplier.get());
                         releasable.close(); // release shard operation lock before responding to caller
                         if (logger.isTraceEnabled()) {
                             logger.trace(