be/src/exprs/function/cast/cast_to_decimal.h

Source
// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License.  You may obtain a copy of the License at
//
//   http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License.

#pragma once

#include <cmath>
#include <type_traits>

#include "common/status.h"
#include "core/data_type/data_type_decimal.h"
#include "core/data_type/data_type_number.h"
#include "core/types.h"
#include "exprs/function/cast/cast_to_basic_number_common.h"

namespace doris {

#define DECIMAL_CONVERT_OVERFLOW_ERROR(value, from_type_name, precision, scale)                    \
    Status(ErrorCode::ARITHMETIC_OVERFLOW_ERRROR,                                                  \
           fmt::format(                                                                            \
                   "Arithmetic overflow when converting value {} from type {} to decimal({}, {})", \
                   value, from_type_name, precision, scale))

struct CastToDecimal {
    template <typename ToCppT>
        requires(IsDecimalNumber<ToCppT>)
    static inline bool from_string(const StringRef& from, ToCppT& to, UInt32 precision,
                                   UInt32 scale, CastParameters& params) {
        if constexpr (IsDecimalV2<ToCppT>) {
            StringParser::ParseResult result = StringParser::PARSE_SUCCESS;
            to = DecimalV2Value(StringParser::string_to_decimal<TYPE_DECIMALV2>(
                    from.data, (int)from.size, DecimalV2Value::PRECISION, DecimalV2Value::SCALE,
                    &result));
            return result == StringParser::PARSE_SUCCESS;
        }

        if constexpr (IsDecimal32<ToCppT>) {
            StringParser::ParseResult result = StringParser::PARSE_SUCCESS;
            to.value = StringParser::string_to_decimal<TYPE_DECIMAL32>(from.data, (int)from.size,
                                                                       precision, scale, &result);
            return result == StringParser::PARSE_SUCCESS;
        }

        if constexpr (IsDecimal64<ToCppT>) {
            StringParser::ParseResult result = StringParser::PARSE_SUCCESS;
            to.value = StringParser::string_to_decimal<TYPE_DECIMAL64>(from.data, (int)from.size,
                                                                       precision, scale, &result);
            return result == StringParser::PARSE_SUCCESS;
        }

        if constexpr (IsDecimal128V3<ToCppT>) {
            StringParser::ParseResult result = StringParser::PARSE_SUCCESS;
            to.value = StringParser::string_to_decimal<TYPE_DECIMAL128I>(from.data, (int)from.size,
                                                                         precision, scale, &result);
            return result == StringParser::PARSE_SUCCESS;
        }

        if constexpr (IsDecimal256<ToCppT>) {
            StringParser::ParseResult result = StringParser::PARSE_SUCCESS;
            to.value = StringParser::string_to_decimal<TYPE_DECIMAL256>(from.data, (int)from.size,
                                                                        precision, scale, &result);
            return result == StringParser::PARSE_SUCCESS;
        }
    }

    // cast int to decimal
    template <typename FromCppT, typename ToCppT,
              typename MaxNativeType =
                      std::conditional_t<(sizeof(FromCppT) > sizeof(typename ToCppT::NativeType)),
                                         FromCppT, typename ToCppT::NativeType>>
        requires(IsDecimalNumber<ToCppT> &&
                 (IsCppTypeInt<FromCppT> || std::is_same_v<FromCppT, UInt8>))
    static inline bool from_int(const FromCppT& from, ToCppT& to, UInt32 to_precision,
                                UInt32 to_scale, CastParameters& params) {
        MaxNativeType scale_multiplier =
                DataTypeDecimal<ToCppT::PType>::get_scale_multiplier(to_scale);
        typename ToCppT::NativeType max_result =
                DataTypeDecimal<ToCppT::PType>::get_max_digits_number(to_precision);
        typename ToCppT::NativeType min_result = -max_result;

        UInt32 from_precision = NumberTraits::max_ascii_len<FromCppT>();
        constexpr UInt32 from_scale = 0;
        constexpr UInt32 to_max_digits = NumberTraits::max_ascii_len<typename ToCppT::NativeType>();

        auto from_max_int_digit_count = from_precision - from_scale;
        auto to_max_int_digit_count = to_precision - to_scale;
        bool narrow_integral = (to_max_int_digit_count < from_max_int_digit_count);
        bool multiply_may_overflow = false;
        if (to_scale > from_scale) {
            multiply_may_overflow = (from_precision + to_scale - from_scale) >= to_max_digits;
        }
        return std::visit(
                [&](auto multiply_may_overflow, auto narrow_integral) {
                    return _from_int<FromCppT, ToCppT, multiply_may_overflow, narrow_integral>(
                            from, to, to_precision, to_scale, scale_multiplier, min_result,
                            max_result, params);
                },
                make_bool_variant(multiply_may_overflow), make_bool_variant(narrow_integral));
    }

    // cast bool to decimal
    template <typename FromCppT, typename ToCppT,
              typename MaxNativeType =
                      std::conditional_t<(sizeof(FromCppT) > sizeof(typename ToCppT::NativeType)),
                                         FromCppT, typename ToCppT::NativeType>>
        requires(IsDecimalNumber<ToCppT> && std::is_same_v<FromCppT, UInt8>)
    static inline bool from_bool(const FromCppT& from, ToCppT& to, UInt32 to_precision,
                                 UInt32 to_scale, CastParameters& params) {
        return from_int<FromCppT, ToCppT, MaxNativeType>(from, to, to_precision, to_scale, params);
    }

    template <typename FromCppT, typename ToCppT>
        requires(IsDecimalNumber<ToCppT> && IsCppTypeFloat<FromCppT>)
    static inline bool from_float(const FromCppT& from, ToCppT& to, UInt32 to_precision,
                                  UInt32 to_scale, CastParameters& params) {
        typename ToCppT::NativeType scale_multiplier =
                DataTypeDecimal<ToCppT::PType>::get_scale_multiplier(to_scale);
        typename ToCppT::NativeType max_result =
                DataTypeDecimal<ToCppT::PType>::get_max_digits_number(to_precision);
        typename ToCppT::NativeType min_result = -max_result;

        return _from_float(from, to, to_precision, to_scale, scale_multiplier, min_result,
                           max_result, params);
    }

    template <typename FromCppT, typename ToCppT,
              typename MaxFieldType = std::conditional_t<
                      (sizeof(FromCppT) == sizeof(ToCppT)) &&
                              (std::is_same_v<ToCppT, Decimal128V3> ||
                               std::is_same_v<FromCppT, Decimal128V3>),
                      Decimal128V3,
                      std::conditional_t<(sizeof(FromCppT) > sizeof(ToCppT)), FromCppT, ToCppT>>>
        requires(IsDecimalNumber<ToCppT> && IsDecimalNumber<FromCppT>)
    static inline bool from_decimal(const FromCppT& from, const UInt32 from_precision,
                                    const UInt32 from_scale, ToCppT& to, UInt32 to_precision,
                                    UInt32 to_scale, CastParameters& params) {
        return _from_decimalv3<FromCppT, ToCppT, MaxFieldType>(from, from_precision, from_scale, to,
                                                               to_precision, to_scale, params);
    }

