fix

databendlabs · Feb 8, 2025 · f2530ea · f2530ea
1 parent db12a2b
commit f2530ea
Showing 1 changed file with 5 additions and 11 deletions.
diff --git a/src/query/service/src/pipelines/processors/transforms/transform_partition_collect.rs b/src/query/service/src/pipelines/processors/transforms/transform_partition_collect.rs
@@ -227,11 +227,6 @@ impl Processor for TransformHilbertPartitionCollect {
             return Ok(Event::NeedConsume);
         }
 
-        if self.need_spill() {
-            self.state = State::Spill;
-            return Ok(Event::Async);
-        }
-
         if let Some(data_block) = self.output_data_blocks.pop() {
             self.output.push_data(Ok(data_block));
             return Ok(Event::NeedConsume);
@@ -256,14 +251,13 @@ impl Processor for TransformHilbertPartitionCollect {
             return Ok(Event::Finished);
         }
 
+        if self.need_spill() {
+            self.state = State::Spill;
+            return Ok(Event::Async);
+        }
+
         if self.input.has_data() {
             self.collect_data_block()?;
-
-            if self.need_spill() {
-                self.state = State::Spill;
-                return Ok(Event::Async);
-            }
-
             if !self.immediate_output_blocks.is_empty() {
                 self.state = State::Flush;
                 return Ok(Event::Async);