SERVER-64517 Recover RecoverableCriticalSection after initialSync and startupRecovery have completed

(cherry picked from commit ce0bbc5ec1728e443c5ff893a78693b24570b80d)

1 files changed, 12 insertions, 0 deletions

diff --git a/src/mongo/db/repl/replica_set_aware_service.h b/src/mongo/db/repl/replica_set_aware_service.h
index eb38cac0252..f7ea8c3a71e 100644
--- a/src/mongo/db/repl/replica_set_aware_service.h
+++ b/src/mongo/db/repl/replica_set_aware_service.h
@@ -118,6 +118,16 @@ public:
     virtual void onStartup(OperationContext* opCtx) = 0;
 
     /**
+     * Called after startup recovery has completed.
+     */
+    virtual void onStartupRecoveryComplete(OperationContext* opCtx) = 0;
+
+    /**
+     * Called after initial sync has completed.
+     */
+    virtual void onInitialSyncComplete(OperationContext* opCtx) = 0;
+
+    /**
      * Called as part of ReplicationCoordinator shutdown.
      */
     virtual void onShutdown() = 0;
@@ -190,6 +200,8 @@ public:
     static ReplicaSetAwareServiceRegistry& get(ServiceContext* serviceContext);
 
     void onStartup(OperationContext* opCtx) final;
+    void onStartupRecoveryComplete(OperationContext* opCtx) final;
+    void onInitialSyncComplete(OperationContext* opCtx) final;
     void onShutdown() final;
     void onStepUpBegin(OperationContext* opCtx, long long term) final;
     void onStepUpComplete(OperationContext* opCtx, long long term) final;