fix match node label (#5165)

* filter out initvid * Update MatchNodeLabelFilter.feature Co-authored-by: codesigner <[email protected]>
vesoft-inc · Dec 30, 2022 · 9eca7a2 · 9eca7a2
1 parent 51ff615
commit 9eca7a2
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 23 deletions.
diff --git a/src/graph/executor/query/TraverseExecutor.cpp b/src/graph/executor/query/TraverseExecutor.cpp
@@ -169,30 +169,20 @@ folly::Future<Status> TraverseExecutor::handleResponse(RpcResponse&& resps) {
     // match (v)-[e:Rel]-(v1:Label1)-[e1*2]->() where id(v0) in [6, 23] return v1
     // the attributes of v1 will be obtained in the second traverse operator
     // If the conditions are not met, the path in the previous step needs to be filtered out
-    if (!dst2PathsMap_.empty()) {
-      std::unordered_set<Value, VertexHash, VertexEqual> existVids;
-      existVids.reserve(iter->numRows());
-      auto vertices = iter->getVertices();
-      for (auto& vertex : vertices.values) {
-        if (vertex.isVertex()) {
-          existVids.emplace(vertex);
-        }
-      }
-      auto dst2PathIter = dst2PathsMap_.begin();
-      while (dst2PathIter != dst2PathsMap_.end()) {
-        if (existVids.find(dst2PathIter->first) == existVids.end()) {
-          dst2PathIter = dst2PathsMap_.erase(dst2PathIter);
-        } else {
-          dst2PathIter++;
-        }
+    std::unordered_set<Value, VertexHash, VertexEqual> existVids;
+    existVids.reserve(iter->numRows());
+    auto vertices = iter->getVertices();
+    for (auto& vertex : vertices.values) {
+      if (vertex.isVertex()) {
+        existVids.emplace(vertex);
       }
-      auto initVidIter = initVids_.begin();
-      while (initVidIter != initVids_.end()) {
-        if (existVids.find(*initVidIter) == existVids.end()) {
-          initVidIter = initVids_.erase(initVidIter);
-        } else {
-          initVidIter++;
-        }
+    }
+    auto initVidIter = initVids_.begin();
+    while (initVidIter != initVids_.end()) {
+      if (existVids.find(*initVidIter) == existVids.end()) {
+        initVidIter = initVids_.erase(initVidIter);
+      } else {
+        initVidIter++;
       }
     }
   }

diff --git a/tests/tck/features/bugfix/MatchNodeLabelFilter.feature b/tests/tck/features/bugfix/MatchNodeLabelFilter.feature
@@ -51,3 +51,12 @@ Feature: match node label filter
     Then the result should be, in any order:
       | labels(v1)             | count(*) |
       | ["bachelor", "player"] | 34       |
+    When executing query:
+      """
+      MATCH (v:bachelor)<-[e*2..2]-()<-[e1]-()
+      WHERE id(v) in ['Tony Parker', 'Spurs', 'Tim Duncan']
+      return labels(v), count(*)
+      """
+    Then the result should be, in any order:
+      | labels(v)              | count(*) |
+      | ["bachelor", "player"] | 184      |