    template <typename FromCppT, typename ToCppT,
              typename MaxFieldType = std::conditional_t<
                      (sizeof(FromCppT) == sizeof(ToCppT)) &&
                              (std::is_same_v<ToCppT, Decimal128V3> ||
                               std::is_same_v<FromCppT, Decimal128V3>),
                      Decimal128V3,
                      std::conditional_t<(sizeof(FromCppT) > sizeof(ToCppT)), FromCppT, ToCppT>>>
        requires(IsDecimalNumber<ToCppT> && IsDecimalNumber<FromCppT>)
    static inline bool _from_decimalv3(const FromCppT& from, const UInt32 from_precision,
                                       const UInt32 from_scale, ToCppT& to, UInt32 to_precision,
                                       UInt32 to_scale, CastParameters& params) {
        using MaxNativeType = typename MaxFieldType::NativeType;

        auto from_max_int_digit_count = from_precision - from_scale;
        auto to_max_int_digit_count = to_precision - to_scale;
        bool narrow_integral =
                (to_max_int_digit_count < from_max_int_digit_count) ||
                (to_max_int_digit_count == from_max_int_digit_count && to_scale < from_scale);

        UInt32 to_max_digits = NumberTraits::max_ascii_len<typename ToCppT::NativeType>();
        bool multiply_may_overflow = false;
        if (to_scale > from_scale) {
            multiply_may_overflow = (from_precision + to_scale - from_scale) >= to_max_digits;
        }

        typename ToCppT::NativeType max_result =
                DataTypeDecimal<ToCppT::PType>::get_max_digits_number(to_precision);
        typename ToCppT::NativeType min_result = -max_result;

        MaxNativeType multiplier {};
        if (from_scale < to_scale) {
            multiplier = DataTypeDecimal<MaxFieldType::PType>::get_scale_multiplier(to_scale -
                                                                                    from_scale);
        } else if (from_scale > to_scale) {
            multiplier = DataTypeDecimal<MaxFieldType::PType>::get_scale_multiplier(from_scale -
                                                                                    to_scale);
        }

        return std::visit(
                [&](auto multiply_may_overflow, auto narrow_integral) {
                    return _from_decimal<FromCppT, ToCppT, multiply_may_overflow, narrow_integral>(
                            from, from_precision, from_scale, to, to_precision, to_scale,
                            min_result, max_result, multiplier, params);
                },
                make_bool_variant(multiply_may_overflow), make_bool_variant(narrow_integral));
    }

    template <typename FromCppT, typename ToCppT, bool multiply_may_overflow, bool narrow_integral,
              typename MaxFieldType = std::conditional_t<
                      (sizeof(FromCppT) == sizeof(ToCppT)) &&
                              (std::is_same_v<ToCppT, Decimal128V3> ||
                               std::is_same_v<FromCppT, Decimal128V3>),
                      Decimal128V3,
                      std::conditional_t<(sizeof(FromCppT) > sizeof(ToCppT)), FromCppT, ToCppT>>>
        requires(IsDecimalNumber<ToCppT> && IsDecimalNumber<FromCppT>)
    static inline bool _from_decimal(const FromCppT& from, const UInt32 from_precision,
                                     const UInt32 from_scale, ToCppT& to, UInt32 to_precision,
                                     UInt32 to_scale, const ToCppT::NativeType& min_result,
                                     const ToCppT::NativeType& max_result,
                                     const typename MaxFieldType::NativeType& scale_multiplier,
                                     CastParameters& params) {
        using MaxNativeType = typename MaxFieldType::NativeType;

        if (from_scale < to_scale) {
            return _from_decimal_smaller_scale<FromCppT, ToCppT, multiply_may_overflow,
                                               narrow_integral>(
                    from, from_precision, from_scale, to, to_precision, to_scale, scale_multiplier,
                    min_result, max_result, params);
        } else if (from_scale == to_scale) {
            return _from_decimal_same_scale<FromCppT, ToCppT, MaxNativeType, narrow_integral>(
                    from, from_precision, from_scale, to, to_precision, to_scale, min_result,
                    max_result, params);
        } else {
            return _from_decimal_bigger_scale<FromCppT, ToCppT, multiply_may_overflow,
                                              narrow_integral>(
                    from, from_precision, from_scale, to, to_precision, to_scale, scale_multiplier,
                    min_result, max_result, params);
        }
        return true;
    }

    template <typename FromCppT, typename ToCppT>
        requires(IsDecimalNumber<ToCppT> && IsCppTypeFloat<FromCppT> && !IsDecimal128V2<ToCppT>)
    static inline bool _from_float(const FromCppT& from, ToCppT& to, UInt32 to_precision,
                                   UInt32 to_scale,
                                   const typename ToCppT::NativeType& scale_multiplier,
                                   const typename ToCppT::NativeType& min_result,
                                   const typename ToCppT::NativeType& max_result,
                                   CastParameters& params) {
        if (!std::isfinite(from)) {
            params.status = Status(ErrorCode::ARITHMETIC_OVERFLOW_ERRROR,
                                   "Decimal convert overflow. Cannot convert infinity or NaN "
                                   "to decimal");
            return false;
        }
        using DoubleType = std::conditional_t<IsDecimal256<ToCppT>, long double, double>;
        DoubleType tmp = from * static_cast<DoubleType>(scale_multiplier);
        if (tmp <= DoubleType(min_result) || tmp >= DoubleType(max_result)) {
            if (params.is_strict) {
                params.status = DECIMAL_CONVERT_OVERFLOW_ERROR(from, "float/double", to_precision,
                                                               to_scale);
            }
            return false;
        }
        to.value = static_cast<typename ToCppT::NativeType>(static_cast<double>(
                from * static_cast<DoubleType>(scale_multiplier) + ((from >= 0) ? 0.5 : -0.5)));
        return true;
    }

