Skip to content

Commit ee57420

Browse files
authored
Adjust SearchRequest version checks (#38181)
The finalReduce flag is now supported on 6.x too, hence we need to update the version checks in master.
1 parent 7023583 commit ee57420

File tree

3 files changed

+5
-14
lines changed

3 files changed

+5
-14
lines changed

server/src/main/java/org/elasticsearch/action/search/SearchRequest.java

+3-9
Original file line numberDiff line numberDiff line change
@@ -205,17 +205,14 @@ public SearchRequest(StreamInput in) throws IOException {
205205
localClusterAlias = in.readOptionalString();
206206
if (localClusterAlias != null) {
207207
absoluteStartMillis = in.readVLong();
208+
finalReduce = in.readBoolean();
208209
} else {
209210
absoluteStartMillis = DEFAULT_ABSOLUTE_START_MILLIS;
211+
finalReduce = true;
210212
}
211213
} else {
212214
localClusterAlias = null;
213215
absoluteStartMillis = DEFAULT_ABSOLUTE_START_MILLIS;
214-
}
215-
//TODO move to the 6_7_0 branch once backported to 6.x
216-
if (in.getVersion().onOrAfter(Version.V_7_0_0)) {
217-
finalReduce = in.readBoolean();
218-
} else {
219216
finalReduce = true;
220217
}
221218
if (in.getVersion().onOrAfter(Version.V_7_0_0)) {
@@ -245,12 +242,9 @@ public void writeTo(StreamOutput out) throws IOException {
245242
out.writeOptionalString(localClusterAlias);
246243
if (localClusterAlias != null) {
247244
out.writeVLong(absoluteStartMillis);
245+
out.writeBoolean(finalReduce);
248246
}
249247
}
250-
//TODO move to the 6_7_0 branch once backported to 6.x
251-
if (out.getVersion().onOrAfter(Version.V_7_0_0)) {
252-
out.writeBoolean(finalReduce);
253-
}
254248
if (out.getVersion().onOrAfter(Version.V_7_0_0)) {
255249
out.writeBoolean(ccsMinimizeRoundtrips);
256250
}

server/src/test/java/org/elasticsearch/action/search/SearchRequestTests.java

+1-5
Original file line numberDiff line numberDiff line change
@@ -91,14 +91,10 @@ public void testRandomVersionSerialization() throws IOException {
9191
if (version.before(Version.V_6_7_0)) {
9292
assertNull(deserializedRequest.getLocalClusterAlias());
9393
assertAbsoluteStartMillisIsCurrentTime(deserializedRequest);
94+
assertTrue(deserializedRequest.isFinalReduce());
9495
} else {
9596
assertEquals(searchRequest.getLocalClusterAlias(), deserializedRequest.getLocalClusterAlias());
9697
assertEquals(searchRequest.getOrCreateAbsoluteStartMillis(), deserializedRequest.getOrCreateAbsoluteStartMillis());
97-
}
98-
//TODO move to the 6_7_0 branch once backported to 6.x
99-
if (version.before(Version.V_7_0_0)) {
100-
assertTrue(deserializedRequest.isFinalReduce());
101-
} else {
10298
assertEquals(searchRequest.isFinalReduce(), deserializedRequest.isFinalReduce());
10399
}
104100
}

server/src/test/java/org/elasticsearch/action/search/TransportSearchActionSingleNodeTests.java

+1
Original file line numberDiff line numberDiff line change
@@ -171,6 +171,7 @@ public void testFinalReduce() {
171171
assertEquals(2, searchResponse.getHits().getTotalHits().value);
172172
Aggregations aggregations = searchResponse.getAggregations();
173173
LongTerms longTerms = aggregations.get("terms");
174+
assertEquals(2, longTerms.getBuckets().size());
174175
}
175176
}
176177
}

0 commit comments

Comments
 (0)