fix zero cnt

Signed-off-by: Kevin Axel <[email protected]>
risingwavelabs · Sep 22, 2023 · e21c086 · e21c086
1 parent c4d2fda
commit e21c086
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/expr/src/expr/expr_udf.rs b/src/expr/src/expr/expr_udf.rs
@@ -119,6 +119,14 @@ impl UdfExpression {
             uncompact_builder.append(array.datum_at(idx));
             last_u = Some(u);
         }
+        let zeros = if let Some(last_u) = last_u {
+            vis.len() - last_u - 1
+        } else {
+            vis.len()
+        };        
+        for _ in 0..zeros {
+            uncompact_builder.append_null();
+        }
         let array = uncompact_builder.finish();
         Ok(Arc::new(array))
     }