    template <typename FromCppT, typename ToCppT>
        requires(IsDecimal128V2<ToCppT> && IsCppTypeFloat<FromCppT>)
    static inline bool _from_float(const FromCppT& from, ToCppT& to, UInt32 to_precision,
                                   UInt32 to_scale,
                                   const typename ToCppT::NativeType& scale_multiplier,
                                   const typename ToCppT::NativeType& min_result,
                                   const typename ToCppT::NativeType& max_result,
                                   CastParameters& params) {
        if (!std::isfinite(from)) {
            params.status = Status(ErrorCode::ARITHMETIC_OVERFLOW_ERRROR,
                                   "Decimal convert overflow. Cannot convert infinity or NaN "
                                   "to decimal");
            return false;
        }
        using DoubleType = std::conditional_t<IsDecimal256<ToCppT>, long double, double>;
        DoubleType tmp = from * static_cast<DoubleType>(scale_multiplier);
        if (tmp <= DoubleType(min_result) || tmp >= DoubleType(max_result)) {
            if (params.is_strict) {
                params.status = DECIMAL_CONVERT_OVERFLOW_ERROR(from, "float/double", to_precision,
                                                               to_scale);
            }
            return false;
        }
        to = DecimalV2Value(static_cast<typename ToCppT::NativeType>(static_cast<double>(
                from * static_cast<DoubleType>(scale_multiplier) + ((from >= 0) ? 0.5 : -0.5))));
        return true;
    }

    template <typename FromCppT, typename ToCppT,
              typename MaxFieldType = std::conditional_t<
                      (sizeof(FromCppT) == sizeof(ToCppT)) &&
                              (std::is_same_v<ToCppT, Decimal128V3> ||
                               std::is_same_v<FromCppT, Decimal128V3>),
                      Decimal128V3,
                      std::conditional_t<(sizeof(FromCppT) > sizeof(ToCppT)), FromCppT, ToCppT>>>
        requires(IsDecimalNumber<ToCppT> && IsDecimalNumber<FromCppT>)
    static inline bool from_decimal(const FromCppT& from, const UInt32 from_precision,
                                    const UInt32 from_scale, UInt32 from_original_precision,
                                    UInt32 from_original_scale, ToCppT& to, UInt32 to_precision,
                                    UInt32 to_scale, CastParameters& params) {
        return _from_decimalv2<FromCppT, ToCppT, MaxFieldType>(
                from, from_precision, from_scale, from_original_precision, from_original_scale, to,
                to_precision, to_scale, params);
    }

    template <typename FromCppT, typename ToCppT,
              typename MaxFieldType = std::conditional_t<
                      (sizeof(FromCppT) == sizeof(ToCppT)) &&
                              (std::is_same_v<ToCppT, Decimal128V3> ||
                               std::is_same_v<FromCppT, Decimal128V3>),
                      Decimal128V3,
                      std::conditional_t<(sizeof(FromCppT) > sizeof(ToCppT)), FromCppT, ToCppT>>>
        requires(IsDecimalNumber<ToCppT> && IsDecimalNumber<FromCppT>)
    static inline bool _from_decimalv2(const FromCppT& from, const UInt32 from_precision,
                                       const UInt32 from_scale, UInt32 from_original_precision,
                                       UInt32 from_original_scale, ToCppT& to, UInt32 to_precision,
                                       UInt32 to_scale, CastParameters& params) {
        using MaxNativeType = typename MaxFieldType::NativeType;

        auto from_max_int_digit_count = from_original_precision - from_original_scale;
        auto to_max_int_digit_count = to_precision - to_scale;
        bool narrow_integral = (to_max_int_digit_count < from_max_int_digit_count) ||
                               (to_max_int_digit_count == from_max_int_digit_count &&
                                to_scale < from_original_scale);

        constexpr UInt32 to_max_digits = NumberTraits::max_ascii_len<typename ToCppT::NativeType>();
        bool multiply_may_overflow = false;
        if (to_scale > from_scale) {
            multiply_may_overflow = (from_precision + to_scale - from_scale) >= to_max_digits;
        }

        typename ToCppT::NativeType max_result =
                DataTypeDecimal<ToCppT::PType>::get_max_digits_number(to_precision);
        typename ToCppT::NativeType min_result = -max_result;

        return std::visit(
                [&](auto multiply_may_overflow, auto narrow_integral) {
                    if (from_scale < to_scale) {
                        MaxNativeType multiplier =
                                DataTypeDecimal<MaxFieldType::PType>::get_scale_multiplier(
                                        to_scale - from_scale);
                        return _from_decimal_smaller_scale<FromCppT, ToCppT, multiply_may_overflow,
                                                           narrow_integral>(
                                from, from_precision, from_scale, to, to_precision, to_scale,
                                multiplier, min_result, max_result, params);
                    } else if (from_scale == to_scale) {
                        return _from_decimal_same_scale<FromCppT, ToCppT, MaxNativeType,
                                                        narrow_integral>(
                                from, from_precision, from_scale, to, to_precision, to_scale,
                                min_result, max_result, params);
                    } else {
                        MaxNativeType multiplier =
                                DataTypeDecimal<MaxFieldType::PType>::get_scale_multiplier(
                                        from_scale - to_scale);
                        return _from_decimal_bigger_scale<FromCppT, ToCppT, multiply_may_overflow,
                                                          narrow_integral>(
                                from, from_precision, from_scale, to, to_precision, to_scale,
                                multiplier, min_result, max_result, params);
                    }
                    return true;
                },
                make_bool_variant(multiply_may_overflow), make_bool_variant(narrow_integral));
    }

    template <typename FromCppT, typename ToCppT, bool multiply_may_overflow, bool narrow_integral,
              typename MaxNativeType =
                      std::conditional_t<(sizeof(FromCppT) > sizeof(typename ToCppT::NativeType)),
                                         FromCppT, typename ToCppT::NativeType>>
        requires(IsDecimalNumber<ToCppT> && !IsDecimal128V2<ToCppT> &&
                 (IsCppTypeInt<FromCppT> || std::is_same_v<FromCppT, UInt8>))
    static inline bool _from_int(const FromCppT& from, ToCppT& to, UInt32 precision, UInt32 scale,
                                 const MaxNativeType& scale_multiplier,
                                 const typename ToCppT::NativeType& min_result,
                                 const typename ToCppT::NativeType& max_result,
                                 CastParameters& params) {
        MaxNativeType tmp;
        if constexpr (multiply_may_overflow) {
            if (common::mul_overflow(static_cast<MaxNativeType>(from), scale_multiplier, tmp)) {
                if (params.is_strict) {
                    params.status = DECIMAL_CONVERT_OVERFLOW_ERROR(from, int_type_name<FromCppT>,
                                                                   precision, scale);
                }
                return false;
            }
            if constexpr (narrow_integral) {
                if (tmp < min_result || tmp > max_result) {
                    if (params.is_strict) {
                        params.status = DECIMAL_CONVERT_OVERFLOW_ERROR(
                                from, int_type_name<FromCppT>, precision, scale);
                    }
                    return false;
                }
            }
            to.value = static_cast<typename ToCppT::NativeType>(tmp);
        } else {
            tmp = scale_multiplier * from;
            if constexpr (narrow_integral) {
                if (tmp < min_result || tmp > max_result) {
                    if (params.is_strict) {
                        params.status = DECIMAL_CONVERT_OVERFLOW_ERROR(
                                from, int_type_name<FromCppT>, precision, scale);
                    }
                    return false;
                }
            }
            to.value = static_cast<typename ToCppT::NativeType>(tmp);
        }

        return true;
    }

