Add missing reset for start iterator in case of neighbor node queries.

IntelLabs · May 24, 2018 · 6b44419 · 6b44419
1 parent f3ae132
commit 6b44419
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/PMGDQueryHandler.cc b/src/PMGDQueryHandler.cc
@@ -386,6 +386,8 @@ int PMGDQueryHandler::query_node(const protobufs::QueryNode &qn,
     if (!bool(ni)) {
         set_response(response, PMGDCmdResponse::Empty,
                        "Null search iterator\n");
+        if (has_link)
+            start_ni->reset();
         return -1;
     }
 
@@ -413,6 +415,8 @@ int PMGDQueryHandler::query_node(const protobufs::QueryNode &qn,
         if (bool(*tni)) {  // Not unique and that is an error here.
             set_response(response, PMGDCmdResponse::NotUnique,
                            "Query response not unique\n");
+            if (has_link)
+                start_ni->reset();
             delete tni;
             return -1;
         }