Skip to content

Commit f2530ea

Browse files
committed
fix
1 parent db12a2b commit f2530ea

File tree

1 file changed

+5
-11
lines changed

1 file changed

+5
-11
lines changed

src/query/service/src/pipelines/processors/transforms/transform_partition_collect.rs

Lines changed: 5 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -227,11 +227,6 @@ impl Processor for TransformHilbertPartitionCollect {
227227
return Ok(Event::NeedConsume);
228228
}
229229

230-
if self.need_spill() {
231-
self.state = State::Spill;
232-
return Ok(Event::Async);
233-
}
234-
235230
if let Some(data_block) = self.output_data_blocks.pop() {
236231
self.output.push_data(Ok(data_block));
237232
return Ok(Event::NeedConsume);
@@ -256,14 +251,13 @@ impl Processor for TransformHilbertPartitionCollect {
256251
return Ok(Event::Finished);
257252
}
258253

254+
if self.need_spill() {
255+
self.state = State::Spill;
256+
return Ok(Event::Async);
257+
}
258+
259259
if self.input.has_data() {
260260
self.collect_data_block()?;
261-
262-
if self.need_spill() {
263-
self.state = State::Spill;
264-
return Ok(Event::Async);
265-
}
266-
267261
if !self.immediate_output_blocks.is_empty() {
268262
self.state = State::Flush;
269263
return Ok(Event::Async);

0 commit comments

Comments
 (0)