decimal_types_bool_cast_native_support

Author: Bhargava Vadlamani

native/spark-expr/src/conversion_funcs/cast.rs

@@ -1015,7 +1015,7 @@ fn cast_array(
         {
             spark_cast_nonintegral_numeric_to_integral(&array, eval_mode, from_type, to_type)
         }
-        (Decimal128(_p, _s), Boolean) => spark_cast_decimal_to_boolean(&array, from_type, to_type),
+        (Decimal128(_p, _s), Boolean) => spark_cast_decimal_to_boolean(&array),
         (Utf8View, Utf8) => Ok(cast_with_options(&array, to_type, &CAST_OPTIONS)?),
         (Struct(_), Utf8) => Ok(casts_struct_to_string(array.as_struct(), cast_options)?),
         (Struct(_), Struct(_)) => Ok(cast_struct_to_struct(
@@ -1529,29 +1529,19 @@ where
     Ok(Arc::new(output_array))
 }
 
-fn spark_cast_decimal_to_boolean(
-    array: &dyn Array,
-    from_type: &DataType,
-    to_type: &DataType,
-) -> SparkResult<ArrayRef> {
-    match (from_type, to_type) {
-        (DataType::Decimal128(_p, _s), DataType::Boolean) => {
-            let decimal_array = array.as_primitive::<Decimal128Type>();
-            let mut result = BooleanBuilder::with_capacity(decimal_array.len());
-            for i in 0..decimal_array.len() {
-                if decimal_array.is_null(i) {
-                    result.append_null()
-                } else if decimal_array.value(i).is_zero() {
-                    result.append_value(false);
-                } else {
-                    result.append_value(true);
-                }
-            }
-            Ok(Arc::new(result.finish()))
+fn spark_cast_decimal_to_boolean(array: &dyn Array) -> SparkResult<ArrayRef> {
+    let decimal_array = array.as_primitive::<Decimal128Type>();
+    let mut result = BooleanBuilder::with_capacity(decimal_array.len());
+    for i in 0..decimal_array.len() {
+        if decimal_array.is_null(i) {
+            result.append_null()
+        } else {
+            result.append_value(!decimal_array.value(i).is_zero());
         }
-        _ => panic!("invalid cast from decimal type: {from_type} to boolean type: {to_type}"),
     }
+    Ok(Arc::new(result.finish()))
 }
+
 fn spark_cast_nonintegral_numeric_to_integral(
     array: &dyn Array,
     eval_mode: EvalMode,