Add NearestNeighborQueryNode.

1 files changed, 20 insertions, 0 deletions

diff --git a/searchlib/src/vespa/searchlib/query/streaming/querynode.cpp b/searchlib/src/vespa/searchlib/query/streaming/querynode.cpp
index 6d59886a4f5..226cb92c894 100644
--- a/searchlib/src/vespa/searchlib/query/streaming/querynode.cpp
+++ b/searchlib/src/vespa/searchlib/query/streaming/querynode.cpp
@@ -1,6 +1,7 @@
 // Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
 
 #include "query.h"
+#include "nearest_neighbor_query_node.h"
 #include <vespa/searchlib/parsequery/stackdumpiterator.h>
 #include <charconv>
 #include <vespa/log/log.h>
@@ -77,6 +78,9 @@ QueryNode::Build(const QueryNode * parent, const QueryNodeResultFactory & factor
                                          queryRep.getIndexName(),
                                          QueryTerm::Type::GEO_LOCATION);
         break;
+    case ParseItem::ITEM_NEAREST_NEIGHBOR:
+        qn = build_nearest_neighbor_query_node(factory, queryRep);
+        break;
     case ParseItem::ITEM_NUMTERM:
     case ParseItem::ITEM_TERM:
     case ParseItem::ITEM_PREFIXTERM:
@@ -191,4 +195,20 @@ const HitList & QueryNode::evaluateHits(HitList & hl) const
     return hl;
 }
 
+std::unique_ptr<QueryNode>
+QueryNode::build_nearest_neighbor_query_node(const QueryNodeResultFactory& factory, SimpleQueryStackDumpIterator& query_rep)
+{
+    vespalib::stringref query_tensor_name = query_rep.getTerm();
+    vespalib::stringref field_name = query_rep.getIndexName();
+    int32_t id = query_rep.getUniqueId();
+    search::query::Weight weight = query_rep.GetWeight();
+    double distance_threshold = query_rep.getDistanceThreshold();
+    return std::make_unique<NearestNeighborQueryNode>(factory.create(),
+                                                      query_tensor_name,
+                                                      field_name,
+                                                      id,
+                                                      weight,
+                                                      distance_threshold);
+}
+
 }