SERVER-10026 returnKey in new system

1 files changed, 3 insertions, 0 deletions

diff --git a/src/mongo/db/query/planner_access.cpp b/src/mongo/db/query/planner_access.cpp
index 6f28702931f..744f17be890 100644
--- a/src/mongo/db/query/planner_access.cpp
+++ b/src/mongo/db/query/planner_access.cpp
@@ -136,6 +136,7 @@ namespace mongo {
             isn->indexIsMultiKey = index.multikey;
             isn->bounds.fields.resize(index.keyPattern.nFields());
             isn->maxScan = query.getParsed().getMaxScan();
+            isn->addKeyMetadata = query.getParsed().returnKey();
 
             IndexBoundsBuilder::translate(expr, index.keyPattern.firstElement(),
                                           &isn->bounds.fields[0], tightnessOut);
@@ -841,6 +842,7 @@ namespace mongo {
         isn->indexIsMultiKey = index.multikey;
         isn->bounds.fields.resize(index.keyPattern.nFields());
         isn->maxScan = query.getParsed().getMaxScan();
+        isn->addKeyMetadata = query.getParsed().returnKey();
 
         // TODO: can we use simple bounds with this compound idx?
         BSONObjIterator it(isn->indexKeyPattern);
@@ -924,6 +926,7 @@ namespace mongo {
         isn->indexIsMultiKey = index.multikey;
         isn->direction = 1;
         isn->maxScan = query.getParsed().getMaxScan();
+        isn->addKeyMetadata = query.getParsed().returnKey();
         isn->bounds.isSimpleRange = true;
         isn->bounds.startKey = startKey;
         isn->bounds.endKey = endKey;