SERVER-36483 Do not reap transaction entries for prepared transactions

1 files changed, 11 insertions, 2 deletions

diff --git a/src/mongo/db/session_catalog_mongod.cpp b/src/mongo/db/session_catalog_mongod.cpp
index d17280e9ca3..39880028b28 100644
--- a/src/mongo/db/session_catalog_mongod.cpp
+++ b/src/mongo/db/session_catalog_mongod.cpp
@@ -202,8 +202,17 @@ void MongoDSessionCatalog::invalidateSessions(OperationContext* opCtx,
     auto sessionKillTokens = std::make_shared<std::vector<SessionCatalog::KillToken>>();
 
     if (singleSessionDoc) {
-        sessionKillTokens->emplace_back(catalog->killSession(LogicalSessionId::parse(
-            IDLParserErrorContext("lsid"), singleSessionDoc->getField("_id").Obj())));
+        const auto lsid = LogicalSessionId::parse(IDLParserErrorContext("lsid"),
+                                                  singleSessionDoc->getField("_id").Obj());
+        catalog->scanSession(lsid, [&sessionKillTokens](const ObservableSession& session) {
+            const auto participant = TransactionParticipant::get(session);
+            uassert(ErrorCodes::PreparedTransactionInProgress,
+                    str::stream() << "Cannot modify the entry for session "
+                                  << session.getSessionId().getId()
+                                  << " because it is in the prepared state",
+                    !participant.transactionIsPrepared());
+            sessionKillTokens->emplace_back(session.kill());
+        });
     } else {
         SessionKiller::Matcher matcher(
             KillAllSessionsByPatternSet{makeKillAllSessionsByPattern(opCtx)});