Skip to content

Commit d95db48

Browse files
committed
Properly implement advanceExact()
1 parent cca6e39 commit d95db48

File tree

2 files changed

+2
-6
lines changed

2 files changed

+2
-6
lines changed

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

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,6 @@
1414
import org.apache.lucene.index.LeafReader;
1515
import org.apache.lucene.index.LeafReaderContext;
1616
import org.apache.lucene.index.VectorSimilarityFunction;
17-
import org.apache.lucene.search.DocIdSetIterator;
1817
import org.apache.lucene.search.DoubleValues;
1918
import org.apache.lucene.search.DoubleValuesSource;
2019
import org.apache.lucene.search.IndexSearcher;
@@ -56,8 +55,7 @@ public double doubleValue() throws IOException {
5655

5756
@Override
5857
public boolean advanceExact(int doc) throws IOException {
59-
docId = doc;
60-
return iterator.advance(docId) != DocIdSetIterator.NO_MORE_DOCS;
58+
return iterator.advance(doc) == doc;
6159
}
6260
};
6361
}

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

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,6 @@
1414
import org.apache.lucene.index.LeafReader;
1515
import org.apache.lucene.index.LeafReaderContext;
1616
import org.apache.lucene.index.VectorSimilarityFunction;
17-
import org.apache.lucene.search.DocIdSetIterator;
1817
import org.apache.lucene.search.DoubleValues;
1918
import org.apache.lucene.search.DoubleValuesSource;
2019
import org.apache.lucene.search.IndexSearcher;
@@ -56,8 +55,7 @@ public double doubleValue() throws IOException {
5655

5756
@Override
5857
public boolean advanceExact(int doc) throws IOException {
59-
docId = doc;
60-
return iterator.advance(docId) != DocIdSetIterator.NO_MORE_DOCS;
58+
return iterator.advance(doc) == doc;
6159
}
6260
};
6361
}

0 commit comments

Comments
 (0)