chore(query): skip empty block for udf script (#16182)

databendlabs · Aug 5, 2024 · d462ca5 · d462ca5
1 parent c3047a6
commit d462ca5
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/query/service/src/pipelines/processors/transforms/transform_udf_script.rs b/src/query/service/src/pipelines/processors/transforms/transform_udf_script.rs
@@ -224,7 +224,13 @@ impl TransformUdfScript {
 impl Transform for TransformUdfScript {
     const NAME: &'static str = "UDFScriptTransform";
 
+    const SKIP_EMPTY_DATA_BLOCK: bool = true;
+
     fn transform(&mut self, mut data_block: DataBlock) -> Result<DataBlock> {
+        if data_block.is_empty() {
+            return Ok(data_block);
+        }
+
         let index = self.index_seq.fetch_add(1, Ordering::SeqCst);
         for func in &self.funcs {
             let num_rows = data_block.num_rows();