Add solr query
This commit is contained in:
parent
3a6883c274
commit
e66bc6ce53
10
pom.xml
10
pom.xml
@ -206,11 +206,11 @@
|
|||||||
<artifactId>lucene-core</artifactId>
|
<artifactId>lucene-core</artifactId>
|
||||||
<version>${lucene.version}</version>
|
<version>${lucene.version}</version>
|
||||||
</dependency>
|
</dependency>
|
||||||
<dependency>
|
<dependency>
|
||||||
<groupId>org.apache.lucene</groupId>
|
<groupId>org.apache.lucene</groupId>
|
||||||
<artifactId>lucene-join</artifactId>
|
<artifactId>lucene-join</artifactId>
|
||||||
<version>${lucene.version}</version>
|
<version>${lucene.version}</version>
|
||||||
</dependency>
|
</dependency>
|
||||||
<dependency>
|
<dependency>
|
||||||
<groupId>org.apache.lucene</groupId>
|
<groupId>org.apache.lucene</groupId>
|
||||||
<artifactId>lucene-analysis-common</artifactId>
|
<artifactId>lucene-analysis-common</artifactId>
|
||||||
|
@ -6,7 +6,7 @@ superTypesData:
|
|||||||
Query: [
|
Query: [
|
||||||
BoxedQuery, TermQuery, IntTermQuery, IntNDTermQuery, LongTermQuery, LongNDTermQuery, FloatTermQuery,
|
BoxedQuery, TermQuery, IntTermQuery, IntNDTermQuery, LongTermQuery, LongNDTermQuery, FloatTermQuery,
|
||||||
FloatNDTermQuery, DoubleTermQuery, DoubleNDTermQuery,
|
FloatNDTermQuery, DoubleTermQuery, DoubleNDTermQuery,
|
||||||
PhraseQuery, WildcardQuery, SynonymQuery, FuzzyQuery, MatchAllDocsQuery, MatchNoDocsQuery,
|
PhraseQuery, SolrTextQuery, WildcardQuery, SynonymQuery, FuzzyQuery, MatchAllDocsQuery, MatchNoDocsQuery,
|
||||||
BooleanQuery, SortedNumericDocValuesFieldSlowRangeQuery, SortedDocFieldExistsQuery,
|
BooleanQuery, SortedNumericDocValuesFieldSlowRangeQuery, SortedDocFieldExistsQuery,
|
||||||
ConstantScoreQuery, BoostQuery, IntPointRangeQuery, IntNDPointRangeQuery, LongPointRangeQuery,
|
ConstantScoreQuery, BoostQuery, IntPointRangeQuery, IntNDPointRangeQuery, LongPointRangeQuery,
|
||||||
FloatPointRangeQuery, DoublePointRangeQuery, LongNDPointRangeQuery, FloatNDPointRangeQuery,
|
FloatPointRangeQuery, DoublePointRangeQuery, LongNDPointRangeQuery, FloatNDPointRangeQuery,
|
||||||
@ -136,6 +136,14 @@ baseTypesData:
|
|||||||
# counted as characters from the beginning of the phrase.
|
# counted as characters from the beginning of the phrase.
|
||||||
phrase: TermPosition[]
|
phrase: TermPosition[]
|
||||||
slop: int
|
slop: int
|
||||||
|
# Query that matches a phrase. (Solr)
|
||||||
|
SolrTextQuery:
|
||||||
|
data:
|
||||||
|
# Field name
|
||||||
|
field: String
|
||||||
|
# Text query
|
||||||
|
phrase: String
|
||||||
|
slop: int
|
||||||
# Advanced query that matches text allowing asterisks in the query
|
# Advanced query that matches text allowing asterisks in the query
|
||||||
WildcardQuery:
|
WildcardQuery:
|
||||||
data:
|
data:
|
||||||
|
@ -1,5 +1,8 @@
|
|||||||
package it.cavallium.dbengine.client.query;
|
package it.cavallium.dbengine.client.query;
|
||||||
|
|
||||||
|
import com.google.common.xml.XmlEscapers;
|
||||||
|
import it.cavallium.dbengine.client.query.current.data.BooleanQuery;
|
||||||
|
import it.cavallium.dbengine.client.query.current.data.BooleanQueryBuilder;
|
||||||
import it.cavallium.dbengine.client.query.current.data.BooleanQueryPart;
|
import it.cavallium.dbengine.client.query.current.data.BooleanQueryPart;
|
||||||
import it.cavallium.dbengine.client.query.current.data.BoostQuery;
|
import it.cavallium.dbengine.client.query.current.data.BoostQuery;
|
||||||
import it.cavallium.dbengine.client.query.current.data.BoxedQuery;
|
import it.cavallium.dbengine.client.query.current.data.BoxedQuery;
|
||||||
@ -34,9 +37,12 @@ import it.cavallium.dbengine.client.query.current.data.LongPointRangeQuery;
|
|||||||
import it.cavallium.dbengine.client.query.current.data.LongPointSetQuery;
|
import it.cavallium.dbengine.client.query.current.data.LongPointSetQuery;
|
||||||
import it.cavallium.dbengine.client.query.current.data.LongTermQuery;
|
import it.cavallium.dbengine.client.query.current.data.LongTermQuery;
|
||||||
import it.cavallium.dbengine.client.query.current.data.NumericSort;
|
import it.cavallium.dbengine.client.query.current.data.NumericSort;
|
||||||
|
import it.cavallium.dbengine.client.query.current.data.OccurMust;
|
||||||
|
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.PhraseQuery;
|
||||||
import it.cavallium.dbengine.client.query.current.data.PointConfig;
|
import it.cavallium.dbengine.client.query.current.data.PointConfig;
|
||||||
import it.cavallium.dbengine.client.query.current.data.PointType;
|
import it.cavallium.dbengine.client.query.current.data.PointType;
|
||||||
|
import it.cavallium.dbengine.client.query.current.data.SolrTextQuery;
|
||||||
import it.cavallium.dbengine.client.query.current.data.SortedDocFieldExistsQuery;
|
import it.cavallium.dbengine.client.query.current.data.SortedDocFieldExistsQuery;
|
||||||
import it.cavallium.dbengine.client.query.current.data.SortedNumericDocValuesFieldSlowRangeQuery;
|
import it.cavallium.dbengine.client.query.current.data.SortedNumericDocValuesFieldSlowRangeQuery;
|
||||||
import it.cavallium.dbengine.client.query.current.data.SynonymQuery;
|
import it.cavallium.dbengine.client.query.current.data.SynonymQuery;
|
||||||
@ -47,9 +53,13 @@ import it.cavallium.dbengine.client.query.current.data.WildcardQuery;
|
|||||||
import it.cavallium.dbengine.lucene.RandomSortField;
|
import it.cavallium.dbengine.lucene.RandomSortField;
|
||||||
import java.text.DecimalFormat;
|
import java.text.DecimalFormat;
|
||||||
import java.text.NumberFormat;
|
import java.text.NumberFormat;
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.Comparator;
|
||||||
|
import java.util.List;
|
||||||
import java.util.Map;
|
import java.util.Map;
|
||||||
import java.util.function.Function;
|
import java.util.function.Function;
|
||||||
import java.util.stream.Collectors;
|
import java.util.stream.Collectors;
|
||||||
|
import org.apache.commons.lang3.StringUtils;
|
||||||
import org.apache.lucene.analysis.Analyzer;
|
import org.apache.lucene.analysis.Analyzer;
|
||||||
import org.apache.lucene.analysis.miscellaneous.PerFieldAnalyzerWrapper;
|
import org.apache.lucene.analysis.miscellaneous.PerFieldAnalyzerWrapper;
|
||||||
import org.apache.lucene.document.DoublePoint;
|
import org.apache.lucene.document.DoublePoint;
|
||||||
@ -61,6 +71,7 @@ import org.apache.lucene.index.Term;
|
|||||||
import org.apache.lucene.queryparser.flexible.core.QueryNodeException;
|
import org.apache.lucene.queryparser.flexible.core.QueryNodeException;
|
||||||
import org.apache.lucene.queryparser.flexible.standard.StandardQueryParser;
|
import org.apache.lucene.queryparser.flexible.standard.StandardQueryParser;
|
||||||
import org.apache.lucene.queryparser.flexible.standard.config.PointsConfig;
|
import org.apache.lucene.queryparser.flexible.standard.config.PointsConfig;
|
||||||
|
import org.apache.lucene.queryparser.xml.builders.UserInputQueryBuilder;
|
||||||
import org.apache.lucene.search.BooleanClause.Occur;
|
import org.apache.lucene.search.BooleanClause.Occur;
|
||||||
import org.apache.lucene.search.BooleanQuery.Builder;
|
import org.apache.lucene.search.BooleanQuery.Builder;
|
||||||
import org.apache.lucene.search.DocValuesFieldExistsQuery;
|
import org.apache.lucene.search.DocValuesFieldExistsQuery;
|
||||||
@ -72,9 +83,13 @@ import org.apache.lucene.search.Sort;
|
|||||||
import org.apache.lucene.search.SortField;
|
import org.apache.lucene.search.SortField;
|
||||||
import org.apache.lucene.search.SortField.Type;
|
import org.apache.lucene.search.SortField.Type;
|
||||||
import org.apache.lucene.search.SortedNumericSortField;
|
import org.apache.lucene.search.SortedNumericSortField;
|
||||||
|
import org.jetbrains.annotations.Nullable;
|
||||||
|
|
||||||
public class QueryParser {
|
public class QueryParser {
|
||||||
|
|
||||||
|
private static final String[] QUERY_STRING_FIND = {"\\", "\""};
|
||||||
|
private static final String[] QUERY_STRING_REPLACE = {"\\\\", "\\\""};
|
||||||
|
|
||||||
public static Query toQuery(it.cavallium.dbengine.client.query.current.data.Query query, Analyzer analyzer) {
|
public static Query toQuery(it.cavallium.dbengine.client.query.current.data.Query query, Analyzer analyzer) {
|
||||||
if (query == null) {
|
if (query == null) {
|
||||||
return null;
|
return null;
|
||||||
@ -350,6 +365,389 @@ public class QueryParser {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public static void toQueryXML(StringBuilder out,
|
||||||
|
it.cavallium.dbengine.client.query.current.data.Query query,
|
||||||
|
@Nullable Float boost) {
|
||||||
|
if (query == null) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
switch (query.getBaseType$()) {
|
||||||
|
case StandardQuery -> {
|
||||||
|
var standardQuery = (it.cavallium.dbengine.client.query.current.data.StandardQuery) query;
|
||||||
|
|
||||||
|
out.append("<UserQuery");
|
||||||
|
if (standardQuery.defaultFields().size() > 1) {
|
||||||
|
throw new UnsupportedOperationException("Maximum supported default fields count: 1");
|
||||||
|
}
|
||||||
|
if (boost != null) {
|
||||||
|
out.append(" boost=\"").append(boost).append("\"");
|
||||||
|
}
|
||||||
|
if (standardQuery.defaultFields().size() == 1) {
|
||||||
|
out
|
||||||
|
.append(" fieldName=\"")
|
||||||
|
.append(XmlEscapers.xmlAttributeEscaper().escape(standardQuery.defaultFields().get(0)))
|
||||||
|
.append("\"");
|
||||||
|
}
|
||||||
|
if (!standardQuery.termFields().isEmpty()) {
|
||||||
|
throw new UnsupportedOperationException("Term fields unsupported");
|
||||||
|
}
|
||||||
|
if (!standardQuery.pointsConfig().isEmpty()) {
|
||||||
|
throw new UnsupportedOperationException("Points config unsupported");
|
||||||
|
}
|
||||||
|
out.append(">");
|
||||||
|
out.append(XmlEscapers.xmlContentEscaper().escape(standardQuery.query()));
|
||||||
|
out.append("</UserQuery>\n");
|
||||||
|
}
|
||||||
|
case BooleanQuery -> {
|
||||||
|
var booleanQuery = (it.cavallium.dbengine.client.query.current.data.BooleanQuery) query;
|
||||||
|
|
||||||
|
out.append("<BooleanQuery");
|
||||||
|
if (boost != null) {
|
||||||
|
out.append(" boost=\"").append(boost).append("\"");
|
||||||
|
}
|
||||||
|
out.append(" minimumNumberShouldMatch=\"").append(booleanQuery.minShouldMatch()).append("\"");
|
||||||
|
out.append(">\n");
|
||||||
|
|
||||||
|
for (BooleanQueryPart part : booleanQuery.parts()) {
|
||||||
|
out.append("<Clause");
|
||||||
|
out.append(" occurs=\"").append(switch (part.occur().getBaseType$()) {
|
||||||
|
case OccurFilter -> "filter";
|
||||||
|
case OccurMust -> "must";
|
||||||
|
case OccurShould -> "should";
|
||||||
|
case OccurMustNot -> "mustNot";
|
||||||
|
default -> throw new IllegalStateException("Unexpected value: " + part.occur().getBaseType$());
|
||||||
|
}).append("\"");
|
||||||
|
out.append(">\n");
|
||||||
|
toQueryXML(out, part.query(), null);
|
||||||
|
out.append("</Clause>\n");
|
||||||
|
}
|
||||||
|
out.append("</BooleanQuery>\n");
|
||||||
|
}
|
||||||
|
case IntPointExactQuery -> {
|
||||||
|
var intPointExactQuery = (IntPointExactQuery) query;
|
||||||
|
out.append("<PointRangeQuery type=\"int\"");
|
||||||
|
if (boost != null) {
|
||||||
|
out.append(" boost=\"").append(boost).append("\"");
|
||||||
|
}
|
||||||
|
out.append(" fieldName=\"").append(XmlEscapers.xmlAttributeEscaper().escape(intPointExactQuery.field())).append("\"");
|
||||||
|
out.append(" lowerTerm=\"").append(intPointExactQuery.value()).append("\"");
|
||||||
|
out.append(" upperTerm=\"").append(intPointExactQuery.value()).append("\"");
|
||||||
|
out.append(" />\n");
|
||||||
|
}
|
||||||
|
case IntNDPointExactQuery -> {
|
||||||
|
var intPointExactQuery = (IntPointExactQuery) query;
|
||||||
|
throw new UnsupportedOperationException("N-dimensional point queries are not supported");
|
||||||
|
}
|
||||||
|
case LongPointExactQuery -> {
|
||||||
|
var longPointExactQuery = (LongPointExactQuery) query;
|
||||||
|
out.append("<PointRangeQuery type=\"long\"");
|
||||||
|
if (boost != null) {
|
||||||
|
out.append(" boost=\"").append(boost).append("\"");
|
||||||
|
}
|
||||||
|
out.append(" fieldName=\"").append(XmlEscapers.xmlAttributeEscaper().escape(longPointExactQuery.field())).append("\"");
|
||||||
|
out.append(" lowerTerm=\"").append(longPointExactQuery.value()).append("\"");
|
||||||
|
out.append(" upperTerm=\"").append(longPointExactQuery.value()).append("\"");
|
||||||
|
out.append(" />\n");
|
||||||
|
}
|
||||||
|
case FloatPointExactQuery -> {
|
||||||
|
var floatPointExactQuery = (FloatPointExactQuery) query;
|
||||||
|
out.append("<PointRangeQuery type=\"float\"");
|
||||||
|
if (boost != null) {
|
||||||
|
out.append(" boost=\"").append(boost).append("\"");
|
||||||
|
}
|
||||||
|
out.append(" fieldName=\"").append(XmlEscapers.xmlAttributeEscaper().escape(floatPointExactQuery.field())).append("\"");
|
||||||
|
out.append(" lowerTerm=\"").append(floatPointExactQuery.value()).append("\"");
|
||||||
|
out.append(" upperTerm=\"").append(floatPointExactQuery.value()).append("\"");
|
||||||
|
out.append(" />\n");
|
||||||
|
}
|
||||||
|
case DoublePointExactQuery -> {
|
||||||
|
var doublePointExactQuery = (DoublePointExactQuery) query;
|
||||||
|
out.append("<PointRangeQuery type=\"double\"");
|
||||||
|
if (boost != null) {
|
||||||
|
out.append(" boost=\"").append(boost).append("\"");
|
||||||
|
}
|
||||||
|
out.append(" fieldName=\"").append(XmlEscapers.xmlAttributeEscaper().escape(doublePointExactQuery.field())).append("\"");
|
||||||
|
out.append(" lowerTerm=\"").append(doublePointExactQuery.value()).append("\"");
|
||||||
|
out.append(" upperTerm=\"").append(doublePointExactQuery.value()).append("\"");
|
||||||
|
out.append(" />\n");
|
||||||
|
}
|
||||||
|
case LongNDPointExactQuery -> {
|
||||||
|
var longndPointExactQuery = (LongNDPointExactQuery) query;
|
||||||
|
throw new UnsupportedOperationException("N-dimensional point queries are not supported");
|
||||||
|
}
|
||||||
|
case FloatNDPointExactQuery -> {
|
||||||
|
var floatndPointExactQuery = (FloatNDPointExactQuery) query;
|
||||||
|
throw new UnsupportedOperationException("N-dimensional point queries are not supported");
|
||||||
|
}
|
||||||
|
case DoubleNDPointExactQuery -> {
|
||||||
|
var doublendPointExactQuery = (DoubleNDPointExactQuery) query;
|
||||||
|
throw new UnsupportedOperationException("N-dimensional point queries are not supported");
|
||||||
|
}
|
||||||
|
case IntPointSetQuery -> {
|
||||||
|
var intPointSetQuery = (IntPointSetQuery) query;
|
||||||
|
// Polyfill
|
||||||
|
toQueryXML(out, BooleanQuery.of(intPointSetQuery.values().intStream()
|
||||||
|
.mapToObj(val -> IntPointExactQuery.of(intPointSetQuery.field(), val))
|
||||||
|
.map(q -> BooleanQueryPart.of(q, OccurShould.of()))
|
||||||
|
.toList(), 1), boost);
|
||||||
|
}
|
||||||
|
case LongPointSetQuery -> {
|
||||||
|
var longPointSetQuery = (LongPointSetQuery) query;
|
||||||
|
// Polyfill
|
||||||
|
toQueryXML(out, BooleanQuery.of(longPointSetQuery.values().longStream()
|
||||||
|
.mapToObj(val -> LongPointExactQuery.of(longPointSetQuery.field(), val))
|
||||||
|
.map(q -> BooleanQueryPart.of(q, OccurShould.of()))
|
||||||
|
.toList(), 1), boost);
|
||||||
|
}
|
||||||
|
case FloatPointSetQuery -> {
|
||||||
|
var floatPointSetQuery = (FloatPointSetQuery) query;
|
||||||
|
// Polyfill
|
||||||
|
toQueryXML(out, BooleanQuery.of(floatPointSetQuery.values().stream()
|
||||||
|
.map(val -> FloatPointExactQuery.of(floatPointSetQuery.field(), val))
|
||||||
|
.map(q -> BooleanQueryPart.of(q, OccurShould.of()))
|
||||||
|
.toList(), 1), boost);
|
||||||
|
}
|
||||||
|
case DoublePointSetQuery -> {
|
||||||
|
var doublePointSetQuery = (DoublePointSetQuery) query;
|
||||||
|
// Polyfill
|
||||||
|
toQueryXML(out, BooleanQuery.of(doublePointSetQuery.values().doubleStream()
|
||||||
|
.mapToObj(val -> DoublePointExactQuery.of(doublePointSetQuery.field(), val))
|
||||||
|
.map(q -> BooleanQueryPart.of(q, OccurShould.of()))
|
||||||
|
.toList(), 1), boost);
|
||||||
|
}
|
||||||
|
case TermQuery -> {
|
||||||
|
var termQuery = (TermQuery) query;
|
||||||
|
out
|
||||||
|
.append("<TermQuery");
|
||||||
|
if (boost != null) {
|
||||||
|
out.append(" boost=\"").append(boost).append("\"");
|
||||||
|
}
|
||||||
|
out
|
||||||
|
.append(" fieldName=\"")
|
||||||
|
.append(XmlEscapers.xmlAttributeEscaper().escape(termQuery.term().field()))
|
||||||
|
.append("\"");
|
||||||
|
out.append(">");
|
||||||
|
out.append(XmlEscapers.xmlContentEscaper().escape(termQuery.term().value()));
|
||||||
|
out.append("</TermQuery>\n");
|
||||||
|
}
|
||||||
|
case IntTermQuery -> {
|
||||||
|
var intTermQuery = (IntTermQuery) query;
|
||||||
|
throw new UnsupportedOperationException("Non-string term fields are not supported");
|
||||||
|
}
|
||||||
|
case IntNDTermQuery -> {
|
||||||
|
var intNDTermQuery = (IntNDTermQuery) query;
|
||||||
|
throw new UnsupportedOperationException("Non-string term fields are not supported");
|
||||||
|
}
|
||||||
|
case LongTermQuery -> {
|
||||||
|
var longTermQuery = (LongTermQuery) query;
|
||||||
|
throw new UnsupportedOperationException("Non-string term fields are not supported");
|
||||||
|
}
|
||||||
|
case LongNDTermQuery -> {
|
||||||
|
var longNDTermQuery = (LongNDTermQuery) query;
|
||||||
|
throw new UnsupportedOperationException("Non-string term fields are not supported");
|
||||||
|
}
|
||||||
|
case FloatTermQuery -> {
|
||||||
|
var floatTermQuery = (FloatTermQuery) query;
|
||||||
|
throw new UnsupportedOperationException("Non-string term fields are not supported");
|
||||||
|
}
|
||||||
|
case FloatNDTermQuery -> {
|
||||||
|
var floatNDTermQuery = (FloatNDTermQuery) query;
|
||||||
|
throw new UnsupportedOperationException("Non-string term fields are not supported");
|
||||||
|
}
|
||||||
|
case DoubleTermQuery -> {
|
||||||
|
var doubleTermQuery = (DoubleTermQuery) query;
|
||||||
|
throw new UnsupportedOperationException("Non-string term fields are not supported");
|
||||||
|
}
|
||||||
|
case DoubleNDTermQuery -> {
|
||||||
|
var doubleNDTermQuery = (DoubleNDTermQuery) query;
|
||||||
|
throw new UnsupportedOperationException("Non-string term fields are not supported");
|
||||||
|
}
|
||||||
|
case FieldExistsQuery -> {
|
||||||
|
var fieldExistQuery = (FieldExistsQuery) query;
|
||||||
|
out.append("<UserQuery");
|
||||||
|
if (boost != null) {
|
||||||
|
out.append(" boost=\"").append(boost).append("\"");
|
||||||
|
}
|
||||||
|
out.append(">");
|
||||||
|
ensureValidField(fieldExistQuery.field());
|
||||||
|
out.append(fieldExistQuery.field());
|
||||||
|
out.append(":[* TO *]");
|
||||||
|
out.append("</UserQuery>\n");
|
||||||
|
}
|
||||||
|
case SolrTextQuery -> {
|
||||||
|
var solrTextQuery = (SolrTextQuery) query;
|
||||||
|
out.append("<UserQuery");
|
||||||
|
if (boost != null) {
|
||||||
|
out.append(" boost=\"").append(boost).append("\"");
|
||||||
|
}
|
||||||
|
out.append(">");
|
||||||
|
ensureValidField(solrTextQuery.field());
|
||||||
|
out.append(solrTextQuery.field());
|
||||||
|
out.append(":");
|
||||||
|
out.append("\"").append(escapeQueryStringValue(solrTextQuery.phrase())).append("\"");
|
||||||
|
if (solrTextQuery.slop() > 0) {
|
||||||
|
out.append("~").append(solrTextQuery.slop());
|
||||||
|
}
|
||||||
|
out.append("</UserQuery>\n");
|
||||||
|
}
|
||||||
|
case BoostQuery -> {
|
||||||
|
var boostQuery = (BoostQuery) query;
|
||||||
|
toQueryXML(out, boostQuery.query(), boostQuery.scoreBoost());
|
||||||
|
}
|
||||||
|
case ConstantScoreQuery -> {
|
||||||
|
var constantScoreQuery = (ConstantScoreQuery) query;
|
||||||
|
out.append("<ConstantScoreQuery");
|
||||||
|
if (boost != null) {
|
||||||
|
out.append(" boost=\"").append(boost).append("\"");
|
||||||
|
}
|
||||||
|
out.append(">\n");
|
||||||
|
toQueryXML(out, query, null);
|
||||||
|
out.append("</ConstantScoreQuery>\n");
|
||||||
|
}
|
||||||
|
case BoxedQuery -> {
|
||||||
|
toQueryXML(out, ((BoxedQuery) query).query(), boost);
|
||||||
|
}
|
||||||
|
case FuzzyQuery -> {
|
||||||
|
var fuzzyQuery = (it.cavallium.dbengine.client.query.current.data.FuzzyQuery) query;
|
||||||
|
new FuzzyQuery(toTerm(fuzzyQuery.term()),
|
||||||
|
fuzzyQuery.maxEdits(),
|
||||||
|
fuzzyQuery.prefixLength(),
|
||||||
|
fuzzyQuery.maxExpansions(),
|
||||||
|
fuzzyQuery.transpositions()
|
||||||
|
);
|
||||||
|
throw new UnsupportedOperationException("Fuzzy query is not supported, use span queries");
|
||||||
|
}
|
||||||
|
case IntPointRangeQuery -> {
|
||||||
|
var intPointRangeQuery = (IntPointRangeQuery) query;
|
||||||
|
out.append("<PointRangeQuery type=\"int\"");
|
||||||
|
if (boost != null) {
|
||||||
|
out.append(" boost=\"").append(boost).append("\"");
|
||||||
|
}
|
||||||
|
out.append(" fieldName=\"").append(XmlEscapers.xmlAttributeEscaper().escape(intPointRangeQuery.field())).append("\"");
|
||||||
|
out.append(" lowerTerm=\"").append(intPointRangeQuery.min()).append("\"");
|
||||||
|
out.append(" upperTerm=\"").append(intPointRangeQuery.max()).append("\"");
|
||||||
|
out.append(" />\n");
|
||||||
|
}
|
||||||
|
case IntNDPointRangeQuery -> {
|
||||||
|
var intndPointRangeQuery = (IntNDPointRangeQuery) query;
|
||||||
|
throw new UnsupportedOperationException("N-dimensional point queries are not supported");
|
||||||
|
}
|
||||||
|
case LongPointRangeQuery -> {
|
||||||
|
var longPointRangeQuery = (LongPointRangeQuery) query;
|
||||||
|
out.append("<PointRangeQuery type=\"long\"");
|
||||||
|
if (boost != null) {
|
||||||
|
out.append(" boost=\"").append(boost).append("\"");
|
||||||
|
}
|
||||||
|
out.append(" fieldName=\"").append(XmlEscapers.xmlAttributeEscaper().escape(longPointRangeQuery.field())).append("\"");
|
||||||
|
out.append(" lowerTerm=\"").append(longPointRangeQuery.min()).append("\"");
|
||||||
|
out.append(" upperTerm=\"").append(longPointRangeQuery.max()).append("\"");
|
||||||
|
out.append(" />\n");
|
||||||
|
}
|
||||||
|
case FloatPointRangeQuery -> {
|
||||||
|
var floatPointRangeQuery = (FloatPointRangeQuery) query;
|
||||||
|
out.append("<PointRangeQuery type=\"float\"");
|
||||||
|
if (boost != null) {
|
||||||
|
out.append(" boost=\"").append(boost).append("\"");
|
||||||
|
}
|
||||||
|
out.append(" fieldName=\"").append(XmlEscapers.xmlAttributeEscaper().escape(floatPointRangeQuery.field())).append("\"");
|
||||||
|
out.append(" lowerTerm=\"").append(floatPointRangeQuery.min()).append("\"");
|
||||||
|
out.append(" upperTerm=\"").append(floatPointRangeQuery.max()).append("\"");
|
||||||
|
out.append(" />\n");
|
||||||
|
}
|
||||||
|
case DoublePointRangeQuery -> {
|
||||||
|
var doublePointRangeQuery = (DoublePointRangeQuery) query;
|
||||||
|
out.append("<PointRangeQuery type=\"double\"");
|
||||||
|
if (boost != null) {
|
||||||
|
out.append(" boost=\"").append(boost).append("\"");
|
||||||
|
}
|
||||||
|
out.append(" fieldName=\"").append(XmlEscapers.xmlAttributeEscaper().escape(doublePointRangeQuery.field())).append("\"");
|
||||||
|
out.append(" lowerTerm=\"").append(doublePointRangeQuery.min()).append("\"");
|
||||||
|
out.append(" upperTerm=\"").append(doublePointRangeQuery.max()).append("\"");
|
||||||
|
out.append(" />\n");
|
||||||
|
}
|
||||||
|
case LongNDPointRangeQuery -> {
|
||||||
|
var longndPointRangeQuery = (LongNDPointRangeQuery) query;
|
||||||
|
throw new UnsupportedOperationException("N-dimensional point queries are not supported");
|
||||||
|
}
|
||||||
|
case FloatNDPointRangeQuery -> {
|
||||||
|
var floatndPointRangeQuery = (FloatNDPointRangeQuery) query;
|
||||||
|
throw new UnsupportedOperationException("N-dimensional point queries are not supported");
|
||||||
|
}
|
||||||
|
case DoubleNDPointRangeQuery -> {
|
||||||
|
var doublendPointRangeQuery = (DoubleNDPointRangeQuery) query;
|
||||||
|
throw new UnsupportedOperationException("N-dimensional point queries are not supported");
|
||||||
|
}
|
||||||
|
case MatchAllDocsQuery -> {
|
||||||
|
out.append("<UserQuery");
|
||||||
|
if (boost != null) {
|
||||||
|
out.append(" boost=\"").append(boost).append("\"");
|
||||||
|
}
|
||||||
|
out.append(">");
|
||||||
|
out.append("*:*");
|
||||||
|
out.append("</UserQuery>\n");
|
||||||
|
}
|
||||||
|
case MatchNoDocsQuery -> {
|
||||||
|
out.append("<UserQuery");
|
||||||
|
if (boost != null) {
|
||||||
|
out.append(" boost=\"").append(boost).append("\"");
|
||||||
|
}
|
||||||
|
out.append(">");
|
||||||
|
//todo: check if it's correct
|
||||||
|
out.append("!*:*");
|
||||||
|
out.append("</UserQuery>\n");
|
||||||
|
}
|
||||||
|
case PhraseQuery -> {
|
||||||
|
//todo: check if it's correct
|
||||||
|
|
||||||
|
var phraseQuery = (PhraseQuery) query;
|
||||||
|
out.append("<SpanNear");
|
||||||
|
if (boost != null) {
|
||||||
|
out.append(" boost=\"").append(boost).append("\"");
|
||||||
|
}
|
||||||
|
out.append(" inOrder=\"true\"");
|
||||||
|
out.append(">\n");
|
||||||
|
phraseQuery.phrase().stream().sorted(Comparator.comparingInt(TermPosition::position)).forEach(term -> {
|
||||||
|
out
|
||||||
|
.append("<SpanTerm fieldName=\"")
|
||||||
|
.append(XmlEscapers.xmlAttributeEscaper().escape(term.term().field()))
|
||||||
|
.append("\">")
|
||||||
|
.append(XmlEscapers.xmlContentEscaper().escape(term.term().value()))
|
||||||
|
.append("</SpanTerm>\n");
|
||||||
|
});
|
||||||
|
out.append("</SpanNear>\n");
|
||||||
|
}
|
||||||
|
case SortedDocFieldExistsQuery -> {
|
||||||
|
var sortedDocFieldExistsQuery = (SortedDocFieldExistsQuery) query;
|
||||||
|
throw new UnsupportedOperationException("Field existence query is not supported");
|
||||||
|
}
|
||||||
|
case SynonymQuery -> {
|
||||||
|
var synonymQuery = (SynonymQuery) query;
|
||||||
|
throw new UnsupportedOperationException("Synonym query is not supported");
|
||||||
|
}
|
||||||
|
case SortedNumericDocValuesFieldSlowRangeQuery -> {
|
||||||
|
throw new UnsupportedOperationException("Slow range query is not supported");
|
||||||
|
}
|
||||||
|
case WildcardQuery -> {
|
||||||
|
var wildcardQuery = (WildcardQuery) query;
|
||||||
|
throw new UnsupportedOperationException("Wildcard query is not supported");
|
||||||
|
}
|
||||||
|
default -> throw new IllegalStateException("Unexpected value: " + query.getBaseType$());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private static String escapeQueryStringValue(String text) {
|
||||||
|
return StringUtils.replaceEach(text, QUERY_STRING_FIND, QUERY_STRING_REPLACE);
|
||||||
|
}
|
||||||
|
|
||||||
|
private static void ensureValidField(String field) {
|
||||||
|
field.codePoints().forEach(codePoint -> {
|
||||||
|
if (!Character.isLetterOrDigit(codePoint) && codePoint != '_') {
|
||||||
|
throw new UnsupportedOperationException(
|
||||||
|
"Invalid character \"" + codePoint + "\" in field name \"" + field + "\"");
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
private static NumberFormat toNumberFormat(it.cavallium.dbengine.client.query.current.data.NumberFormat numberFormat) {
|
private static NumberFormat toNumberFormat(it.cavallium.dbengine.client.query.current.data.NumberFormat numberFormat) {
|
||||||
return switch (numberFormat.getBaseType$()) {
|
return switch (numberFormat.getBaseType$()) {
|
||||||
case NumberFormatDecimal -> new DecimalFormat();
|
case NumberFormatDecimal -> new DecimalFormat();
|
||||||
|
@ -38,12 +38,19 @@ public class QueryUtils {
|
|||||||
return transformQuery(field, luceneQuery);
|
return transformQuery(field, luceneQuery);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Deprecated: use solr SolrTextQuery
|
||||||
|
*/
|
||||||
|
@Deprecated
|
||||||
public static Query phraseSearch(TextFieldsAnalyzer preferredAnalyzer, String field, String text, int slop) {
|
public static Query phraseSearch(TextFieldsAnalyzer preferredAnalyzer, String field, String text, int slop) {
|
||||||
var qb = new QueryBuilder(LuceneUtils.getAnalyzer(preferredAnalyzer));
|
var qb = new QueryBuilder(LuceneUtils.getAnalyzer(preferredAnalyzer));
|
||||||
var luceneQuery = qb.createPhraseQuery(field, text, slop);
|
var luceneQuery = qb.createPhraseQuery(field, text, slop);
|
||||||
return transformQuery(field, luceneQuery);
|
return transformQuery(field, luceneQuery);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Deprecated: use solr SolrTextQuery
|
||||||
|
*/
|
||||||
public static Query exactSearch(TextFieldsAnalyzer preferredAnalyzer, String field, String text) {
|
public static Query exactSearch(TextFieldsAnalyzer preferredAnalyzer, String field, String text) {
|
||||||
var qb = new QueryBuilder(LuceneUtils.getAnalyzer(preferredAnalyzer));
|
var qb = new QueryBuilder(LuceneUtils.getAnalyzer(preferredAnalyzer));
|
||||||
var luceneQuery = qb.createPhraseQuery(field, text);
|
var luceneQuery = qb.createPhraseQuery(field, text);
|
||||||
|
Loading…
Reference in New Issue
Block a user