SERVER-17280 geo near must propagate saveState()/restoreState() to the DensityEstimator

1 files changed, 10 insertions, 5 deletions

diff --git a/src/mongo/db/exec/near.h b/src/mongo/db/exec/near.h
index 60384679c00..d7d27a78331 100644
--- a/src/mongo/db/exec/near.h
+++ b/src/mongo/db/exec/near.h
@@ -147,15 +147,20 @@ namespace mongo {
                                       Collection* collection,
                                       WorkingSetID* out) = 0;
 
-        /**
-         * Does any invalidation work specific to the search type.
-         */
+    private:
+
+        //
+        // Save/restore/invalidate work specific to the search type.
+        //
+
+        virtual void finishSaveState() = 0;
+
+        virtual void finishRestoreState(OperationContext* txn) = 0;
+
         virtual void finishInvalidate(OperationContext* txn,
                                       const RecordId& dl,
                                       InvalidationType type) = 0;
 
-    private:
-
         //
         // Generic methods for progressive search functionality
         //