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

1 files changed, 27 insertions, 0 deletions

diff --git a/src/mongo/db/s/transaction_coordinator_service.cpp b/src/mongo/db/s/transaction_coordinator_service.cpp
index bdc4cc373d9..f936323dcee 100644
--- a/src/mongo/db/s/transaction_coordinator_service.cpp
+++ b/src/mongo/db/s/transaction_coordinator_service.cpp
@@ -370,4 +370,31 @@ void TransactionCoordinatorService::cancelIfCommitNotYetStarted(
     }
 }
 
+const std::vector<SharedSemiFuture<void>>
+TransactionCoordinatorService::getAllRemovalFuturesForCoordinatorsForInternalTransactions(
+    OperationContext* opCtx) {
+    std::vector<SharedSemiFuture<void>> coordinatorStateDocRemovalFutures;
+    std::shared_ptr<CatalogAndScheduler> cas = _getCatalogAndScheduler(opCtx);
+    auto& catalog = cas->catalog;
+
+    auto predicate = [](const LogicalSessionId lsid,
+                        const TxnNumberAndRetryCounter txnNumberAndRetryCounter,
+                        const std::shared_ptr<TransactionCoordinator> transactionCoordinator) {
+        TransactionCoordinator::Step step = transactionCoordinator->getStep();
+        if (step > TransactionCoordinator::Step::kInactive && getParentSessionId(lsid)) {
+            return true;
+        }
+        return false;
+    };
+
+    auto visitorAction = [&](const LogicalSessionId lsid,
+                             const TxnNumberAndRetryCounter txnNumberAndRetryCounter,
+                             const std::shared_ptr<TransactionCoordinator> transactionCoordinator) {
+        coordinatorStateDocRemovalFutures.push_back(
+            transactionCoordinator->getCoordinatorDocRemovalFuture());
+    };
+    catalog.filter(predicate, visitorAction);
+    return coordinatorStateDocRemovalFutures;
+}
+
 }  // namespace mongo