Skip to content

Commit 9603a7c

Browse files
Fix failing UTs
1 parent d24dab3 commit 9603a7c

File tree

6 files changed

+10
-7
lines changed

6 files changed

+10
-7
lines changed

server/src/main/java/org/elasticsearch/TransportVersions.java

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -364,6 +364,7 @@ static TransportVersion def(int id) {
364364
public static final TransportVersion EXTENDED_SNAPSHOT_STATS_IN_NODE_INFO = def(9_137_0_00);
365365
public static final TransportVersion SIMULATE_INGEST_MAPPING_MERGE_TYPE = def(9_138_0_00);
366366
public static final TransportVersion ESQL_LOOKUP_JOIN_ON_MANY_FIELDS = def(9_139_0_00);
367+
public static final TransportVersion ESQL_LOOKUP_JOIN_PRE_JOIN_FILTER = def(9_140_0_00);
367368

368369
/*
369370
* STOP! READ THIS FIRST! No, really,

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/enrich/ExpressionQueryList.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -85,8 +85,8 @@ private void buildPrePostJoinFilter(PhysicalPlan rightPreJoinPlan, ClusterServic
8585
} catch (IOException e) {
8686
throw new IllegalArgumentException("Failed to translate pre-join filter: " + esQueryExec, e);
8787
}
88-
} else {
89-
throw new IllegalArgumentException("Unsupported pre-join filter type: " + preJoinFilters.getClass().getName());
88+
} else if (rightPreJoinPlan != null) {
89+
throw new IllegalArgumentException("Unsupported pre-join filter type: " + rightPreJoinPlan.getClass().getName());
9090
}
9191
}
9292

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/enrich/LookupFromIndexOperator.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ public String describe() {
6262
.append(" inputChannel=")
6363
.append(matchField.channel());
6464
}
65-
stringBuilder.append(" rightPreJoinPlan=").append(rightPreJoinPlan);
65+
stringBuilder.append(" right_pre_join_plan=").append(rightPreJoinPlan);
6666
stringBuilder.append("]");
6767
return stringBuilder.toString();
6868
}

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/enrich/LookupFromIndexService.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -220,7 +220,7 @@ static TransportRequest readFrom(StreamInput in, BlockFactory blockFactory) thro
220220
source = new Source(source.source(), sourceText);
221221
}
222222
PhysicalPlan rightPreJoinPlan = null;
223-
if (in.getTransportVersion().onOrAfter(TransportVersions.ESQL_LOOKUP_JOIN_ON_MANY_FIELDS)) {
223+
if (in.getTransportVersion().onOrAfter(TransportVersions.ESQL_LOOKUP_JOIN_PRE_JOIN_FILTER)) {
224224
rightPreJoinPlan = planIn.readOptionalNamedWriteable(PhysicalPlan.class);
225225
}
226226
TransportRequest result = new TransportRequest(
@@ -275,7 +275,7 @@ public void writeTo(StreamOutput out) throws IOException {
275275
if (out.getTransportVersion().onOrAfter(TransportVersions.ESQL_LOOKUP_JOIN_SOURCE_TEXT)) {
276276
out.writeString(source.text());
277277
}
278-
if (out.getTransportVersion().onOrAfter(TransportVersions.ESQL_LOOKUP_JOIN_ON_MANY_FIELDS)) {
278+
if (out.getTransportVersion().onOrAfter(TransportVersions.ESQL_LOOKUP_JOIN_PRE_JOIN_FILTER)) {
279279
planOut.writeOptionalNamedWriteable(rightPreJoinPlan);
280280
} else if (rightPreJoinPlan != null) {
281281
throw new EsqlIllegalArgumentException("LOOKUP JOIN with pre-join filter is not supported on remote node");

x-pack/plugin/esql/src/test/java/org/elasticsearch/xpack/esql/enrich/LookupFromIndexOperatorTests.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -187,7 +187,7 @@ protected Matcher<String> expectedToStringOfSimple() {
187187
for (int i = 0; i < numberOfJoinColumns; i++) {
188188
sb.append(" input_type=LONG match_field=match").append(i).append(" inputChannel=").append(i);
189189
}
190-
sb.append("]");
190+
sb.append(" right_pre_join_plan=null]");
191191
return matchesPattern(sb.toString());
192192
}
193193

x-pack/plugin/esql/src/test/java/org/elasticsearch/xpack/esql/optimizer/LogicalPlanOptimizerTests.java

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7055,7 +7055,9 @@ public void testLookupJoinPushDownSeparatedForConjunctionBetweenLeftAndRightFiel
70557055
assertThat(literal.value(), equalTo(1));
70567056

70577057
var leftRel = as(filter.child(), EsRelation.class);
7058-
var rightRel = as(join.right(), EsRelation.class);
7058+
var filterRight = as(join.right(), Filter.class);
7059+
assertEquals("language_name == \"English\"", filterRight.condition().toString());
7060+
var rightRel = as(filterRight.child(), EsRelation.class);
70597061
}
70607062

70617063
/**

0 commit comments

Comments
 (0)