    template <typename FromCppT, typename ToCppT, bool multiply_may_overflow, bool narrow_integral,
              typename MaxNativeType =
                      std::conditional_t<(sizeof(FromCppT) > sizeof(typename ToCppT::NativeType)),
                                         FromCppT, typename ToCppT::NativeType>>
        requires(IsDecimalV2<ToCppT> && (IsCppTypeInt<FromCppT> || std::is_same_v<FromCppT, UInt8>))
    static inline bool _from_int(const FromCppT& from, ToCppT& to, UInt32 precision, UInt32 scale,
                                 const MaxNativeType& scale_multiplier,
                                 const typename ToCppT::NativeType& min_result,
                                 const typename ToCppT::NativeType& max_result,
                                 CastParameters& params) {
        MaxNativeType tmp;
        if constexpr (multiply_may_overflow) {
            if (common::mul_overflow(static_cast<MaxNativeType>(from), scale_multiplier, tmp)) {
                if (params.is_strict) {
                    params.status = DECIMAL_CONVERT_OVERFLOW_ERROR(from, int_type_name<FromCppT>,
                                                                   precision, scale);
                }
                return false;
            }
            if constexpr (narrow_integral) {
                if (tmp < min_result || tmp > max_result) {
                    if (params.is_strict) {
                        params.status = DECIMAL_CONVERT_OVERFLOW_ERROR(
                                from, int_type_name<FromCppT>, precision, scale);
                    }
                    return false;
                }
            }
            to = DecimalV2Value(static_cast<typename ToCppT::NativeType>(tmp));
        } else {
            tmp = scale_multiplier * from;
            if constexpr (narrow_integral) {
                if (tmp < min_result || tmp > max_result) {
                    if (params.is_strict) {
                        params.status = DECIMAL_CONVERT_OVERFLOW_ERROR(
                                from, int_type_name<FromCppT>, precision, scale);
                    }
                    return false;
                }
            }
            to = DecimalV2Value(static_cast<typename ToCppT::NativeType>(tmp));
        }

        return true;
    }

    template <
            typename FromCppT, typename ToCppT, bool multiply_may_overflow, bool narrow_integral,
            typename MaxNativeType = std::conditional_t<
                    (sizeof(FromCppT) == sizeof(ToCppT)) &&
                            (std::is_same_v<ToCppT, Decimal128V3> ||
                             std::is_same_v<FromCppT, Decimal128V3>),
                    Decimal128V3::NativeType,
                    std::conditional_t<(sizeof(FromCppT) > sizeof(ToCppT)),
                                       typename FromCppT::NativeType, typename ToCppT::NativeType>>>
        requires(IsDecimalNumber<ToCppT> && IsDecimalNumber<FromCppT>)
    static inline bool _from_decimal_smaller_scale(
            const FromCppT& from, const UInt32 precision_from, const UInt32 scale_from, ToCppT& to,
            UInt32 precision_to, UInt32 scale_to, const MaxNativeType& scale_multiplier,
            const typename ToCppT::NativeType& min_result,
            const typename ToCppT::NativeType& max_result, CastParameters& params) {
        MaxNativeType res;
        if constexpr (multiply_may_overflow) {
            if constexpr (IsDecimal128V2<FromCppT>) {
                if (common::mul_overflow(static_cast<MaxNativeType>(from.value()), scale_multiplier,
                                         res)) {
                    if (params.is_strict) {
                        params.status = DECIMAL_CONVERT_OVERFLOW_ERROR(
                                decimal_to_string(from.value(), scale_from),
                                fmt::format("decimal({}, {})", precision_from, scale_from),
                                precision_to, scale_to);
                    }
                    return false;
                } else {
                    if (UNLIKELY(res > max_result || res < -max_result)) {
                        if (params.is_strict) {
                            params.status = DECIMAL_CONVERT_OVERFLOW_ERROR(
                                    decimal_to_string(from.value(), scale_from),
                                    fmt::format("decimal({}, {})", precision_from, scale_from),
                                    precision_to, scale_to);
                        }
                        return false;
                    } else {
                        to = ToCppT(res);
                    }
                }
            } else {
                if (common::mul_overflow(static_cast<MaxNativeType>(from.value), scale_multiplier,
                                         res)) {
                    if (params.is_strict) {
                        params.status = DECIMAL_CONVERT_OVERFLOW_ERROR(
                                decimal_to_string(from.value, scale_from),
                                fmt::format("decimal({}, {})", precision_from, scale_from),
                                precision_to, scale_to);
                    }
                    return false;
                } else {
                    if (UNLIKELY(res > max_result || res < -max_result)) {
                        if (params.is_strict) {
                            params.status = DECIMAL_CONVERT_OVERFLOW_ERROR(
                                    decimal_to_string(from.value, scale_from),
                                    fmt::format("decimal({}, {})", precision_from, scale_from),
                                    precision_to, scale_to);
                        }
                        return false;
                    } else {
                        to = ToCppT(res);
                    }
                }
            }
        } else {
            if constexpr (IsDecimal128V2<FromCppT>) {
                res = from.value() * scale_multiplier;
            } else {
                res = from.value * scale_multiplier;
            }
            if constexpr (narrow_integral) {
                if (UNLIKELY(res > max_result || res < -max_result)) {
                    if constexpr (IsDecimal128V2<FromCppT>) {
                        if (params.is_strict) {
                            params.status = DECIMAL_CONVERT_OVERFLOW_ERROR(
                                    decimal_to_string(from.value(), scale_from),
                                    fmt::format("decimal({}, {})", precision_from, scale_from),
                                    precision_to, scale_to);
                        }
                    } else {
                        if (params.is_strict) {
                            params.status = DECIMAL_CONVERT_OVERFLOW_ERROR(
                                    decimal_to_string(from.value, scale_from),
                                    fmt::format("decimal({}, {})", precision_from, scale_from),
                                    precision_to, scale_to);
                        }
                    }
                    return false;
                }
            }
            to = ToCppT(res);
        }
        return true;
    }

