upd

Author: idegtiarenko

x-pack/plugin/esql/compute/src/main/java/org/elasticsearch/compute/operator/exchange/ExchangeSourceHandler.java

@@ -31,7 +31,6 @@
  */
 public final class ExchangeSourceHandler {
 
-    private volatile boolean completed = false;
     private final ExchangeBuffer buffer;
     private final Executor fetchExecutor;
 
@@ -58,8 +57,8 @@ public ExchangeSourceHandler(int maxBufferSize, Executor fetchExecutor) {
         this.outstandingSources = new PendingInstances(() -> finishEarly(true, ActionListener.noop()));
     }
 
-    public boolean isCompleted() {
-        return completed;
+    public boolean isFinished() {
+        return buffer.isFinished();
     }
 
     private void checkFailure() {
@@ -305,7 +304,6 @@ public Releasable addEmptySink() {
      * @param drainingPages whether to discard pages already fetched in the exchange
      */
     public void finishEarly(boolean drainingPages, ActionListener<Void> listener) {
-        completed = true;
         buffer.finish(drainingPages);
         try (EsqlRefCountingListener refs = new EsqlRefCountingListener(listener)) {
             for (RemoteSink remoteSink : remoteSinks.values()) {

x-pack/plugin/esql/src/internalClusterTest/java/org/elasticsearch/xpack/esql/action/ManyShardsIT.java

@@ -261,6 +261,7 @@ public void testLimitConcurrentShards() {
     }
 
     public void testCancelUnnecessaryRequests() {
+        assumeTrue("Requires pragmas", canUseQueryPragmas());
         internalCluster().ensureAtLeastNumDataNodes(3);
 
         var coordinatingNode = internalCluster().getNodeNames()[0];

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

@@ -112,7 +112,7 @@ protected void sendRequest(
                 Map<Index, AliasFilter> aliasFilters,
                 NodeListener nodeListener
             ) {
-                if (exchangeSource.isCompleted()) {
+                if (exchangeSource.isFinished()) {
                     nodeListener.onResponse(new DataNodeComputeResponse(List.of(), Map.of()));
                     return;
                 }