Correctly implement profiling. Rename ProfilingQuery to QueryProfilerProvider

Author: carlosdelest

server/src/main/java/org/elasticsearch/index/mapper/vectors/VectorSimilarityByteValueSource.java

@@ -18,6 +18,8 @@
 import org.apache.lucene.search.DoubleValues;
 import org.apache.lucene.search.DoubleValuesSource;
 import org.apache.lucene.search.IndexSearcher;
+import org.elasticsearch.search.profile.query.QueryProfiler;
+import org.elasticsearch.search.vectors.QueryProfilerProvider;
 
 import java.io.IOException;
 import java.util.Arrays;
@@ -27,11 +29,12 @@
  * DoubleValuesSource that is used to calculate scores according to a similarity function for a KnnByteVectorField, using the
  * original vector values stored in the index
  */
-public class VectorSimilarityByteValueSource extends DoubleValuesSource {
+public class VectorSimilarityByteValueSource extends DoubleValuesSource implements QueryProfilerProvider {
 
     private final String field;
     private final byte[] target;
     private final VectorSimilarityFunction vectorSimilarityFunction;
+    private long vectorOpsCount;
 
     public VectorSimilarityByteValueSource(String field, byte[] target, VectorSimilarityFunction vectorSimilarityFunction) {
         this.field = field;
@@ -51,6 +54,7 @@ public DoubleValues getValues(LeafReaderContext ctx, DoubleValues scores) throws
 
             @Override
             public double doubleValue() throws IOException {
+                vectorOpsCount++;
                 return vectorSimilarityFunction.compare(target, vectorValues.vectorValue(docId));
             }
 
@@ -72,6 +76,11 @@ public DoubleValuesSource rewrite(IndexSearcher reader) throws IOException {
         return this;
     }
 
+    @Override
+    public void profile(QueryProfiler queryProfiler) {
+        queryProfiler.addVectorOpsCount(vectorOpsCount);
+    }
+
     @Override
     public int hashCode() {
         return Objects.hash(field, Arrays.hashCode(target), vectorSimilarityFunction);