SERVER-14633 rename prepareToYield() and recoverFromYield() to saveState() and restoreState()

Also renames PlanExecutor::getStages() to PlanExecutor::getRootStage()

1 files changed, 3 insertions, 3 deletions

diff --git a/src/mongo/db/commands/pipeline_command.cpp b/src/mongo/db/commands/pipeline_command.cpp
index 36a9a7c75a8..20d05a86e90 100644
--- a/src/mongo/db/commands/pipeline_command.cpp
+++ b/src/mongo/db/commands/pipeline_command.cpp
@@ -116,10 +116,10 @@ namespace {
 
         // Manage our OperationContext. We intentionally don't propagate to the child
         // Runner as that is handled by DocumentSourceCursor as it needs to.
-        virtual void prepareToYield() {
+        virtual void saveState() {
             _pipeline->getContext()->opCtx = NULL;
         }
-        virtual void recoverFromYield(OperationContext* opCtx) {
+        virtual void restoreState(OperationContext* opCtx) {
             _pipeline->getContext()->opCtx = opCtx;
         }
 
@@ -234,7 +234,7 @@ namespace {
 
             if (resultsArray.len() + next.objsize() > byteLimit) {
                 // Get the pipeline proxy stage wrapped by this PlanExecutor.
-                PipelineProxyStage* proxy = static_cast<PipelineProxyStage*>(exec->getStages());
+                PipelineProxyStage* proxy = static_cast<PipelineProxyStage*>(exec->getRootStage());
                 // too big. next will be the first doc in the second batch
                 proxy->pushBack(next);
                 break;