fix removeproject error

vesoft-inc · Jul 25, 2022 · 3bd1d70 · 3bd1d70
1 parent 82118e8
commit 3bd1d70
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 3 deletions.
diff --git a/src/graph/optimizer/rule/RemoveNoopProjectRule.cpp b/src/graph/optimizer/rule/RemoveNoopProjectRule.cpp
@@ -55,6 +55,7 @@ namespace opt {
     PlanNode::Kind::kSubgraph,
     PlanNode::Kind::kDataCollect,
     PlanNode::Kind::kLeftJoin,
+    PlanNode::Kind::kUnwind,
     PlanNode::Kind::kInnerJoin,
     PlanNode::Kind::kBiLeftJoin,
     PlanNode::Kind::kBiInnerJoin,

diff --git a/src/graph/planner/plan/Query.cpp b/src/graph/planner/plan/Query.cpp
@@ -369,6 +369,7 @@ std::unique_ptr<PlanNodeDescription> Unwind::explain() const {
 
 PlanNode* Unwind::clone() const {
   auto* newUnwind = Unwind::make(qctx_, nullptr);
+  newUnwind->setFromPipe(fromPipe_);
   newUnwind->cloneMembers(*this);
   return newUnwind;
 }

diff --git a/src/graph/planner/plan/Query.h b/src/graph/planner/plan/Query.h
@@ -845,8 +845,8 @@ class Unwind final : public SingleInputNode {
     return fromPipe_;
   }
 
-  void setFromPipe() {
-    fromPipe_ = true;
+  void setFromPipe(bool fromPipe) {
+    fromPipe_ = fromPipe;
   }
 
   PlanNode* clone() const override;

diff --git a/src/graph/validator/UnwindValidator.cpp b/src/graph/validator/UnwindValidator.cpp
@@ -41,7 +41,7 @@ Status UnwindValidator::validateImpl() {
 Status UnwindValidator::toPlan() {
   auto *unwind = Unwind::make(qctx_, nullptr, unwindExpr_, alias_);
   unwind->setColNames({alias_});
-  unwind->setFromPipe();
+  unwind->setFromPipe(true);
   root_ = tail_ = unwind;
   return Status::OK();
 }