Skip to content

Commit 9973de3

Browse files
committed
more cleanup
Signed-off-by: Atri Sharma <atri.jiit@gmail.com>
1 parent 3f78994 commit 9973de3

File tree

3 files changed

+35
-13
lines changed

3 files changed

+35
-13
lines changed

server/src/main/java/org/opensearch/action/search/QueryPhaseResultConsumer.java

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@ public class QueryPhaseResultConsumer extends ArraySearchPhaseResults<SearchPhas
7777
private final Executor executor;
7878
private final CircuitBreaker circuitBreaker;
7979
private final SearchPhaseController controller;
80-
protected final SearchProgressListener progressListener;
80+
private final SearchProgressListener progressListener;
8181
private final ReduceContextBuilder aggReduceContextBuilder;
8282
private final NamedWriteableRegistry namedWriteableRegistry;
8383

@@ -151,6 +151,14 @@ int getBatchReduceSize(int requestBatchedReduceSize, int minBatchReduceSize) {
151151
return (hasAggs || hasTopDocs) ? Math.min(requestBatchedReduceSize, minBatchReduceSize) : minBatchReduceSize;
152152
}
153153

154+
/**
155+
* Protected accessor for progressListener to allow subclasses to access it.
156+
* @return the search progress listener
157+
*/
158+
protected SearchProgressListener progressListener() {
159+
return this.progressListener;
160+
}
161+
154162
@Override
155163
public void close() {
156164
Releasables.close(pendingMerges);

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

Lines changed: 13 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -281,9 +281,14 @@ public SearchRequest(StreamInput in) throws IOException {
281281
if (in.getVersion().onOrAfter(Version.V_2_12_0)) {
282282
phaseTook = in.readOptionalBoolean();
283283
}
284-
// Read streaming fields
285-
streamingScoring = in.readBoolean();
286-
streamingSearchMode = in.readOptionalString();
284+
// Read streaming fields - gated on version for BWC
285+
if (in.getVersion().onOrAfter(Version.V_3_3_0)) {
286+
streamingScoring = in.readBoolean();
287+
streamingSearchMode = in.readOptionalString();
288+
} else {
289+
streamingScoring = false;
290+
streamingSearchMode = null;
291+
}
287292
}
288293

289294
@Override
@@ -318,9 +323,11 @@ public void writeTo(StreamOutput out) throws IOException {
318323
if (out.getVersion().onOrAfter(Version.V_2_12_0)) {
319324
out.writeOptionalBoolean(phaseTook);
320325
}
321-
// Write streaming fields
322-
out.writeBoolean(streamingScoring);
323-
out.writeOptionalString(streamingSearchMode);
326+
// Write streaming fields - gated on version for BWC
327+
if (out.getVersion().onOrAfter(Version.V_3_3_0)) {
328+
out.writeBoolean(streamingScoring);
329+
out.writeOptionalString(streamingSearchMode);
330+
}
324331
}
325332

326333
@Override

server/src/main/java/org/opensearch/search/internal/ShardSearchRequest.java

Lines changed: 13 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -276,9 +276,14 @@ public ShardSearchRequest(StreamInput in) throws IOException {
276276
keepAlive = in.readOptionalTimeValue();
277277
originalIndices = OriginalIndices.readOriginalIndices(in);
278278
assert keepAlive == null || readerId != null : "readerId: " + readerId + " keepAlive: " + keepAlive;
279-
// Read streaming fields
280-
streamingSearch = in.readBoolean();
281-
streamingSearchMode = in.readOptionalString();
279+
// Read streaming fields - gated on version for BWC
280+
if (in.getVersion().onOrAfter(Version.V_3_3_0)) {
281+
streamingSearch = in.readBoolean();
282+
streamingSearchMode = in.readOptionalString();
283+
} else {
284+
streamingSearch = false;
285+
streamingSearchMode = null;
286+
}
282287
}
283288

284289
public ShardSearchRequest(ShardSearchRequest clone) {
@@ -311,9 +316,11 @@ public void writeTo(StreamOutput out) throws IOException {
311316
super.writeTo(out);
312317
innerWriteTo(out, false);
313318
OriginalIndices.writeOriginalIndices(originalIndices, out);
314-
// Write streaming fields after OriginalIndices
315-
out.writeBoolean(streamingSearch);
316-
out.writeOptionalString(streamingSearchMode);
319+
// Write streaming fields after OriginalIndices - gated on version for BWC
320+
if (out.getVersion().onOrAfter(Version.V_3_3_0)) {
321+
out.writeBoolean(streamingSearch);
322+
out.writeOptionalString(streamingSearchMode);
323+
}
317324
}
318325

319326
protected final void innerWriteTo(StreamOutput out, boolean asKey) throws IOException {

0 commit comments

Comments
 (0)