96 lines
3.9 KiB
Java
96 lines
3.9 KiB
Java
package it.cavallium.dbengine.client.query;
|
|
|
|
import it.cavallium.dbengine.client.query.current.data.BooleanQuery;
|
|
import it.cavallium.dbengine.client.query.current.data.BooleanQueryPart;
|
|
import it.cavallium.dbengine.client.query.current.data.Occur;
|
|
import it.cavallium.dbengine.client.query.current.data.OccurFilter;
|
|
import it.cavallium.dbengine.client.query.current.data.OccurMust;
|
|
import it.cavallium.dbengine.client.query.current.data.OccurMustNot;
|
|
import it.cavallium.dbengine.client.query.current.data.OccurShould;
|
|
import it.cavallium.dbengine.client.query.current.data.PhraseQuery;
|
|
import it.cavallium.dbengine.client.query.current.data.Query;
|
|
import it.cavallium.dbengine.client.query.current.data.SynonymQuery;
|
|
import it.cavallium.dbengine.client.query.current.data.TermAndBoost;
|
|
import it.cavallium.dbengine.client.query.current.data.TermPosition;
|
|
import it.cavallium.dbengine.client.query.current.data.TermQuery;
|
|
import it.cavallium.dbengine.lucene.LuceneUtils;
|
|
import it.cavallium.dbengine.lucene.analyzer.TextFieldsAnalyzer;
|
|
import java.util.ArrayList;
|
|
import org.apache.lucene.search.BooleanClause;
|
|
import org.apache.lucene.util.QueryBuilder;
|
|
import org.jetbrains.annotations.NotNull;
|
|
|
|
@SuppressWarnings("unused")
|
|
public class QueryUtils {
|
|
|
|
public static Query approximateSearch(TextFieldsAnalyzer preferredAnalyzer, String field, String text) {
|
|
var qb = new QueryBuilder(LuceneUtils.getAnalyzer(preferredAnalyzer));
|
|
var luceneQuery = qb.createMinShouldMatchQuery(field, text, 0.75f);
|
|
return transformQuery(field, luceneQuery);
|
|
}
|
|
|
|
public static Query exactSearch(TextFieldsAnalyzer preferredAnalyzer, String field, String text) {
|
|
var qb = new QueryBuilder(LuceneUtils.getAnalyzer(preferredAnalyzer));
|
|
var luceneQuery = qb.createPhraseQuery(field, text);
|
|
return transformQuery(field, luceneQuery);
|
|
}
|
|
|
|
@NotNull
|
|
private static Query transformQuery(String field, org.apache.lucene.search.Query luceneQuery) {
|
|
if (luceneQuery == null) {
|
|
return TermQuery.of(it.cavallium.dbengine.client.query.current.data.Term.of(field, ""));
|
|
}
|
|
if (luceneQuery instanceof org.apache.lucene.search.TermQuery) {
|
|
return TermQuery.of(QueryParser.toQueryTerm(((org.apache.lucene.search.TermQuery) luceneQuery).getTerm()));
|
|
}
|
|
if (luceneQuery instanceof org.apache.lucene.search.BooleanQuery) {
|
|
var booleanQuery = (org.apache.lucene.search.BooleanQuery) luceneQuery;
|
|
var queryParts = new ArrayList<BooleanQueryPart>();
|
|
for (BooleanClause booleanClause : booleanQuery) {
|
|
org.apache.lucene.search.Query queryPartQuery = booleanClause.getQuery();
|
|
|
|
Occur occur;
|
|
switch (booleanClause.getOccur()) {
|
|
case MUST:
|
|
occur = OccurMust.of();
|
|
break;
|
|
case FILTER:
|
|
occur = OccurFilter.of();
|
|
break;
|
|
case SHOULD:
|
|
occur = OccurShould.of();
|
|
break;
|
|
case MUST_NOT:
|
|
occur = OccurMustNot.of();
|
|
break;
|
|
default:
|
|
throw new IllegalArgumentException();
|
|
}
|
|
queryParts.add(BooleanQueryPart.of(transformQuery(field, queryPartQuery), occur));
|
|
}
|
|
return BooleanQuery.of(queryParts.toArray(BooleanQueryPart[]::new), booleanQuery.getMinimumNumberShouldMatch());
|
|
}
|
|
if (luceneQuery instanceof org.apache.lucene.search.PhraseQuery) {
|
|
var phraseQuery = (org.apache.lucene.search.PhraseQuery) luceneQuery;
|
|
int slop = phraseQuery.getSlop();
|
|
var terms = phraseQuery.getTerms();
|
|
var positions = phraseQuery.getPositions();
|
|
TermPosition[] termPositions = new TermPosition[terms.length];
|
|
for (int i = 0; i < terms.length; i++) {
|
|
var term = terms[i];
|
|
var position = positions[i];
|
|
termPositions[i] = TermPosition.of(QueryParser.toQueryTerm(term), position);
|
|
}
|
|
return PhraseQuery.of(termPositions, slop);
|
|
}
|
|
org.apache.lucene.search.SynonymQuery synonymQuery = (org.apache.lucene.search.SynonymQuery) luceneQuery;
|
|
return SynonymQuery.of(field,
|
|
synonymQuery
|
|
.getTerms()
|
|
.stream()
|
|
.map(term -> TermAndBoost.of(QueryParser.toQueryTerm(term), 1))
|
|
.toArray(TermAndBoost[]::new)
|
|
);
|
|
}
|
|
}
|