1111
1212import org .apache .lucene .codecs .hnsw .FlatVectorsReader ;
1313import org .apache .lucene .index .ByteVectorValues ;
14+ import org .apache .lucene .index .FieldInfo ;
1415import org .apache .lucene .index .FloatVectorValues ;
1516import org .apache .lucene .search .KnnCollector ;
1617import org .apache .lucene .util .Accountable ;
2021
2122import java .io .IOException ;
2223import java .util .Collection ;
24+ import java .util .Map ;
2325
2426class MergeReaderWrapper extends FlatVectorsReader {
2527
@@ -57,6 +59,16 @@ public ByteVectorValues getByteVectorValues(String field) throws IOException {
5759 return mainReader .getByteVectorValues (field );
5860 }
5961
62+ @ Override
63+ public void search (String field , float [] target , KnnCollector knnCollector , Bits acceptDocs ) throws IOException {
64+ mainReader .search (field , target , knnCollector , acceptDocs );
65+ }
66+
67+ @ Override
68+ public void search (String field , byte [] target , KnnCollector knnCollector , Bits acceptDocs ) throws IOException {
69+ mainReader .search (field , target , knnCollector , acceptDocs );
70+ }
71+
6072 @ Override
6173 public FlatVectorsReader getMergeInstance () {
6274 return mergeReader ;
@@ -73,13 +85,8 @@ public Collection<Accountable> getChildResources() {
7385 }
7486
7587 @ Override
76- public void search (String field , float [] target , KnnCollector knnCollector , Bits acceptDocs ) throws IOException {
77- mainReader .search (field , target , knnCollector , acceptDocs );
78- }
79-
80- @ Override
81- public void search (String field , byte [] target , KnnCollector knnCollector , Bits acceptDocs ) throws IOException {
82- mainReader .search (field , target , knnCollector , acceptDocs );
88+ public Map <String , Long > getOffHeapByteSize (FieldInfo fieldInfo ) {
89+ return mainReader .getOffHeapByteSize (fieldInfo );
8390 }
8491
8592 @ Override
0 commit comments