|
6 | 6 | */ |
7 | 7 | package org.elasticsearch.xpack.esql.core.querydsl.query; |
8 | 8 |
|
| 9 | +import org.apache.lucene.search.MatchAllDocsQuery; |
| 10 | +import org.apache.lucene.search.MatchNoDocsQuery; |
| 11 | +import org.elasticsearch.common.regex.Regex; |
| 12 | +import org.elasticsearch.core.Nullable; |
| 13 | +import org.elasticsearch.index.mapper.IndexFieldMapper; |
| 14 | +import org.elasticsearch.index.mapper.MappedFieldType; |
| 15 | +import org.elasticsearch.index.query.CoordinatorRewriteContext; |
| 16 | +import org.elasticsearch.index.query.MatchAllQueryBuilder; |
| 17 | +import org.elasticsearch.index.query.MatchNoneQueryBuilder; |
9 | 18 | import org.elasticsearch.index.query.QueryBuilder; |
| 19 | +import org.elasticsearch.index.query.QueryRewriteContext; |
10 | 20 | import org.elasticsearch.index.query.WildcardQueryBuilder; |
11 | 21 | import org.elasticsearch.xpack.esql.core.tree.Source; |
12 | 22 |
|
| 23 | +import java.util.Locale; |
13 | 24 | import java.util.Objects; |
14 | 25 |
|
15 | 26 | import static org.elasticsearch.index.query.QueryBuilders.wildcardQuery; |
@@ -44,7 +55,24 @@ public Boolean caseInsensitive() { |
44 | 55 |
|
45 | 56 | @Override |
46 | 57 | protected QueryBuilder asBuilder() { |
47 | | - WildcardQueryBuilder wb = wildcardQuery(field, query); |
| 58 | + WildcardQueryBuilder wb = new WildcardQueryBuilder(field, query) { |
| 59 | + @Override |
| 60 | + protected QueryBuilder maybeRewriteBasedOnConstantFields(@Nullable MappedFieldType fieldType, QueryRewriteContext context) { |
| 61 | + if (fieldType instanceof IndexFieldMapper.IndexFieldType) { |
| 62 | + String value = value(); |
| 63 | + String indexName = context.getFullyQualifiedIndex().getName(); |
| 64 | + if (WildcardQuery.this.caseInsensitive) { |
| 65 | + value = value.toLowerCase(Locale.ROOT); |
| 66 | + indexName = indexName.toLowerCase(Locale.ROOT); |
| 67 | + } |
| 68 | + if (Regex.simpleMatch(value, indexName)) { |
| 69 | + return new MatchAllQueryBuilder(); |
| 70 | + } |
| 71 | + return new MatchNoneQueryBuilder("The \"" + getName() + "\" query was rewritten to a \"match_none\" query."); |
| 72 | + } |
| 73 | + return super.maybeRewriteBasedOnConstantFields(fieldType, context); |
| 74 | + } |
| 75 | + }; |
48 | 76 | // ES does not allow case_insensitive to be set to "false", it should be either "true" or not specified |
49 | 77 | return caseInsensitive == false ? wb : wb.caseInsensitive(caseInsensitive); |
50 | 78 | } |
|
0 commit comments