Method invoke_with_args

datafusion/functions/src/unicode/translate.rs:93–158

(&self, args: ScalarFunctionArgs)

    }

    fn invoke_with_args(&self, args: ScalarFunctionArgs) -> Result<ColumnarValue> {
        // When from and to are scalars, pre-build the translation map once
        if let (Some(from_str), Some(to_str)) = (
            try_as_scalar_str(&args.args[1]),
            try_as_scalar_str(&args.args[2]),
        ) {
            let to_chars: Vec<char> = to_str.chars().collect();

            let mut from_map: HashMap<char, usize> = HashMap::new();
            for (index, c) in from_str.chars().enumerate() {
                from_map.entry(c).or_insert(index);
            }

            let ascii_table = build_ascii_translate_table(from_str, to_str);

            let string_array = args.args[0].to_array_of_size(args.number_rows)?;
            let len = string_array.len();

            let result = match string_array.data_type() {
                DataType::Utf8View => {
                    let arr = string_array.as_string_view();
                    let builder = StringViewBuilder::with_capacity(len);
                    translate_with_map(
                        arr,
                        &from_map,
                        &to_chars,
                        ascii_table.as_ref(),
                        builder,
                    )
                }
                DataType::Utf8 => {
                    let arr = string_array.as_string::<i32>();
                    let builder =
                        StringBuilder::with_capacity(len, arr.value_data().len());
                    translate_with_map(
                        arr,
                        &from_map,
                        &to_chars,
                        ascii_table.as_ref(),
                        builder,
                    )
                }
                DataType::LargeUtf8 => {
                    let arr = string_array.as_string::<i64>();
                    let builder =
                        LargeStringBuilder::with_capacity(len, arr.value_data().len());
                    translate_with_map(
                        arr,
                        &from_map,
                        &to_chars,
                        ascii_table.as_ref(),
                        builder,
                    )
                }
                other => {
                    return exec_err!(
                        "Unsupported data type {other:?} for function translate"
                    );