From 91ff4b129547c72b8899c0dcd3009bc9091181e0 Mon Sep 17 00:00:00 2001 From: Gal Lalouche Date: Sun, 29 Jun 2025 13:15:15 +0300 Subject: [PATCH] ESQL: Allow ExchangeSinkOperator and LimitOperator in Driver assertion --- .../main/java/org/elasticsearch/compute/operator/Driver.java | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/x-pack/plugin/esql/compute/src/main/java/org/elasticsearch/compute/operator/Driver.java b/x-pack/plugin/esql/compute/src/main/java/org/elasticsearch/compute/operator/Driver.java index 775ac401cd916..0ad3c64fcaf09 100644 --- a/x-pack/plugin/esql/compute/src/main/java/org/elasticsearch/compute/operator/Driver.java +++ b/x-pack/plugin/esql/compute/src/main/java/org/elasticsearch/compute/operator/Driver.java @@ -267,7 +267,8 @@ private IsBlockedResult runSingleLoopIteration() { if (op.isFinished() == false && nextOp.needsInput()) { driverContext.checkForEarlyTermination(); - assert nextOp.isFinished() == false : "next operator should not be finished yet: " + nextOp; + assert nextOp.isFinished() == false || nextOp instanceof ExchangeSinkOperator || nextOp instanceof LimitOperator + : "next operator should not be finished yet: " + nextOp; Page page = op.getOutput(); if (page == null) { // No result, just move to the next iteration