SERVER-33551 Switch Pipeline to use session interface to determine if in a transaction or snapshot read.

1 files changed, 1 insertions, 4 deletions

diff --git a/src/mongo/db/pipeline/pipeline.cpp b/src/mongo/db/pipeline/pipeline.cpp
index fafcd0d910f..2f6fce9e526 100644
--- a/src/mongo/db/pipeline/pipeline.cpp
+++ b/src/mongo/db/pipeline/pipeline.cpp
@@ -200,9 +200,6 @@ void Pipeline::validateFacetPipeline() const {
 }
 
 void Pipeline::validateCommon() const {
-    // TODO SERVER-33551: Don't use presence of WUOW to decide whether we are in a snapshot read or
-    // multi-doc transaction.
-    const bool isSnapshotReadOrTxn = static_cast<bool>(pCtx->opCtx->getWriteUnitOfWork());
     size_t i = 0;
     for (auto&& stage : _sources) {
         auto constraints = stage->constraints(_splitState);
@@ -231,7 +228,7 @@ void Pipeline::validateCommon() const {
                 str::stream() << stage->getSourceName() << " can only be run on mongoS",
                 !(constraints.hostRequirement == HostTypeRequirement::kMongoS && !pCtx->inMongos));
 
-        if (isSnapshotReadOrTxn) {
+        if (pCtx->inSnapshotReadOrMultiDocumentTransaction) {
             uassert(50742,
                     str::stream() << "Stage not supported with readConcern level \"snapshot\" "
                                      "or inside of a multi-document transaction: "