SERVER-30871 Permit blocking aggregation stages to run on mongoS if allowDiskUse is false

1 files changed, 8 insertions, 7 deletions

diff --git a/src/mongo/db/pipeline/document_source_sequential_document_cache.h b/src/mongo/db/pipeline/document_source_sequential_document_cache.h
index af96ae8e2ab..1d2474235f2 100644
--- a/src/mongo/db/pipeline/document_source_sequential_document_cache.h
+++ b/src/mongo/db/pipeline/document_source_sequential_document_cache.h
@@ -50,13 +50,14 @@ public:
     }
 
     StageConstraints constraints() const {
-        StageConstraints constraints;
-
-        if (_cache->isServing()) {
-            constraints.requiredPosition = PositionRequirement::kFirst;
-            constraints.requiresInputDocSource = false;
-        }
-
+        StageConstraints constraints(StreamType::kStreaming,
+                                     _cache->isServing() ? PositionRequirement::kFirst
+                                                         : PositionRequirement::kNone,
+                                     HostTypeRequirement::kNone,
+                                     DiskUseRequirement::kNoDiskUse,
+                                     FacetRequirement::kNotAllowed);
+
+        constraints.requiresInputDocSource = (_cache->isBuilding());
         return constraints;
     }