honeymoose
/
OpenSearch
mirror of https://github.com/honeymoose/OpenSearch.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Optimize conversion to a cacheable DocIdSet

This commit is contained in:
Igor Motov 2013-02-13 21:04:54 -05:00
parent 883c593d7e
commit 6b49457d9d
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/main/java/org/elasticsearch/common/lucene/docset/DocIdSets.java
View File

@ -82,7 +82,12 @@ public class DocIdSets {
if (set instanceof FixedBitSet) { if (set instanceof FixedBitSet) {
return set; return set;
} }
return toFixedBitSet(set.iterator(), reader.maxDoc()); FixedBitSet fixedBitSet = new FixedBitSet(reader.maxDoc());
do {
fixedBitSet.set(doc);
doc = it.nextDoc();
} while (doc != DocIdSetIterator.NO_MORE_DOCS);
return fixedBitSet;
} }
/** /**
@ -100,16 +105,16 @@ public class DocIdSets {
if (iterator == null) { if (iterator == null) {
return new Bits.MatchNoBits(reader.maxDoc()); return new Bits.MatchNoBits(reader.maxDoc());
} }
return new FixedBitSet(toFixedBitSet(iterator, reader.maxDoc())); return toFixedBitSet(iterator, reader.maxDoc());
} }
/** /**
* Creates a {@link FixedBitSet} from an iterator. * Creates a {@link FixedBitSet} from an iterator.
*/ */
public static FixedBitSet toFixedBitSet(DocIdSetIterator iteartor, int numBits) throws IOException { public static FixedBitSet toFixedBitSet(DocIdSetIterator iterator, int numBits) throws IOException {
FixedBitSet set = new FixedBitSet(numBits); FixedBitSet set = new FixedBitSet(numBits);
int doc; int doc;
while ((doc = iteartor.nextDoc()) != DocIdSetIterator.NO_MORE_DOCS) { while ((doc = iterator.nextDoc()) != DocIdSetIterator.NO_MORE_DOCS) {
set.set(doc); set.set(doc);
} }
return set; return set;