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

Indicate frontier init length.

This commit is contained in:
zhouhui 2024-10-15 17:31:19 +08:00
parent 6e9c431f95
commit 912674af66
2 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
lucene/core/src/java/org/apache/lucene/codecs/lucene90/blocktree/Lucene90BlockTreeTermsWriter.java
View File

@ -545,6 +545,7 @@ public final class Lucene90BlockTreeTermsWriter extends FieldsConsumer {
.suffixRAMLimitMB(0d)
.dataOutput(getOnHeapReaderWriter(pageBits))
.setVersion(fstVersion)
.setInitLength(prefix.length + 1)
.build();
// if (DEBUG) {
// System.out.println(" compile index for prefix=" + prefix);

14
lucene/core/src/java/org/apache/lucene/util/fst/FSTCompiler.java
View File

@ -164,7 +164,8 @@ public class FSTCompiler<T> {
boolean allowFixedLengthArcs,
DataOutput dataOutput,
float directAddressingMaxOversizingFactor,
int version) {
int version,
int initLength) {
this.allowFixedLengthArcs = allowFixedLengthArcs;
this.directAddressingMaxOversizingFactor = directAddressingMaxOversizingFactor;
this.version = version;
@ -185,7 +186,7 @@ public class FSTCompiler<T> {
NO_OUTPUT = outputs.getNoOutput();
@SuppressWarnings({"rawtypes", "unchecked"})
final UnCompiledNode<T>[] f = (UnCompiledNode<T>[]) new UnCompiledNode[10];
final UnCompiledNode<T>[] f = (UnCompiledNode<T>[]) new UnCompiledNode[initLength];
frontier = f;
for (int idx = 0; idx < frontier.length; idx++) {
frontier[idx] = new UnCompiledNode<>(this, idx);
@ -247,6 +248,7 @@ public class FSTCompiler<T> {
private DataOutput dataOutput;
private float directAddressingMaxOversizingFactor = DIRECT_ADDRESSING_MAX_OVERSIZING_FACTOR;
private int version = FST.VERSION_CURRENT;
private int initLength = 10;
/**
* @param inputType The input type (transition labels). Can be anything from {@link INPUT_TYPE}
@ -346,6 +348,11 @@ public class FSTCompiler<T> {
return this;
}
public Builder<T> setInitLength(int initLength) {
this.initLength = initLength;
return this;
}
/** Creates a new {@link FSTCompiler}. */
public FSTCompiler<T> build() {
// create a default DataOutput if not specified
@ -359,7 +366,8 @@ public class FSTCompiler<T> {
allowFixedLengthArcs,
dataOutput,
directAddressingMaxOversizingFactor,
version);
version,
initLength);
}
}