Apache
/
lucene
mirror of https://github.com/apache/lucene.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Remove BytesRef usage from SortingCodeReader 

Follow-up of #12102
This commit is contained in:
Luca Cavanna 2023-01-23 22:27:13 +01:00
parent 4594400216
commit f5bd28662f
1 changed files with 4 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
lucene/core/src/java/org/apache/lucene/index/SortingCodecReader.java
View File

@ -34,9 +34,7 @@ import org.apache.lucene.codecs.TermVectorsReader;
import org.apache.lucene.search.Sort; import org.apache.lucene.search.Sort;
import org.apache.lucene.search.SortField; import org.apache.lucene.search.SortField;
import org.apache.lucene.search.TopDocs; import org.apache.lucene.search.TopDocs;
import org.apache.lucene.util.ArrayUtil;
import org.apache.lucene.util.Bits; import org.apache.lucene.util.Bits;
import org.apache.lucene.util.BytesRef;
import org.apache.lucene.util.FixedBitSet; import org.apache.lucene.util.FixedBitSet;
import org.apache.lucene.util.IOSupplier; import org.apache.lucene.util.IOSupplier;
import org.apache.lucene.util.packed.PackedInts; import org.apache.lucene.util.packed.PackedInts;
@ -273,7 +271,7 @@ public final class SortingCodecReader extends FilterCodecReader {
final int size; final int size;
final int dimension; final int dimension;
final FixedBitSet docsWithField; final FixedBitSet docsWithField;
final BytesRef[] binaryVectors; final byte[][] vectors;
private int docId = -1; private int docId = -1;
@ -281,13 +279,11 @@ public final class SortingCodecReader extends FilterCodecReader {
this.size = delegate.size(); this.size = delegate.size();
this.dimension = delegate.dimension(); this.dimension = delegate.dimension();
docsWithField = new FixedBitSet(sortMap.size()); docsWithField = new FixedBitSet(sortMap.size());
binaryVectors = new BytesRef[sortMap.size()]; vectors = new byte[sortMap.size()][];
for (int doc = delegate.nextDoc(); doc != NO_MORE_DOCS; doc = delegate.nextDoc()) { for (int doc = delegate.nextDoc(); doc != NO_MORE_DOCS; doc = delegate.nextDoc()) {
int newDocID = sortMap.oldToNew(doc); int newDocID = sortMap.oldToNew(doc);
docsWithField.set(newDocID); docsWithField.set(newDocID);
binaryVectors[newDocID] = vectors[newDocID] = delegate.vectorValue().clone();
new BytesRef(
ArrayUtil.copyOfSubArray(delegate.vectorValue(), 0, delegate.vectorValue().length));
} }
} }
@ -303,7 +299,7 @@ public final class SortingCodecReader extends FilterCodecReader {
@Override @Override
public byte[] vectorValue() throws IOException { public byte[] vectorValue() throws IOException {
return binaryVectors[docId].bytes; return vectors[docId];
} }
@Override @Override