SERVER-37179 Pass the reference of OperationSessionInfoFromClient around.

1 files changed, 24 insertions, 9 deletions

diff --git a/src/mongo/db/initialize_operation_session_info.cpp b/src/mongo/db/initialize_operation_session_info.cpp
index a7f4b003499..e513126f173 100644
--- a/src/mongo/db/initialize_operation_session_info.cpp
+++ b/src/mongo/db/initialize_operation_session_info.cpp
@@ -40,12 +40,11 @@
 
 namespace mongo {
 
-boost::optional<OperationSessionInfoFromClient> initializeOperationSessionInfo(
-    OperationContext* opCtx,
-    const BSONObj& requestBody,
-    bool requiresAuth,
-    bool isReplSetMemberOrMongos,
-    bool supportsDocLocking) {
+OperationSessionInfoFromClient initializeOperationSessionInfo(OperationContext* opCtx,
+                                                              const BSONObj& requestBody,
+                                                              bool requiresAuth,
+                                                              bool isReplSetMemberOrMongos,
+                                                              bool supportsDocLocking) {
     auto osi = OperationSessionInfoFromClient::parse("OperationSessionInfo"_sd, requestBody);
 
     if (opCtx->getClient()->isInDirectClient()) {
@@ -61,7 +60,7 @@ boost::optional<OperationSessionInfoFromClient> initializeOperationSessionInfo(
                 !osi.getAutocommit());
         uassert(
             50889, "It is illegal to provide a txnNumber for this command", !osi.getTxnNumber());
-        return boost::none;
+        return {};
     }
 
     {
@@ -71,7 +70,7 @@ boost::optional<OperationSessionInfoFromClient> initializeOperationSessionInfo(
         AuthorizationSession* authSession = AuthorizationSession::get(opCtx->getClient());
         if (authSession && authSession->isUsingLocalhostBypass() &&
             !authSession->isAuthenticated()) {
-            return boost::none;
+            return {};
         }
     }
 
@@ -82,7 +81,7 @@ boost::optional<OperationSessionInfoFromClient> initializeOperationSessionInfo(
         if (!lsc) {
             // Ignore session information if the logical session cache has not been set up, e.g. on
             // the embedded version of mongod.
-            return boost::none;
+            return {};
         }
 
         opCtx->setLogicalSessionId(makeLogicalSessionId(osi.getSessionId().get(), opCtx));
@@ -133,6 +132,22 @@ boost::optional<OperationSessionInfoFromClient> initializeOperationSessionInfo(
                 osi.getStartTransaction().value());
     }
 
+    // Populate the session info for doTxn command.
+    if (requestBody.firstElementFieldName() == "doTxn"_sd) {
+        uassert(ErrorCodes::InvalidOptions,
+                "doTxn can only be run with a transaction number.",
+                osi.getTxnNumber());
+        uassert(ErrorCodes::OperationNotSupportedInTransaction,
+                "doTxn can not be run in a transaction",
+                !osi.getAutocommit());
+        // 'autocommit' and 'startTransaction' are populated for 'doTxn' to get the oplog
+        // entry generation behavior used for multi-document transactions. The 'doTxn'
+        // command still logically behaves as a commit.
+        osi.setAutocommit(false);
+        osi.setStartTransaction(true);
+    }
+
+
     return osi;
 }