[spark tests]

lakehq · Oct 10, 2024 · b194325 · b194325
1 parent 706d873
commit b194325
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/crates/sail-plan/src/extension/function/spark_murmur3_hash.rs b/crates/sail-plan/src/extension/function/spark_murmur3_hash.rs
@@ -38,6 +38,11 @@ impl ScalarUDFImpl for SparkMurmur3Hash {
 
     fn invoke(&self, args: &[ColumnarValue]) -> Result<ColumnarValue> {
         let length = args.len();
+        if length < 1 {
+            return Err(datafusion::error::DataFusionError::Internal(
+                "spark_hash requires at least one argument".to_string(),
+            ));
+        }
         let seed = &args[length - 1];
         let mut args = args.to_vec();
         match seed {

diff --git a/crates/sail-plan/src/extension/function/spark_xxhash64.rs b/crates/sail-plan/src/extension/function/spark_xxhash64.rs
@@ -38,6 +38,11 @@ impl ScalarUDFImpl for SparkXxhash64 {
 
     fn invoke(&self, args: &[ColumnarValue]) -> Result<ColumnarValue> {
         let length = args.len();
+        if length < 1 {
+            return Err(datafusion::error::DataFusionError::Internal(
+                "spark_xxhash64 requires at least one argument".to_string(),
+            ));
+        }
         let seed = &args[length - 1];
         let mut args = args.to_vec();
         match seed {