Class ESKnnFloatVectorQuery

java.lang.Object
org.apache.lucene.search.Query
org.apache.lucene.search.KnnFloatVectorQuery
org.elasticsearch.search.vectors.ESKnnFloatVectorQuery
All Implemented Interfaces:
QueryProfilerProvider

public class ESKnnFloatVectorQuery extends org.apache.lucene.search.KnnFloatVectorQuery implements QueryProfilerProvider
  • Field Summary

    Fields
    Modifier and Type
    Field
    Description
    protected final String
     
    protected final org.apache.lucene.search.Query
     
    protected final int
     
    protected final org.apache.lucene.search.knn.KnnSearchStrategy
     

    Fields inherited from class org.apache.lucene.search.KnnFloatVectorQuery

    target
  • Constructor Summary

    Constructors
    Constructor
    Description
    ESKnnFloatVectorQuery(String field, float[] target, int k, int numCands, org.apache.lucene.search.Query filter, org.apache.lucene.search.knn.KnnSearchStrategy strategy)
     
    ESKnnFloatVectorQuery(String field, float[] target, int k, int numCands, org.apache.lucene.search.Query filter, org.apache.lucene.search.knn.KnnSearchStrategy strategy, boolean earlyTermination)
     
  • Method Summary

    Modifier and Type
    Method
    Description
    protected org.apache.lucene.search.TopDocs
    exactSearch(org.apache.lucene.index.LeafReaderContext arg0, org.apache.lucene.search.DocIdSetIterator arg1, org.apache.lucene.index.QueryTimeout arg2)
     
     
    org.apache.lucene.search.Query
     
    int
     
    protected org.apache.lucene.search.knn.KnnCollectorManager
    getKnnCollectorManager(int k, org.apache.lucene.search.IndexSearcher searcher)
     
    org.apache.lucene.search.knn.KnnSearchStrategy
     
    org.apache.lucene.search.knn.KnnSearchStrategy
     
    int
     
    protected org.apache.lucene.search.TopDocs
    mergeLeafResults(org.apache.lucene.search.TopDocs[] perLeafResults)
     
    void
    profile(QueryProfiler queryProfiler)
    Store the profiling information in the QueryProfiler
    org.apache.lucene.search.Query
    rewrite(org.apache.lucene.search.IndexSearcher arg0)
     
    protected org.apache.lucene.search.TopDocs
    searchLeaf(org.apache.lucene.index.LeafReaderContext arg0, org.apache.lucene.search.Weight arg1, org.apache.lucene.search.TimeLimitingKnnCollectorManager arg2)
     
    void
    visit(org.apache.lucene.search.QueryVisitor arg0)
     

    Methods inherited from class org.apache.lucene.search.KnnFloatVectorQuery

    approximateSearch, equals, getTargetCopy, hashCode, toString

    Methods inherited from class org.apache.lucene.search.Query

    classHash, createWeight, sameClassAs, toString

    Methods inherited from class java.lang.Object

    clone, finalize, getClass, notify, notifyAll, wait, wait, wait
  • Field Details

    • field

      protected final String field
    • k

      protected final int k
    • filter

      protected final org.apache.lucene.search.Query filter
    • searchStrategy

      protected final org.apache.lucene.search.knn.KnnSearchStrategy searchStrategy
  • Constructor Details

    • ESKnnFloatVectorQuery

      public ESKnnFloatVectorQuery(String field, float[] target, int k, int numCands, org.apache.lucene.search.Query filter, org.apache.lucene.search.knn.KnnSearchStrategy strategy)
    • ESKnnFloatVectorQuery

      public ESKnnFloatVectorQuery(String field, float[] target, int k, int numCands, org.apache.lucene.search.Query filter, org.apache.lucene.search.knn.KnnSearchStrategy strategy, boolean earlyTermination)
  • Method Details

    • mergeLeafResults

      protected org.apache.lucene.search.TopDocs mergeLeafResults(org.apache.lucene.search.TopDocs[] perLeafResults)
    • profile

      public void profile(QueryProfiler queryProfiler)
      Description copied from interface: QueryProfilerProvider
      Store the profiling information in the QueryProfiler
      Specified by:
      profile in interface QueryProfilerProvider
      Parameters:
      queryProfiler - an instance of KnnFloatVectorField.
    • kParam

      public int kParam()
    • getStrategy

      public org.apache.lucene.search.knn.KnnSearchStrategy getStrategy()
    • getKnnCollectorManager

      protected org.apache.lucene.search.knn.KnnCollectorManager getKnnCollectorManager(int k, org.apache.lucene.search.IndexSearcher searcher)
    • rewrite

      public org.apache.lucene.search.Query rewrite(org.apache.lucene.search.IndexSearcher arg0) throws IOException
      Overrides:
      rewrite in class org.apache.lucene.search.Query
      Throws:
      IOException
    • searchLeaf

      protected org.apache.lucene.search.TopDocs searchLeaf(org.apache.lucene.index.LeafReaderContext arg0, org.apache.lucene.search.Weight arg1, org.apache.lucene.search.TimeLimitingKnnCollectorManager arg2) throws IOException
      Throws:
      IOException
    • exactSearch

      protected org.apache.lucene.search.TopDocs exactSearch(org.apache.lucene.index.LeafReaderContext arg0, org.apache.lucene.search.DocIdSetIterator arg1, org.apache.lucene.index.QueryTimeout arg2) throws IOException
      Throws:
      IOException
    • visit

      public void visit(org.apache.lucene.search.QueryVisitor arg0)
      Specified by:
      visit in class org.apache.lucene.search.Query
    • getField

      public String getField()
    • getK

      public int getK()
    • getFilter

      public org.apache.lucene.search.Query getFilter()
    • getSearchStrategy

      public org.apache.lucene.search.knn.KnnSearchStrategy getSearchStrategy()