SERVER-41404: [FLE] Accessors needed for $graphLookup

1 files changed, 20 insertions, 0 deletions

diff --git a/src/mongo/db/pipeline/document_source_graph_lookup.h b/src/mongo/db/pipeline/document_source_graph_lookup.h
index 535e630acc0..6f47af20c9f 100644
--- a/src/mongo/db/pipeline/document_source_graph_lookup.h
+++ b/src/mongo/db/pipeline/document_source_graph_lookup.h
@@ -46,6 +46,26 @@ public:
 
     const char* getSourceName() const final;
 
+    const FieldPath& getConnectFromField() const {
+        return _connectFromField;
+    }
+
+    const FieldPath& getConnectToField() const {
+        return _connectToField;
+    }
+
+    Expression* getStartWithField() const {
+        return _startWith.get();
+    }
+
+    boost::optional<BSONObj> getAdditionalFilter() const {
+        return _additionalFilter;
+    };
+
+    void setAdditionalFilter(boost::optional<BSONObj> additionalFilter) {
+        _additionalFilter = additionalFilter ? additionalFilter->getOwned() : additionalFilter;
+    };
+
     void serializeToArray(
         std::vector<Value>& array,
         boost::optional<ExplainOptions::Verbosity> explain = boost::none) const final;