fix: add group topn to executor factory (#655)

Signed-off-by: Alex Chi <[email protected]>

Author: skyzh

src/execution/executor_factory.cpp

@@ -32,13 +32,15 @@
 #include "execution/executors/sort_executor.h"
 #include "execution/executors/topn_check_executor.h"
 #include "execution/executors/topn_executor.h"
+#include "execution/executors/topn_per_group_executor.h"
 #include "execution/executors/update_executor.h"
 #include "execution/executors/values_executor.h"
 #include "execution/executors/window_function_executor.h"
 #include "execution/plans/filter_plan.h"
 #include "execution/plans/mock_scan_plan.h"
 #include "execution/plans/projection_plan.h"
 #include "execution/plans/sort_plan.h"
+#include "execution/plans/topn_per_group_plan.h"
 #include "execution/plans/topn_plan.h"
 #include "execution/plans/values_plan.h"
 #include "execution/plans/window_plan.h"
@@ -180,6 +182,13 @@ auto ExecutorFactory::CreateExecutor(ExecutorContext *exec_ctx, const AbstractPl
       return std::make_unique<TopNExecutor>(exec_ctx, topn_plan, std::move(child));
     }
 
+      // Create a new groupTopN executor
+    case PlanType::TopNPerGroup: {
+      const auto *group_topn_plan = dynamic_cast<const TopNPerGroupPlanNode *>(plan.get());
+      auto child = ExecutorFactory::CreateExecutor(exec_ctx, group_topn_plan->GetChildPlan());
+      return std::make_unique<TopNPerGroupExecutor>(exec_ctx, group_topn_plan, std::move(child));
+    }
+
     default:
       UNREACHABLE("Unsupported plan type.");
   }

src/include/execution/executors/topn_per_group_executor.h

@@ -48,6 +48,8 @@ class TopNPerGroupExecutor : public AbstractExecutor {
    */
   auto Next(Tuple *tuple, RID *rid) -> bool override;
 
+  auto GetOutputSchema() const -> const Schema & override { return plan_->OutputSchema(); }
+
  private:
   /** The TopNPerGroup plan node to be executed */
   [[maybe_unused]] const TopNPerGroupPlanNode *plan_;