    template <typename FromCppT, typename ToCppT, typename ScaleT, bool narrow_integral>
        requires(IsDecimalNumber<ToCppT> && IsDecimalNumber<FromCppT>)
    static inline bool _from_decimal_same_scale(const FromCppT& from, const UInt32 precision_from,
                                                const UInt32 scale_from, ToCppT& to,
                                                UInt32 precision_to, UInt32 scale_to,
                                                const typename ToCppT::NativeType& min_result,
                                                const typename ToCppT::NativeType& max_result,
                                                CastParameters& params) {
        if constexpr (IsDecimal128V2<FromCppT>) {
            if constexpr (narrow_integral) {
                if (UNLIKELY(from.value() > max_result || from.value() < -max_result)) {
                    if (params.is_strict) {
                        params.status = DECIMAL_CONVERT_OVERFLOW_ERROR(
                                decimal_to_string(from.value(), scale_from),
                                fmt::format("decimal({}, {})", precision_from, scale_from),
                                precision_to, scale_to);
                    }
                    return false;
                }
            }
            to = ToCppT(from.value());
        } else {
            if constexpr (narrow_integral) {
                if (UNLIKELY(from.value > max_result || from.value < -max_result)) {
                    if (params.is_strict) {
                        params.status = DECIMAL_CONVERT_OVERFLOW_ERROR(
                                decimal_to_string(from.value, scale_from),
                                fmt::format("decimal({}, {})", precision_from, scale_from),
                                precision_to, scale_to);
                    }
                    return false;
                }
            }
            to = ToCppT(from.value);
        }
        return true;
    }

