Обобщение над добавлением целых типов без знака и со знаком

Question

Мне нужна функция Rust, позволяющая добавлять тип u32 (u64, u128) к типу i32 (i64, i128) при проверке на переполнение.

Моя реализация:

/// Add u32 to i32. In case of an overflow, return None.
fn checked_add_i32_u32(a: i32, b: u32) -> Option<i32> {
    let b_half = (b / 2) as i32;
    let b_rem = (b % 2) as i32;

    Some(a.checked_add(b_half)?.checked_add(b_half)?
        .checked_add(b_rem)?)
}

/// Add u64 to i64. In case of an overflow, return None.
fn checked_add_i64_u64(a: i64, b: u64) -> Option<i64> {
    let b_half = (b / 2) as i64;
    let b_rem = (b % 2) as i64;

    Some(a.checked_add(b_half)?.checked_add(b_half)?
        .checked_add(b_rem)?)
}

У меня есть другая похожая программа, которая делает то же самое для u128 и i128.Я чувствую, что повторяюсь.Мои тесты для этих функций также выглядят очень похоже.

Есть ли способ, которым я мог бы реорганизовать свой код и вместо него иметь только одну функцию?Я не уверен, как обобщать отношения между u32 и i32 (или u64 и i64, u128 и i128).

Answer 1

Вы можете использовать макрос:

trait CustomAdd: Copy {
    type Unsigned;

    fn my_checked_add(self, b: Self::Unsigned) -> Option<Self>;
}

macro_rules! impl_custom_add {
    ( $i:ty, $u:ty ) => {
        impl CustomAdd for $i {
            type Unsigned = $u;

            fn my_checked_add(self, b: $u) -> Option<$i> {
                let b_half = (b / 2) as $i;
                let b_rem = (b % 2) as $i;

                Some(self.checked_add(b_half)?.checked_add(b_half)?
                    .checked_add(b_rem)?)
            }
        }
    }
}

impl_custom_add!(i32, u32);
impl_custom_add!(i64, u64);
// etc.

#[test]
fn tests() {
    assert_eq!(123.my_checked_add(10_u32), Some(133));
}