如何实现支持所有数值类型的Rust通用翻倍函数？

问题分析 #

你之前的代码依赖From<f32>约束，导致整数类型无法调用——标准库不会为整数实现From<f32>，因为浮点数转整数可能丢失精度（比如2.5转整数的歧义）；换成From<u32>也不行，因为浮点类型和有符号整数（如i32）都不满足该约束：浮点类型没有From<u32>实现（反过来的转换也不安全），而i32无法直接从u32转换（u32的取值范围超出i32）。

解决方案 #

方案1：仅用标准库实现

利用TryFrom<u8> trait，因为常量2属于u8的安全取值范围，所有数值类型（整数、浮点）都能安全转换这个值，因此unwrap()不会触发panic：

use std::ops::Mul;
use std::convert::TryFrom;

fn double<A>(x: A) -> A
where
    A: Mul<Output = A> + TryFrom<u8>,
{
    x * A::try_from(2).unwrap()
}

fn main() {
    println!("{}", double(2i32));    // 输出 4
    println!("{}", double(3.5f64));  // 输出 7.0
    println!("{}", double(10u64));   // 输出 20
}

方案2：用num-traits库更优雅实现

num-traits是Rust生态中常用的数值工具库，提供了统一的数值类型接口，无需手动处理转换：

1. 先在Cargo.toml添加依赖：

[dependencies]
num-traits = "0.2"

2. 实现代码：

use num_traits::One;
use std::ops::Mul;

fn double<A>(x: A) -> A
where
    A: Mul<Output = A> + One,
{
    x * (A::one() + A::one())
}

fn main() {
    println!("{}", double(2i32));    // 输出 4
    println!("{}", double(3.5f64));  // 输出 7.0
    println!("{}", double(10u64));   // 输出 20
}

通过One::one()获取当前类型的1，相加得到2，完全避免了类型转换的问题，代码更简洁安全。

内容的提问来源于stack exchange，提问作者Daniel Fainberg