public class FirstPassGroupingCollector<T> extends SimpleCollector
GroupSelector
See org.apache.lucene.search.grouping
for more
details including a full code example.
Modifier and Type | Field and Description |
---|---|
private FieldComparator<?>[] |
comparators |
private int |
compIDXEnd |
private int |
docBase |
private java.util.HashMap<T,CollectedSearchGroup<T>> |
groupMap |
private GroupSelector<T> |
groupSelector |
private LeafFieldComparator[] |
leafComparators |
private boolean |
needsScores |
protected java.util.TreeSet<CollectedSearchGroup<T>> |
orderedGroups |
private int[] |
reversed |
private int |
spareSlot |
private int |
topNGroups |
Constructor and Description |
---|
FirstPassGroupingCollector(GroupSelector<T> groupSelector,
Sort groupSort,
int topNGroups)
Create the first pass collector.
|
Modifier and Type | Method and Description |
---|---|
private void |
buildSortedSet() |
void |
collect(int doc)
Called once for every document matching a query, with the unbased document
number.
|
protected void |
doSetNextReader(LeafReaderContext readerContext)
This method is called before collecting
context . |
GroupSelector<T> |
getGroupSelector() |
java.util.Collection<SearchGroup<T>> |
getTopGroups(int groupOffset)
Returns top groups, starting from offset.
|
private boolean |
isCompetitive(int doc) |
ScoreMode |
scoreMode()
Indicates what features are required from the scorer.
|
void |
setScorer(Scorable scorer)
Called before successive calls to
LeafCollector.collect(int) . |
getLeafCollector
private final GroupSelector<T> groupSelector
private final FieldComparator<?>[] comparators
private final LeafFieldComparator[] leafComparators
private final int[] reversed
private final int topNGroups
private final boolean needsScores
private final java.util.HashMap<T,CollectedSearchGroup<T>> groupMap
private final int compIDXEnd
protected java.util.TreeSet<CollectedSearchGroup<T>> orderedGroups
private int docBase
private int spareSlot
public FirstPassGroupingCollector(GroupSelector<T> groupSelector, Sort groupSort, int topNGroups)
groupSelector
- a GroupSelector used to defined groupsgroupSort
- The Sort
used to sort the
groups. The top sorted document within each group
according to groupSort, determines how that group
sorts against other groups. This must be non-null,
ie, if you want to groupSort by relevance use
Sort.RELEVANCE.topNGroups
- How many top groups to keep.public ScoreMode scoreMode()
Collector
public java.util.Collection<SearchGroup<T>> getTopGroups(int groupOffset) throws java.io.IOException
groupOffset
- The offset in the collected groupsjava.io.IOException
public void setScorer(Scorable scorer) throws java.io.IOException
LeafCollector
LeafCollector.collect(int)
. Implementations
that need the score of the current document (passed-in to
LeafCollector.collect(int)
), should save the passed-in Scorer and call
scorer.score() when needed.setScorer
in interface LeafCollector
setScorer
in class SimpleCollector
java.io.IOException
private boolean isCompetitive(int doc) throws java.io.IOException
java.io.IOException
public void collect(int doc) throws java.io.IOException
LeafCollector
Note: The collection of the current segment can be terminated by throwing
a CollectionTerminatedException
. In this case, the last docs of the
current LeafReaderContext
will be skipped and IndexSearcher
will swallow the exception and continue collection with the next leaf.
Note: This is called in an inner search loop. For good search performance,
implementations of this method should not call IndexSearcher.doc(int)
or
IndexReader.document(int)
on every hit.
Doing so can slow searches by an order of magnitude or more.
collect
in interface LeafCollector
collect
in class SimpleCollector
java.io.IOException
private void buildSortedSet() throws java.io.IOException
java.io.IOException
protected void doSetNextReader(LeafReaderContext readerContext) throws java.io.IOException
SimpleCollector
context
.doSetNextReader
in class SimpleCollector
java.io.IOException
public GroupSelector<T> getGroupSelector()