SERVER-31447 Use correct collation for update lookup

1 files changed, 8 insertions, 6 deletions

diff --git a/src/mongo/db/pipeline/document_source.h b/src/mongo/db/pipeline/document_source.h
index 2f6bec9fbcc..0ecd0fcf92a 100644
--- a/src/mongo/db/pipeline/document_source.h
+++ b/src/mongo/db/pipeline/document_source.h
@@ -841,13 +841,15 @@ public:
         virtual std::vector<FieldPath> collectDocumentKeyFields(UUID) const = 0;
 
         /**
-         * Returns zero or one documents matching the input filter, or throws if more than one match
-         * was found. The passed ExpressionContext may use a different namespace than the
-         * ExpressionContext used to construct the MongoProcessInterface. Returns boost::none if no
-         * matching documents were found, including cases where the given namespace does not exist.
+         * Returns zero or one documents with the document key 'documentKey'. 'documentKey' is
+         * treated as a unique identifier of a document, and may include an _id or all fields from
+         * the shard key and an _id. Throws if more than one match was found. Returns boost::none if
+         * no matching documents were found, including cases where the given namespace does not
+         * exist.
          */
-        virtual boost::optional<Document> lookupSingleDocument(
-            const boost::intrusive_ptr<ExpressionContext>& expCtx, const Document& filter) = 0;
+        virtual boost::optional<Document> lookupSingleDocument(const NamespaceString& nss,
+                                                               UUID collectionUUID,
+                                                               const Document& documentKey) = 0;
 
         /**
          * Returns a vector of all local cursors.