From d462ca5391cb9cd3c366cf332329c09f60980da8 Mon Sep 17 00:00:00 2001
From: Winter Zhang <coswde@gmail.com>
Date: Mon, 5 Aug 2024 16:27:47 +0800
Subject: [PATCH] chore(query): skip empty block for udf script (#16182)

---
 .../pipelines/processors/transforms/transform_udf_script.rs | 6 ++++++
 1 file changed, 6 insertions(+)

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
index c0894ab0b81b..41809bc87104 100644
--- 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();