    template <
            typename FromCppT, typename ToCppT, bool multiply_may_overflow, bool narrow_integral,
            typename MaxNativeType = std::conditional_t<
                    (sizeof(FromCppT) == sizeof(ToCppT)) &&
                            (std::is_same_v<ToCppT, Decimal128V3> ||
                             std::is_same_v<FromCppT, Decimal128V3>),
                    Decimal128V3::NativeType,
                    std::conditional_t<(sizeof(FromCppT) > sizeof(ToCppT)),
                                       typename FromCppT::NativeType, typename ToCppT::NativeType>>>
        requires(IsDecimalNumber<ToCppT> && IsDecimalNumber<FromCppT>)
    static inline bool _from_decimal_bigger_scale(const FromCppT& from, const UInt32 precision_from,
                                                  const UInt32 scale_from, ToCppT& to,
                                                  UInt32 precision_to, UInt32 scale_to,
                                                  const MaxNativeType& scale_multiplier,
                                                  const typename ToCppT::NativeType& min_result,
                                                  const typename ToCppT::NativeType& max_result,
                                                  CastParameters& params) {
        MaxNativeType res;
        if (from >= FromCppT(0)) {
            if constexpr (narrow_integral) {
                if constexpr (IsDecimal128V2<FromCppT>) {
                    res = (from.value() + scale_multiplier / 2) / scale_multiplier;
                    if (UNLIKELY(res > max_result)) {
                        if (params.is_strict) {
                            params.status = DECIMAL_CONVERT_OVERFLOW_ERROR(
                                    decimal_to_string(from.value(), scale_from),
                                    fmt::format("decimal({}, {})", precision_from, scale_from),
                                    precision_to, scale_to);
                        }
                        return false;
                    }
                } else {
                    res = (from.value + scale_multiplier / 2) / scale_multiplier;
                    if (UNLIKELY(res > max_result)) {
                        if (params.is_strict) {
                            params.status = DECIMAL_CONVERT_OVERFLOW_ERROR(
                                    decimal_to_string(from.value, scale_from),
                                    fmt::format("decimal({}, {})", precision_from, scale_from),
                                    precision_to, scale_to);
                        }
                        return false;
                    }
                }
                to = ToCppT(res);
            } else {
                if constexpr (IsDecimal128V2<FromCppT>) {
                    to = ToCppT((from.value() + scale_multiplier / 2) / scale_multiplier);
                } else {
                    to = ToCppT((from.value + scale_multiplier / 2) / scale_multiplier);
                }
            }
        } else {
            if constexpr (narrow_integral) {
                if constexpr (IsDecimal128V2<FromCppT>) {
                    res = (from.value() - scale_multiplier / 2) / scale_multiplier;
                    if (UNLIKELY(res < -max_result)) {
                        if (params.is_strict) {
                            params.status = DECIMAL_CONVERT_OVERFLOW_ERROR(
                                    decimal_to_string(from.value(), scale_from),
                                    fmt::format("decimal({}, {})", precision_from, scale_from),
                                    precision_to, scale_to);
                        }
                        return false;
                    }
                } else {
                    res = (from.value - scale_multiplier / 2) / scale_multiplier;
                    if (UNLIKELY(res < -max_result)) {
                        if (params.is_strict) {
                            params.status = DECIMAL_CONVERT_OVERFLOW_ERROR(
                                    decimal_to_string(from.value, scale_from),
                                    fmt::format("decimal({}, {})", precision_from, scale_from),
                                    precision_to, scale_to);
                        }
                        return false;
                    }
                }
                to = ToCppT(res);
            } else {
                if constexpr (IsDecimal128V2<FromCppT>) {
                    to = ToCppT((from.value() - scale_multiplier / 2) / scale_multiplier);
                } else {
                    to = ToCppT((from.value - scale_multiplier / 2) / scale_multiplier);
                }
            }
        }
        return true;
    }
};

// Casting from string to decimal types.
template <CastModeType Mode, typename ToDataType>
    requires(IsDataTypeDecimal<ToDataType>)
class CastToImpl<Mode, DataTypeString, ToDataType> : public CastToBase {
    Status execute_impl(FunctionContext* context, Block& block, const ColumnNumbers& arguments,
                        uint32_t result, size_t input_rows_count,
                        const NullMap::value_type* null_map = nullptr) const override {
        const auto& named_from = block.get_by_position(arguments[0]);
        const auto* col_from =
                check_and_get_column<DataTypeString::ColumnType>(named_from.column.get());
        const NullMap::value_type* source_null_map = null_map;
        if (!col_from) {
            const auto* nullable_col_from =
                    check_and_get_column<ColumnNullable>(named_from.column.get());
            if (nullable_col_from) {
                col_from = check_and_get_column<DataTypeString::ColumnType>(
                        &nullable_col_from->get_nested_column());
                source_null_map = nullable_col_from->get_null_map_data().data();
            }
        }
        if (!col_from) {
            return Status::RuntimeError("Illegal column {} of first argument of function cast",
                                        named_from.column->get_name());
        }

        auto to_type = block.get_by_position(result).type;
        auto serde = remove_nullable(to_type)->get_serde();

        // by default framework, to_type is already unwrapped nullable
        MutableColumnPtr column_to = to_type->create_column();
        ColumnNullable::MutablePtr nullable_col_to = ColumnNullable::create(
                std::move(column_to), ColumnUInt8::create(input_rows_count, 0));

        if constexpr (Mode == CastModeType::NonStrictMode) {
            // may write nulls to nullable_col_to
            RETURN_IF_ERROR(serde->from_string_batch(*col_from, *nullable_col_to, {}));
        } else if constexpr (Mode == CastModeType::StrictMode) {
            // WON'T write nulls to nullable_col_to, just raise errors. null_map is only used to skip invalid rows
            RETURN_IF_ERROR(serde->from_string_strict_mode_batch(
                    *col_from, nullable_col_to->get_nested_column(), {}, source_null_map));
        } else {
            return Status::InternalError("Unsupported cast mode");
        }
        if (source_null_map) {
            auto& null_map_to = nullable_col_to->get_null_map_data();
            for (size_t i = 0; i < input_rows_count; ++i) {
                null_map_to[i] |= source_null_map[i];
            }
        }

        block.get_by_position(result).column = std::move(nullable_col_to);
        return Status::OK();
    }
};

// cast bool and int to decimal. when may overflow, result column is nullable.
template <CastModeType CastMode, typename FromDataType, typename ToDataType>
    requires(IsDataTypeDecimal<ToDataType> &&
             (IsDataTypeInt<FromDataType> || IsDataTypeBool<FromDataType>))
class CastToImpl<CastMode, FromDataType, ToDataType> : public CastToBase {
public:
    Status execute_impl(FunctionContext* context, Block& block, const ColumnNumbers& arguments,
                        uint32_t result, size_t input_rows_count,
                        const NullMap::value_type* null_map = nullptr) const override {
        using FromFieldType = typename FromDataType::FieldType;
        using ToFieldType = typename ToDataType::FieldType;
        const ColumnWithTypeAndName& named_from = block.get_by_position(arguments[0]);
        const auto* col_from =
                check_and_get_column<typename FromDataType::ColumnType>(named_from.column.get());
        if (!col_from) {
            return Status::RuntimeError("Illegal column {} of first argument of function cast",
                                        named_from.column->get_name());
        }

        UInt32 from_precision = NumberTraits::max_ascii_len<FromFieldType>();
        constexpr UInt32 from_scale = 0;

        const ColumnWithTypeAndName& named_to = block.get_by_position(result);
        const auto& to_decimal_type = assert_cast<const ToDataType&>(*named_to.type);
        UInt32 to_precision = to_decimal_type.get_precision();
        ToDataType::check_type_precision(to_precision);
        UInt32 to_scale = to_decimal_type.get_scale();
        ToDataType::check_type_scale(to_scale);

        auto from_max_int_digit_count = from_precision - from_scale;
        auto to_max_int_digit_count = to_precision - to_scale;
        // may overflow. nullable result column.
        bool narrow_integral = (to_max_int_digit_count < from_max_int_digit_count);
        // only in non-strict mode and may overflow, we set nullable
        bool set_nullable = (CastMode == CastModeType::NonStrictMode) && narrow_integral;

        constexpr UInt32 to_max_digits =
                NumberTraits::max_ascii_len<typename ToFieldType::NativeType>();
        bool multiply_may_overflow = false;
        if (to_scale > from_scale) {
            multiply_may_overflow = (from_precision + to_scale - from_scale) >= to_max_digits;
        }
        using MaxNativeType = std::conditional_t<(sizeof(FromFieldType) >
                                                  sizeof(typename ToFieldType::NativeType)),
                                                 FromFieldType, typename ToFieldType::NativeType>;
        MaxNativeType scale_multiplier =
                DataTypeDecimal<ToFieldType::PType>::get_scale_multiplier(to_scale);
        typename ToFieldType::NativeType max_result =
                DataTypeDecimal<ToFieldType::PType>::get_max_digits_number(to_precision);
        typename ToFieldType::NativeType min_result = -max_result;

        ColumnUInt8::MutablePtr col_null_map_to;
        NullMap::value_type* null_map_data = nullptr;
        if (narrow_integral) {
            col_null_map_to = ColumnUInt8::create(input_rows_count, 0);
            null_map_data = col_null_map_to->get_data().data();
        }

        auto col_to = ToDataType::ColumnType::create(input_rows_count, to_scale);
        const auto& vec_from = col_from->get_data();
        const auto* vec_from_data = vec_from.data();
        auto& vec_to = col_to->get_data();
        auto* vec_to_data = vec_to.data();

        CastParameters params;
        params.is_strict = (CastMode == CastModeType::StrictMode);
        size_t size = vec_from.size();

        RETURN_IF_ERROR(std::visit(
                [&](auto multiply_may_overflow, auto narrow_integral) {
                    for (size_t i = 0; i < size; i++) {
                        if (!CastToDecimal::_from_int<typename FromDataType::FieldType,
                                                      typename ToDataType::FieldType,
                                                      multiply_may_overflow, narrow_integral>(
                                    vec_from_data[i], vec_to_data[i], to_precision, to_scale,
                                    scale_multiplier, min_result, max_result, params)) {
                            if (set_nullable) {
                                null_map_data[i] = 1;
                            } else {
                                return params.status;
                            }
                        }
                    }
                    return Status::OK();
                },
                make_bool_variant(multiply_may_overflow), make_bool_variant(narrow_integral)));

        if (narrow_integral) {
            block.get_by_position(result).column =
                    ColumnNullable::create(std::move(col_to), std::move(col_null_map_to));
        } else {
            block.get_by_position(result).column = std::move(col_to);
        }
        return Status::OK();
    }
};

// cast float and double to decimal. ALWAYS nullable result column.
template <CastModeType CastMode, typename FromDataType, typename ToDataType>
    requires(IsDataTypeDecimal<ToDataType> && IsDataTypeFloat<FromDataType>)
class CastToImpl<CastMode, FromDataType, ToDataType> : public CastToBase {
public:
    Status execute_impl(FunctionContext* context, Block& block, const ColumnNumbers& arguments,
                        uint32_t result, size_t input_rows_count,
                        const NullMap::value_type* null_map = nullptr) const override {
        using FromFieldType = typename FromDataType::FieldType;
        using ToFieldType = typename ToDataType::FieldType;
        const ColumnWithTypeAndName& named_from = block.get_by_position(arguments[0]);
        const auto* col_from =
                check_and_get_column<typename FromDataType::ColumnType>(named_from.column.get());
        if (!col_from) {
            return Status::RuntimeError("Illegal column {} of first argument of function cast",
                                        named_from.column->get_name());
        }

        UInt32 from_precision = NumberTraits::max_ascii_len<FromFieldType>();
        UInt32 from_scale = 0;

        const ColumnWithTypeAndName& named_to = block.get_by_position(result);
        const auto& to_decimal_type = assert_cast<const ToDataType&>(*named_to.type);
        UInt32 to_precision = to_decimal_type.get_precision();
        ToDataType::check_type_precision(to_precision);
        UInt32 to_scale = to_decimal_type.get_scale();
        ToDataType::check_type_scale(to_scale);

        auto from_max_int_digit_count = from_precision - from_scale;
        auto to_max_int_digit_count = to_precision - to_scale;
        bool narrow_integral =
                (to_max_int_digit_count < from_max_int_digit_count) ||
                (to_max_int_digit_count == from_max_int_digit_count && to_scale < from_scale);
        // only in non-strict mode and may overflow, we set nullable
        bool set_nullable = (CastMode == CastModeType::NonStrictMode) && narrow_integral;

        ColumnUInt8::MutablePtr col_null_map_to = ColumnUInt8::create(input_rows_count, 0);
        NullMap::value_type* null_map_data = col_null_map_to->get_data().data();

        auto col_to = ToDataType::ColumnType::create(input_rows_count, to_scale);
        const auto& vec_from = col_from->get_data();
        const auto* vec_from_data = vec_from.data();
        auto& vec_to = col_to->get_data();
        auto* vec_to_data = vec_to.data();

        CastParameters params;
        params.is_strict = (CastMode == CastModeType::StrictMode);
        size_t size = vec_from.size();

        typename ToFieldType::NativeType scale_multiplier =
                DataTypeDecimal<ToFieldType::PType>::get_scale_multiplier(to_scale);
        typename ToFieldType::NativeType max_result =
                DataTypeDecimal<ToFieldType::PType>::get_max_digits_number(to_precision);
        typename ToFieldType::NativeType min_result = -max_result;
        for (size_t i = 0; i < size; i++) {
            if (!CastToDecimal::_from_float<typename FromDataType::FieldType,
                                            typename ToDataType::FieldType>(
                        vec_from_data[i], vec_to_data[i], to_precision, to_scale, scale_multiplier,
                        min_result, max_result, params)) {
                if (set_nullable) {
                    null_map_data[i] = 1;
                } else {
                    return params.status;
                }
            }
        }

        block.get_by_position(result).column =
                ColumnNullable::create(std::move(col_to), std::move(col_null_map_to));
        return Status::OK();
    }
};

// cast decimalv3 types to decimalv2 types
template <CastModeType CastMode, typename FromDataType, typename ToDataType>
    requires(IsDataTypeDecimalV3<ToDataType> && IsDataTypeDecimalV2<ToDataType>)
class CastToImpl<CastMode, FromDataType, ToDataType> : public CastToBase {
public:
    Status execute_impl(FunctionContext* context, Block& block, const ColumnNumbers& arguments,
                        uint32_t result, size_t input_rows_count,
                        const NullMap::value_type* null_map = nullptr) const override {
        return Status::RuntimeError(
                "not support {} ",
                cast_mode_type_to_string(CastMode, block.get_by_position(arguments[0]).type,
                                         block.get_by_position(result).type));
    }
};

// cast decimalv2 types to decimalv3 types
template <CastModeType CastMode, typename FromDataType, typename ToDataType>
    requires(IsDataTypeDecimalV2<FromDataType> && IsDataTypeDecimalV3<ToDataType>)
class CastToImpl<CastMode, FromDataType, ToDataType> : public CastToBase {
public:
    Status execute_impl(FunctionContext* context, Block& block, const ColumnNumbers& arguments,
                        uint32_t result, size_t input_rows_count,
                        const NullMap::value_type* null_map = nullptr) const override {
        using FromFieldType = typename FromDataType::FieldType;
        using ToFieldType = typename ToDataType::FieldType;
        const ColumnWithTypeAndName& named_from = block.get_by_position(arguments[0]);
        const auto* col_from =
                check_and_get_column<typename FromDataType::ColumnType>(named_from.column.get());
        if (!col_from) {
            return Status::RuntimeError("Illegal column {} of first argument of function cast",
                                        named_from.column->get_name());
        }

        const auto& from_decimal_type = assert_cast<const FromDataType&>(*named_from.type);
        UInt32 from_precision = from_decimal_type.get_precision();
        UInt32 from_scale = from_decimal_type.get_scale();
        UInt32 from_original_precision = from_decimal_type.get_original_precision();
        UInt32 from_original_scale = from_decimal_type.get_original_scale();

        const ColumnWithTypeAndName& named_to = block.get_by_position(result);
        const auto& to_decimal_type = assert_cast<const ToDataType&>(*named_to.type);
        UInt32 to_precision = to_decimal_type.get_precision();
        ToDataType::check_type_precision(to_precision);
        UInt32 to_scale = to_decimal_type.get_scale();
        ToDataType::check_type_scale(to_scale);

        auto from_max_int_digit_count = from_original_precision - from_original_scale;
        auto to_max_int_digit_count = to_precision - to_scale;
        bool narrow_integral = (to_max_int_digit_count < from_max_int_digit_count) ||
                               (to_max_int_digit_count == from_max_int_digit_count &&
                                to_scale < from_original_scale);
        // only in non-strict mode and may overflow, we set nullable
        bool set_nullable = (CastMode == CastModeType::NonStrictMode) && narrow_integral;

        size_t size = col_from->size();
        ColumnUInt8::MutablePtr col_null_map_to;
        NullMap::value_type* null_map_data = nullptr;
        if (narrow_integral) {
            col_null_map_to = ColumnUInt8::create(size, 0);
            null_map_data = col_null_map_to->get_data().data();
        }
        CastParameters params;
        params.is_strict = (CastMode == CastModeType::StrictMode);
        auto col_to = ToDataType::ColumnType::create(size, to_scale);
        const auto& vec_from = col_from->get_data();
        const auto* vec_from_data = vec_from.data();
        auto& vec_to = col_to->get_data();
        auto* vec_to_data = vec_to.data();

        using MaxFieldType =
                std::conditional_t<(sizeof(FromFieldType) == sizeof(ToFieldType)) &&
                                           (std::is_same_v<ToFieldType, Decimal128V3> ||
                                            std::is_same_v<FromFieldType, Decimal128V3>),
                                   Decimal128V3,
                                   std::conditional_t<(sizeof(FromFieldType) > sizeof(ToFieldType)),
                                                      FromFieldType, ToFieldType>>;
        using MaxNativeType = typename MaxFieldType::NativeType;

        constexpr UInt32 to_max_digits =
                NumberTraits::max_ascii_len<typename ToFieldType::NativeType>();
        bool multiply_may_overflow = false;
        if (to_scale > from_scale) {
            multiply_may_overflow = (from_precision + to_scale - from_scale) >= to_max_digits;
        }

        typename ToFieldType::NativeType max_result =
                DataTypeDecimal<ToFieldType::PType>::get_max_digits_number(to_precision);
        typename ToFieldType::NativeType min_result = -max_result;

        MaxNativeType multiplier {};
        if (from_scale < to_scale) {
            multiplier = DataTypeDecimal<MaxFieldType::PType>::get_scale_multiplier(to_scale -
                                                                                    from_scale);
        } else if (from_scale > to_scale) {
            multiplier = DataTypeDecimal<MaxFieldType::PType>::get_scale_multiplier(from_scale -
                                                                                    to_scale);
        }
        RETURN_IF_ERROR(std::visit(
                [&](auto multiply_may_overflow, auto narrow_integral) {
                    for (size_t i = 0; i < size; i++) {
                        if (!CastToDecimal::_from_decimal<FromFieldType, ToFieldType,
                                                          multiply_may_overflow, narrow_integral>(
                                    vec_from_data[i], from_precision, from_scale, vec_to_data[i],
                                    to_precision, to_scale, min_result, max_result, multiplier,
                                    params)) {
                            if (set_nullable) {
                                null_map_data[i] = 1;
                            } else {
                                return params.status;
                            }
                        }
                    }
                    return Status::OK();
                },
                make_bool_variant(multiply_may_overflow), make_bool_variant(narrow_integral)));
        if (narrow_integral) {
            block.get_by_position(result).column =
                    ColumnNullable::create(std::move(col_to), std::move(col_null_map_to));
        } else {
            block.get_by_position(result).column = std::move(col_to);
        }
        return Status::OK();
    }
};

// cast between decimalv3 types
template <CastModeType CastMode, typename FromDataType, typename ToDataType>
    requires(IsDataTypeDecimalV3<ToDataType> && IsDataTypeDecimalV3<FromDataType>)
class CastToImpl<CastMode, FromDataType, ToDataType> : public CastToBase {
public:
    Status execute_impl(FunctionContext* context, Block& block, const ColumnNumbers& arguments,
                        uint32_t result, size_t input_rows_count,
                        const NullMap::value_type* null_map = nullptr) const override {
        using FromFieldType = typename FromDataType::FieldType;
        using ToFieldType = typename ToDataType::FieldType;
        const ColumnWithTypeAndName& named_from = block.get_by_position(arguments[0]);
        const auto* col_from =
                check_and_get_column<typename FromDataType::ColumnType>(named_from.column.get());
        if (!col_from) {
            return Status::RuntimeError("Illegal column {} of first argument of function cast",
                                        named_from.column->get_name());
        }

        const auto& from_decimal_type = assert_cast<const FromDataType&>(*named_from.type);
        UInt32 from_precision = from_decimal_type.get_precision();
        UInt32 from_scale = from_decimal_type.get_scale();

        const ColumnWithTypeAndName& named_to = block.get_by_position(result);
        const auto& to_decimal_type = assert_cast<const ToDataType&>(*named_to.type);
        UInt32 to_precision = to_decimal_type.get_precision();
        ToDataType::check_type_precision(to_precision);
        UInt32 to_scale = to_decimal_type.get_scale();
        ToDataType::check_type_scale(to_scale);

        auto from_max_int_digit_count = from_precision - from_scale;
        auto to_max_int_digit_count = to_precision - to_scale;
        bool narrow_integral =
                (to_max_int_digit_count < from_max_int_digit_count) ||
                (to_max_int_digit_count == from_max_int_digit_count && to_scale < from_scale);
        // only in non-strict mode and may overflow, we set nullable
        bool set_nullable = (CastMode == CastModeType::NonStrictMode) && narrow_integral;

        size_t size = col_from->size();
        ColumnUInt8::MutablePtr col_null_map_to;
        NullMap::value_type* null_map_data = nullptr;
        if (narrow_integral) {
            col_null_map_to = ColumnUInt8::create(size, 0);
            null_map_data = col_null_map_to->get_data().data();
        }
        CastParameters params;
        params.is_strict = (CastMode == CastModeType::StrictMode);
        auto col_to = ToDataType::ColumnType::create(size, to_scale);
        const auto& vec_from = col_from->get_data();
        const auto* vec_from_data = vec_from.data();
        auto& vec_to = col_to->get_data();
        auto* vec_to_data = vec_to.data();

        using MaxFieldType =
                std::conditional_t<(sizeof(FromFieldType) == sizeof(ToFieldType)) &&
                                           (std::is_same_v<ToFieldType, Decimal128V3> ||
                                            std::is_same_v<FromFieldType, Decimal128V3>),
                                   Decimal128V3,
                                   std::conditional_t<(sizeof(FromFieldType) > sizeof(ToFieldType)),
                                                      FromFieldType, ToFieldType>>;
        using MaxNativeType = typename MaxFieldType::NativeType;

        UInt32 to_max_digits = NumberTraits::max_ascii_len<typename ToFieldType::NativeType>();
        bool multiply_may_overflow = false;
        if (to_scale > from_scale) {
            multiply_may_overflow = (from_precision + to_scale - from_scale) >= to_max_digits;
        }

        typename ToFieldType::NativeType max_result =
                DataTypeDecimal<ToFieldType::PType>::get_max_digits_number(to_precision);
        typename ToFieldType::NativeType min_result = -max_result;

        MaxNativeType multiplier {};
        if (from_scale < to_scale) {
            multiplier = DataTypeDecimal<MaxFieldType::PType>::get_scale_multiplier(to_scale -
                                                                                    from_scale);
        } else if (from_scale > to_scale) {
            multiplier = DataTypeDecimal<MaxFieldType::PType>::get_scale_multiplier(from_scale -
                                                                                    to_scale);
        }
        RETURN_IF_ERROR(std::visit(
                [&](auto multiply_may_overflow, auto narrow_integral) {
                    for (size_t i = 0; i < size; i++) {
                        if (!CastToDecimal::_from_decimal<FromFieldType, ToFieldType,
                                                          multiply_may_overflow, narrow_integral>(
                                    vec_from_data[i], from_precision, from_scale, vec_to_data[i],
                                    to_precision, to_scale, min_result, max_result, multiplier,
                                    params)) {
                            if (set_nullable) {
                                null_map_data[i] = 1;
                            } else {
                                return params.status;
                            }
                        }
                    }
                    return Status::OK();
                },
                make_bool_variant(multiply_may_overflow), make_bool_variant(narrow_integral)));
        if (narrow_integral) {
            block.get_by_position(result).column =
                    ColumnNullable::create(std::move(col_to), std::move(col_null_map_to));
        } else {
            block.get_by_position(result).column = std::move(col_to);
        }
        return Status::OK();
    }
};
} // namespace doris

Coverage Report

Created: 2026-05-18 01:25