fix tagless (#4652)

* fix tagless * fix test Co-authored-by: Sophie <[email protected]>
vesoft-inc · Sep 19, 2022 · 480b29a · 480b29a
1 parent c4cd891
commit 480b29a
Show file tree

Hide file tree

Showing 2 changed files with 57 additions and 53 deletions.
diff --git a/src/graph/visitor/PrunePropertiesVisitor.cpp b/src/graph/visitor/PrunePropertiesVisitor.cpp
@@ -358,11 +358,13 @@ void PrunePropertiesVisitor::pruneCurrent(AppendVertices *node) {
       node->setVertexProps(nullptr);
     } else {
       // only get _tag when props is nullptr
-      auto tagId = vertexProps->front().tag_ref().value();
-      VertexProp newVProp;
-      newVProp.tag_ref() = tagId;
-      newVProp.props_ref() = {nebula::kTag};
-      prunedVertexProps->emplace_back(std::move(newVProp));
+      for (auto &vertexProp : *vertexProps) {
+        auto tagId = vertexProp.tag_ref().value();
+        VertexProp newVProp;
+        newVProp.tag_ref() = tagId;
+        newVProp.props_ref() = {nebula::kTag};
+        prunedVertexProps->emplace_back(std::move(newVProp));
+      }
       node->setVertexProps(std::move(prunedVertexProps));
     }
     return;
@@ -373,11 +375,13 @@ void PrunePropertiesVisitor::pruneCurrent(AppendVertices *node) {
       node->setVertexProps(nullptr);
     } else {
       // only get _tag when props is nullptr
-      auto tagId = vertexProps->front().tag_ref().value();
-      VertexProp newVProp;
-      newVProp.tag_ref() = tagId;
-      newVProp.props_ref() = {nebula::kTag};
-      prunedVertexProps->emplace_back(std::move(newVProp));
+      for (auto &vertexProp : *vertexProps) {
+        auto tagId = vertexProp.tag_ref().value();
+        VertexProp newVProp;
+        newVProp.tag_ref() = tagId;
+        newVProp.props_ref() = {nebula::kTag};
+        prunedVertexProps->emplace_back(std::move(newVProp));
+      }
       node->setVertexProps(std::move(prunedVertexProps));
     }
     return;