SERVER-61090 Make setFCV wait for all TransactionCoordinators of internal transactions to be cleaned up

1 files changed, 14 insertions, 0 deletions

diff --git a/src/mongo/db/s/transaction_coordinator.h b/src/mongo/db/s/transaction_coordinator.h
index ba8eaf144e6..ddf4af355ac 100644
--- a/src/mongo/db/s/transaction_coordinator.h
+++ b/src/mongo/db/s/transaction_coordinator.h
@@ -131,6 +131,15 @@ public:
 
     Step getStep() const;
 
+    /**
+     * TODO: SERVER-62375 Remove upgrade/downgrade code for internal transactions.
+     * Returns a future that will be resolved when we remove the coordinator
+     * document.
+     */
+    SharedSemiFuture<void> getCoordinatorDocRemovalFuture() {
+        return _coordinatorDocRemovalPromise.getFuture();
+    }
+
 
 private:
     void _updateAssociatedClient(Client* client);
@@ -196,6 +205,11 @@ private:
     bool _decisionDurable{false};
     SharedPromise<txn::CommitDecision> _decisionPromise;
 
+    // TODO: SERVER-62375 Remove upgrade/downgrade code for internal transactions.
+    // Promise that is set when we have successfully removed the coordinator
+    // document.
+    SharedPromise<void> _coordinatorDocRemovalPromise;
+
     // A list of all promises corresponding to futures that were returned to callers of
     // onCompletion.
     SharedPromise<txn::CommitDecision> _completionPromise;