Skip to content

Commit 9283823

Browse files
committed
use bool
1 parent 039c8fe commit 9283823

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

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

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@
3737
import java.util.Map;
3838
import java.util.Set;
3939
import java.util.concurrent.Executor;
40-
import java.util.concurrent.atomic.AtomicInteger;
40+
import java.util.concurrent.atomic.AtomicBoolean;
4141
import java.util.concurrent.atomic.AtomicReference;
4242
import java.util.function.Function;
4343

@@ -82,10 +82,10 @@ void startComputeOnRemoteCluster(
8282
listener = ActionListener.runBefore(listener, exchangeSource.addEmptySink()::close);
8383
final var childSessionId = computeService.newChildSession(sessionId);
8484
final String clusterAlias = cluster.clusterAlias();
85-
final AtomicInteger pagesFetched = new AtomicInteger();
85+
final AtomicBoolean pagesFetched = new AtomicBoolean();
8686
final AtomicReference<ComputeResponse> finalResponse = new AtomicReference<>();
8787
listener = listener.delegateResponse((l, e) -> {
88-
final boolean receivedResults = finalResponse.get() != null || pagesFetched.get() > 0;
88+
final boolean receivedResults = finalResponse.get() != null || pagesFetched.get();
8989
if (receivedResults == false && EsqlCCSUtils.shouldIgnoreRuntimeError(executionInfo, clusterAlias, e)) {
9090
EsqlCCSUtils.markClusterWithFinalStateAndNoShards(executionInfo, clusterAlias, EsqlExecutionInfo.Cluster.Status.SKIPPED, e);
9191
l.onResponse(List.of());
@@ -122,7 +122,7 @@ void startComputeOnRemoteCluster(
122122
exchangeSource.addRemoteSink(
123123
remoteSink,
124124
failFast,
125-
pagesFetched::incrementAndGet,
125+
() -> pagesFetched.set(true),
126126
queryPragmas.concurrentExchangeClients(),
127127
computeListener.acquireAvoid()
128128
);

0 commit comments

Comments
 (0)