SERVER-31941 Disallow retryable writes in storage engines that do not support document-level locking

1 files changed, 7 insertions, 2 deletions

diff --git a/src/mongo/db/initialize_operation_session_info.cpp b/src/mongo/db/initialize_operation_session_info.cpp
index b4eb785e2c5..9db8ab8c0c8 100644
--- a/src/mongo/db/initialize_operation_session_info.cpp
+++ b/src/mongo/db/initialize_operation_session_info.cpp
@@ -42,7 +42,8 @@ namespace mongo {
 void initializeOperationSessionInfo(OperationContext* opCtx,
                                     const BSONObj& requestBody,
                                     bool requiresAuth,
-                                    bool canAcceptTxnNumber) {
+                                    bool isReplSetMemberOrMongos,
+                                    bool supportsDocLocking) {
     if (!requiresAuth) {
         return;
     }
@@ -85,7 +86,11 @@ void initializeOperationSessionInfo(OperationContext* opCtx,
                 opCtx->getLogicalSessionId());
         uassert(ErrorCodes::IllegalOperation,
                 "Transaction numbers are only allowed on a replica set member or mongos",
-                canAcceptTxnNumber);
+                isReplSetMemberOrMongos);
+        uassert(ErrorCodes::IllegalOperation,
+                "Transaction numbers are only allowed on storage engines that support "
+                "document-level locking",
+                supportsDocLocking);
         uassert(ErrorCodes::BadValue,
                 "Transaction number cannot be negative",
                 *osi.getTxnNumber() >= 0);