Skip to content

Commit b1e44db

Browse files
ES|QL: fix ENRICH validation for use of wildcards (#121911) (#122018)
1 parent a4c1e6a commit b1e44db

File tree

4 files changed

+30
-8
lines changed

4 files changed

+30
-8
lines changed

docs/changelog/121911.yaml

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
pr: 121911
2+
summary: Fix ENRICH validation for use of wildcards
3+
area: ES|QL
4+
type: bug
5+
issues: []

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/parser/ExpressionBuilder.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -795,9 +795,9 @@ public NamedExpression visitEnrichWithClause(EsqlBaseParser.EnrichWithClauseCont
795795

796796
private NamedExpression enrichFieldName(EsqlBaseParser.QualifiedNamePatternContext ctx) {
797797
return visitQualifiedNamePattern(ctx, ne -> {
798-
if (ne instanceof UnresolvedNamePattern up) {
798+
if (ne instanceof UnresolvedNamePattern || ne instanceof UnresolvedStar) {
799799
var src = ne.source();
800-
throw new ParsingException(src, "Using wildcards [*] in ENRICH WITH projections is not allowed [{}]", up.pattern());
800+
throw new ParsingException(src, "Using wildcards [*] in ENRICH WITH projections is not allowed, found [{}]", src.text());
801801
}
802802
});
803803
}

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/parser/LogicalPlanBuilder.java

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -438,7 +438,11 @@ public PlanFactory visitEnrichCommand(EsqlBaseParser.EnrichCommandContext ctx) {
438438
: matchField instanceof UnresolvedStar ? WILDCARD
439439
: null;
440440
if (patternString != null) {
441-
throw new ParsingException(source, "Using wildcards [*] in ENRICH WITH projections is not allowed [{}]", patternString);
441+
throw new ParsingException(
442+
source,
443+
"Using wildcards [*] in ENRICH WITH projections is not allowed, found [{}]",
444+
patternString
445+
);
442446
}
443447

444448
List<NamedExpression> keepClauses = visitList(this, ctx.enrichWithClause(), NamedExpression.class);

x-pack/plugin/esql/src/test/java/org/elasticsearch/xpack/esql/parser/StatementParserTests.java

Lines changed: 18 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1080,15 +1080,28 @@ public void testEnrich() {
10801080
processingCommand("enrich _" + mode.name() + ":countries ON country_code")
10811081
);
10821082

1083-
expectError("from a | enrich countries on foo* ", "Using wildcards [*] in ENRICH WITH projections is not allowed [foo*]");
1084-
expectError("from a | enrich countries on foo with bar*", "Using wildcards [*] in ENRICH WITH projections is not allowed [bar*]");
1083+
expectError("from a | enrich countries on foo* ", "Using wildcards [*] in ENRICH WITH projections is not allowed, found [foo*]");
1084+
expectError("from a | enrich countries on * ", "Using wildcards [*] in ENRICH WITH projections is not allowed, found [*]");
1085+
expectError(
1086+
"from a | enrich countries on foo with bar*",
1087+
"Using wildcards [*] in ENRICH WITH projections is not allowed, found [bar*]"
1088+
);
1089+
expectError("from a | enrich countries on foo with *", "Using wildcards [*] in ENRICH WITH projections is not allowed, found [*]");
10851090
expectError(
10861091
"from a | enrich countries on foo with x = bar* ",
1087-
"Using wildcards [*] in ENRICH WITH projections is not allowed [bar*]"
1092+
"Using wildcards [*] in ENRICH WITH projections is not allowed, found [bar*]"
1093+
);
1094+
expectError(
1095+
"from a | enrich countries on foo with x = * ",
1096+
"Using wildcards [*] in ENRICH WITH projections is not allowed, found [*]"
10881097
);
10891098
expectError(
10901099
"from a | enrich countries on foo with x* = bar ",
1091-
"Using wildcards [*] in ENRICH WITH projections is not allowed [x*]"
1100+
"Using wildcards [*] in ENRICH WITH projections is not allowed, found [x*]"
1101+
);
1102+
expectError(
1103+
"from a | enrich countries on foo with * = bar ",
1104+
"Using wildcards [*] in ENRICH WITH projections is not allowed, found [*]"
10921105
);
10931106
expectError(
10941107
"from a | enrich typo:countries on foo",
@@ -1973,7 +1986,7 @@ public void testParamInInvalidPosition() {
19731986
expectError(
19741987
"from idx1 | " + enrich,
19751988
List.of(paramAsPattern("f1", pattern), paramAsIdentifier("f2", "f.2"), paramAsIdentifier("f3", "f.3*")),
1976-
"Using wildcards [*] in ENRICH WITH projections is not allowed [" + pattern + "]"
1989+
"Using wildcards [*] in ENRICH WITH projections is not allowed, found [" + pattern + "]"
19771990
);
19781991
expectError(
19791992
"from idx1 | " + enrich,

0 commit comments

Comments
 (0)