SERVER-50660 Integrate CancelationTokens with OperationContext

1 files changed, 12 insertions, 1 deletions

diff --git a/src/mongo/db/s/transaction_coordinator_service.cpp b/src/mongo/db/s/transaction_coordinator_service.cpp
index 98f0944d510..d75fc8b3a39 100644
--- a/src/mongo/db/s/transaction_coordinator_service.cpp
+++ b/src/mongo/db/s/transaction_coordinator_service.cpp
@@ -83,7 +83,18 @@ void TransactionCoordinatorService::createCoordinator(OperationContext* opCtx,
     auto coordinator = std::make_shared<TransactionCoordinator>(
         opCtx, lsid, txnNumber, scheduler.makeChildScheduler(), commitDeadline);
 
-    catalog.insert(opCtx, lsid, txnNumber, coordinator);
+    try {
+        catalog.insert(opCtx, lsid, txnNumber, coordinator);
+    } catch (const DBException&) {
+        // Handle the case where the opCtx has been interrupted and we do not successfully insert
+        // the coordinator into the catalog.
+        coordinator->cancelIfCommitNotYetStarted();
+        // Wait for it to finish processing the error before throwing, since leaving this scope will
+        // cause the newly created coordinator to be destroyed. We ignore the return Status since we
+        // want to just rethrow whatever exception was thrown when inserting into the catalog.
+        [[maybe_unused]] auto status = coordinator->onCompletion().waitNoThrow();
+        throw;
+    }
 }