0ct0pu5/fess
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add PhraseQuery

Browse source
This commit is contained in:
Shinsuke Sugaya 2016-01-10 16:33:41 +09:00
parent 1539f2e549
commit acb5f75ee6
1 changed files with 17 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
src/main/java/org/codelibs/fess/helper/QueryHelper.java
View file

@ -41,6 +41,7 @@ import org.apache.lucene.search.BooleanClause;
import org.apache.lucene.search.BooleanQuery;
import org.apache.lucene.search.FuzzyQuery;
import org.apache.lucene.search.MatchAllDocsQuery;
import org.apache.lucene.search.PhraseQuery;
import org.apache.lucene.search.PrefixQuery;
import org.apache.lucene.search.Query;
import org.apache.lucene.search.TermQuery;
@ -270,6 +271,8 @@ public class QueryHelper implements Serializable {
return convertTermQuery(context, (TermQuery) query);
} else if (query instanceof TermRangeQuery) {
return convertTermRangeQuery(context, (TermRangeQuery) query);
} else if (query instanceof PhraseQuery) {
return convertPhraseQuery(context, (PhraseQuery) query);
} else if (query instanceof FuzzyQuery) {
return convertFuzzyQuery(context, (FuzzyQuery) query);
} else if (query instanceof PrefixQuery) {
@ -437,6 +440,20 @@ public class QueryHelper implements Serializable {
}
}
private QueryBuilder convertPhraseQuery(QueryContext context, PhraseQuery query) {
Term[] terms = query.getTerms();
if (terms.length == 0) {
throw new InvalidQueryException(messages -> messages.addErrorsInvalidQueryUnknown(ActionMessages.GLOBAL_PROPERTY_KEY),
"Unknown phrase query: " + query);
}
final String field = terms[0].field();
final String[] texts = StreamUtil.of(terms).map(term -> term.text()).toArray(n -> new String[n]);
final String text = String.join(" ", texts);
context.addFieldLog(field, text);
StreamUtil.of(texts).forEach(t -> context.addHighlightedQuery(t));
return buildDefaultQueryBuilder((f, b) -> QueryBuilders.matchPhraseQuery(f, text).boost(b));
}
private boolean isSearchField(final String field) {
for (final String searchField : searchFields) {
if (searchField.equals(field)) {