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

LUCENE-6510: take path boosts into account when polling TopNSearcher queue 

git-svn-id: https://svn.apache.org/repos/asf/lucene/dev/trunk@1682290 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
Areek Zillur 2015-05-28 17:46:55 +00:00
parent bc636dea84
commit 40f3361338
2 changed files with 49 additions and 49 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
lucene/core/src/java/org/apache/lucene/util/fst/Util.java
View File

@ -273,7 +273,7 @@ public final class Util {
@Override @Override
public String toString() { public String toString() {
return "input=" + input + " cost=" + cost + "context=" + context + "boost=" + boost; return "input=" + input.get() + " cost=" + cost + "context=" + context + "boost=" + boost;
} }
} }
@ -307,7 +307,8 @@ public final class Util {
private final FST.Arc<T> scratchArc = new FST.Arc<>(); private final FST.Arc<T> scratchArc = new FST.Arc<>();
final Comparator<T> comparator; private final Comparator<T> comparator;
private final Comparator<FSTPath<T>> pathComparator;
TreeSet<FSTPath<T>> queue = null; TreeSet<FSTPath<T>> queue = null;
@ -329,7 +330,7 @@ public final class Util {
this.topN = topN; this.topN = topN;
this.maxQueueDepth = maxQueueDepth; this.maxQueueDepth = maxQueueDepth;
this.comparator = comparator; this.comparator = comparator;
this.pathComparator = pathComparator;
queue = new TreeSet<>(pathComparator); queue = new TreeSet<>(pathComparator);
} }
@ -343,7 +344,7 @@ public final class Util {
if (queue.size() == maxQueueDepth) { if (queue.size() == maxQueueDepth) {
FSTPath<T> bottom = queue.last(); FSTPath<T> bottom = queue.last();
int comp = comparator.compare(cost, bottom.cost); int comp = pathComparator.compare(path, bottom);
if (comp > 0) { if (comp > 0) {
// Doesn't compete // Doesn't compete
return; return;

9
lucene/suggest/src/test/org/apache/lucene/search/suggest/document/TestContextQuery.java
View File

@ -478,7 +478,7 @@ public class TestContextQuery extends LuceneTestCase {
@Test @Test
public void testRandomContextQueryScoring() throws Exception { public void testRandomContextQueryScoring() throws Exception {
Analyzer analyzer = new MockAnalyzer(random()); Analyzer analyzer = new MockAnalyzer(random());
RandomIndexWriter iw = new RandomIndexWriter(random(), dir, iwcWithSuggestField(analyzer, "suggest_field")); try(RandomIndexWriter iw = new RandomIndexWriter(random(), dir, iwcWithSuggestField(analyzer, "suggest_field"))) {
int numSuggestions = atLeast(20); int numSuggestions = atLeast(20);
int numContexts = atLeast(5); int numContexts = atLeast(5);
@ -518,7 +518,7 @@ public class TestContextQuery extends LuceneTestCase {
} }
}); });
DirectoryReader reader = iw.getReader(); try(DirectoryReader reader = iw.getReader()) {
SuggestIndexSearcher suggestIndexSearcher = new SuggestIndexSearcher(reader); SuggestIndexSearcher suggestIndexSearcher = new SuggestIndexSearcher(reader);
ContextQuery query = new ContextQuery(new PrefixCompletionQuery(analyzer, new Term("suggest_field", "sugg"))); ContextQuery query = new ContextQuery(new PrefixCompletionQuery(analyzer, new Term("suggest_field", "sugg")));
for (int i = 0; i < contexts.size(); i++) { for (int i = 0; i < contexts.size(); i++) {
@ -526,8 +526,7 @@ public class TestContextQuery extends LuceneTestCase {
} }
TopSuggestDocs suggest = suggestIndexSearcher.suggest(query, 4); TopSuggestDocs suggest = suggestIndexSearcher.suggest(query, 4);
assertSuggestions(suggest, Arrays.copyOfRange(expectedResults, 0, 4)); assertSuggestions(suggest, Arrays.copyOfRange(expectedResults, 0, 4));
}
reader.close(); }
iw.close();
} }
} }