diff --git a/x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/plugin/ClusterComputeHandler.java b/x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/plugin/ClusterComputeHandler.java index e41dd42c7579d..3b599a87afaff 100644 --- a/x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/plugin/ClusterComputeHandler.java +++ b/x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/plugin/ClusterComputeHandler.java @@ -85,7 +85,6 @@ void startComputeOnRemoteCluster( var resp = finalResponse.get(); return Objects.requireNonNullElseGet(resp, () -> new ComputeResponse(profiles)); }))) { - var openExchangeListener = computeListener.acquireAvoid(); ExchangeService.openExchange( transportService, cluster.connection, @@ -93,7 +92,7 @@ void startComputeOnRemoteCluster( queryPragmas.exchangeBufferSize(), esqlExecutor, EsqlCCSUtils.skipUnavailableListener( - openExchangeListener, + computeListener.acquireAvoid(), executionInfo, clusterAlias, EsqlExecutionInfo.Cluster.Status.SKIPPED @@ -104,7 +103,7 @@ void startComputeOnRemoteCluster( computeListener, clusterAlias, executionInfo, - openExchangeListener + l ); var remoteSink = exchangeService.newRemoteSink( diff --git a/x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/plugin/RemoteListenerGroup.java b/x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/plugin/RemoteListenerGroup.java index 9ef085257b87b..54ab269bb91ae 100644 --- a/x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/plugin/RemoteListenerGroup.java +++ b/x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/plugin/RemoteListenerGroup.java @@ -74,7 +74,7 @@ class RemoteListenerGroup { private ActionListener createCancellingListener(String reason, ActionListener delegate, Runnable finishGroup) { return ActionListener.runAfter(delegate.delegateResponse((inner, e) -> { taskManager.cancelTaskAndDescendants(groupTask, reason, true, ActionListener.running(() -> { - EsqlCCSUtils.skipUnavailableListener(delegate, executionInfo, clusterAlias, EsqlExecutionInfo.Cluster.Status.PARTIAL) + EsqlCCSUtils.skipUnavailableListener(inner, executionInfo, clusterAlias, EsqlExecutionInfo.Cluster.Status.PARTIAL) .onFailure(e); })); }), finishGroup);