1 files changed, 6 insertions, 2 deletions
diff --git a/src/mongo/db/query/multi_plan_runner.h b/src/mongo/db/query/multi_plan_runner.h
index c96121987e2..14ad8507bf3 100644
--- a/src/mongo/db/query/multi_plan_runner.h
+++ b/src/mongo/db/query/multi_plan_runner.h
@@ -67,10 +67,14 @@ namespace mongo {
          */
         bool pickBestPlan(size_t* out);
 
-        virtual void yield();
-        virtual void unYield();
+        virtual void saveState();
+        virtual void restoreState();
         virtual void invalidate(const DiskLoc& dl);
 
+        virtual const CanonicalQuery& getQuery() {
+            return *_query;
+        }
+
     private:
         /**
          * Have all our candidate plans do something.