Filter should keep column names. (#4825)

vesoft-inc · Nov 7, 2022 · e96774e · e96774e
1 parent 63a63ac
commit e96774e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/graph/optimizer/rule/PushFilterDownProjectRule.cpp b/src/graph/optimizer/rule/PushFilterDownProjectRule.cpp
@@ -110,6 +110,8 @@ StatusOr<OptRule::TransformResult> PushFilterDownProjectRule::transform(
   auto* newBelowFilterNode = graph::Filter::make(
       octx->qctx(), const_cast<graph::PlanNode*>(oldProjNode->dep()), newFilterPicked);
   newBelowFilterNode->setInputVar(oldProjNode->inputVar());
+  // Filter should keep column names
+  newBelowFilterNode->setColNames(oldProjNode->inputVars()[0]->colNames);
   auto newBelowFilterGroup = OptGroup::create(octx);
   auto newFilterGroupNode = newBelowFilterGroup->makeGroupNode(newBelowFilterNode);
   for (auto dep : projGroupNode->dependencies()) {