SERVER-28211 added InternalPlanner::updateWithIdHack()

1 files changed, 11 insertions, 0 deletions

diff --git a/src/mongo/db/query/internal_plans.h b/src/mongo/db/query/internal_plans.h
index f9da8e89f70..cc9eb7f0d86 100644
--- a/src/mongo/db/query/internal_plans.h
+++ b/src/mongo/db/query/internal_plans.h
@@ -41,6 +41,7 @@ class OperationContext;
 class PlanStage;
 class WorkingSet;
 struct DeleteStageParams;
+struct UpdateStageParams;
 
 /**
  * The internal planner is a one-stop shop for "off-the-shelf" plans.  Most internal procedures
@@ -110,6 +111,16 @@ public:
                                                              PlanExecutor::YieldPolicy yieldPolicy,
                                                              Direction direction = FORWARD);
 
+    /**
+     * Returns an IDHACK => UPDATE plan.
+     */
+    static std::unique_ptr<PlanExecutor> updateWithIdHack(OperationContext* opCtx,
+                                                          Collection* collection,
+                                                          const UpdateStageParams& params,
+                                                          const IndexDescriptor* descriptor,
+                                                          const BSONObj& key,
+                                                          PlanExecutor::YieldPolicy yieldPolicy);
+
 private:
     /**
      * Returns a plan stage that can be used for a collection scan.