Function spark_unhex

datafusion/spark/src/function/math/unhex.rs:152–209

(args: &[ColumnarValue])

}

fn spark_unhex(args: &[ColumnarValue]) -> Result<ColumnarValue, DataFusionError> {
    let [args] = take_function_args("unhex", args)?;

    match args {
        ColumnarValue::Array(array) => match array.data_type() {
            DataType::Utf8 => {
                let array = as_string_array(array)?;
                let capacity = array.values().len().div_ceil(2);
                Ok(ColumnarValue::Array(unhex_array(
                    array.iter(),
                    array.len(),
                    capacity,
                )?))
            }
            DataType::Utf8View => {
                let array = as_string_view_array(array)?;
                // Estimate capacity since StringViewArray data can be scattered or inlined.
                let capacity = array.len() * 32;
                Ok(ColumnarValue::Array(unhex_array(
                    array.iter(),
                    array.len(),
                    capacity,
                )?))
            }
            DataType::LargeUtf8 => {
                let array = as_large_string_array(array)?;
                let capacity = array.values().len().div_ceil(2);
                Ok(ColumnarValue::Array(unhex_array(
                    array.iter(),
                    array.len(),
                    capacity,
                )?))
            }
            _ => exec_err!(
                "unhex only supports string argument, but got: {}",
                array.data_type()
            ),
        },
        ColumnarValue::Scalar(sv) => match sv {
            ScalarValue::Utf8(None)
            | ScalarValue::Utf8View(None)
            | ScalarValue::LargeUtf8(None) => {
                Ok(ColumnarValue::Scalar(ScalarValue::Binary(None)))
            }
            ScalarValue::Utf8(Some(s))
            | ScalarValue::Utf8View(Some(s))
            | ScalarValue::LargeUtf8(Some(s)) => {
                Ok(ColumnarValue::Scalar(ScalarValue::Binary(unhex_scalar(s))))
            }
            _ => {
                exec_err!(
                    "unhex only supports string argument, but got: {}",
                    sv.data_type()
                )
            }
        },
    }
}