optimize for the more common case of single value facets when loading values
This commit is contained in:
parent
59ee8493ce
commit
d283c158d2
|
@ -40,11 +40,11 @@ public class FieldDataLoader {
|
|||
loader.init();
|
||||
|
||||
field = StringHelper.intern(field);
|
||||
int[][] ordinals = new int[reader.maxDoc()][];
|
||||
int[] ordinals = new int[reader.maxDoc()];
|
||||
int[][] multiValueOrdinals = null;
|
||||
|
||||
int t = 1; // current term number
|
||||
|
||||
boolean multiValued = false;
|
||||
TermDocs termDocs = reader.termDocs();
|
||||
TermEnum termEnum = reader.terms(new Term(field));
|
||||
try {
|
||||
|
@ -55,16 +55,43 @@ public class FieldDataLoader {
|
|||
termDocs.seek(termEnum);
|
||||
while (termDocs.next()) {
|
||||
int doc = termDocs.doc();
|
||||
int[] ordinalPerDoc = ordinals[doc];
|
||||
if (multiValueOrdinals != null) {
|
||||
int[] ordinalPerDoc = multiValueOrdinals[doc];
|
||||
if (ordinalPerDoc == null) {
|
||||
ordinalPerDoc = new int[1];
|
||||
ordinalPerDoc[0] = t;
|
||||
ordinals[doc] = ordinalPerDoc;
|
||||
multiValueOrdinals[doc] = ordinalPerDoc;
|
||||
} else {
|
||||
multiValued = true;
|
||||
ordinalPerDoc = Arrays.copyOf(ordinalPerDoc, ordinalPerDoc.length + 1);
|
||||
ordinalPerDoc[ordinalPerDoc.length - 1] = t;
|
||||
ordinals[doc] = ordinalPerDoc;
|
||||
multiValueOrdinals[doc] = ordinalPerDoc;
|
||||
}
|
||||
} else {
|
||||
int ordinal = ordinals[doc];
|
||||
if (ordinal == 0) { // still not multi valued...
|
||||
ordinals[doc] = t;
|
||||
} else {
|
||||
// move to multi valued
|
||||
multiValueOrdinals = new int[reader.maxDoc()][];
|
||||
for (int i = 0; i < ordinals.length; i++) {
|
||||
ordinal = ordinals[i];
|
||||
if (ordinal != 0) {
|
||||
multiValueOrdinals[i] = new int[1];
|
||||
multiValueOrdinals[i][0] = ordinal;
|
||||
}
|
||||
}
|
||||
// now put the current "t" value
|
||||
int[] ordinalPerDoc = multiValueOrdinals[doc];
|
||||
if (ordinalPerDoc == null) {
|
||||
ordinalPerDoc = new int[1];
|
||||
ordinalPerDoc[0] = t;
|
||||
multiValueOrdinals[doc] = ordinalPerDoc;
|
||||
} else {
|
||||
ordinalPerDoc = Arrays.copyOf(ordinalPerDoc, ordinalPerDoc.length + 1);
|
||||
ordinalPerDoc[ordinalPerDoc.length - 1] = t;
|
||||
multiValueOrdinals[doc] = ordinalPerDoc;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -81,17 +108,10 @@ public class FieldDataLoader {
|
|||
termEnum.close();
|
||||
}
|
||||
|
||||
if (multiValued) {
|
||||
return loader.buildMultiValue(field, ordinals);
|
||||
if (multiValueOrdinals != null) {
|
||||
return loader.buildMultiValue(field, multiValueOrdinals);
|
||||
} else {
|
||||
// optimize for a single valued
|
||||
int[] sOrders = new int[reader.maxDoc()];
|
||||
for (int i = 0; i < ordinals.length; i++) {
|
||||
if (ordinals[i] != null) {
|
||||
sOrders[i] = ordinals[i][0];
|
||||
}
|
||||
}
|
||||
return loader.buildSingleValue(field, sOrders);
|
||||
return loader.buildSingleValue(field, ordinals);
|
||||
}
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in New Issue