Rust 测试实例:单位检查

通过实现一个带虚类型参数的 Add trait 可以实现单位检查。这种 Add trait 的 代码如下:

// 这个 `trait` 会要求 `Self + RHS = Output`。` <RHS = Self>` 表示 RHS 的默认值
// 为 Self 类型,也就是如果没有在实现中另行指定,RHS 就取 Self 类型。
pub trait Add <RHS = Self> {
    type Output;

    fn add(self, rhs: RHS) -> Self::Output;
}

// `Output` 必须是 `T <U>` 类型,所以是 `T <U> + T <U> = T <U>`。
impl <U> Add for T <U> {
    type Output = T <U>;
    ...
}

完整实现:

use std::ops::Add;
use std::marker::PhantomData;

/// 创建空枚举类型来表示单位。
#[derive(Debug, Clone, Copy)]
enum Inch {}
#[derive(Debug, Clone, Copy)]
enum Mm {}

/// `Length` 是一个带有虚类型参数 `Unit` 的类型,
/// 而且对于表示长度的类型(即 `f64`)而言,`Length` 不是泛型的。
///
/// `f64` 已经实现了 `Clone` 和 `Copy` trait.
#[derive(Debug, Clone, Copy)]
struct Length <Unit>(f64, PhantomData <Unit>);

/// `Add` trait 定义了 `+` 运算符的行为。
impl <Unit> Add for Length <Unit> {
     type Output = Length <Unit>;

    // add() 返回一个含有和的新的 `Length` 结构体。
    fn add(self, rhs: Length <Unit>) -> Length <Unit> {
        // `+` 调用了针对 `f64` 类型的 `Add` 实现。
        Length(self.0 + rhs.0, PhantomData)
    }
}

fn main() {
    // 指定 `one_foot` 拥有虚类型参数 `Inch`。
    let one_foot:  Length <Inch> = Length(12.0, PhantomData);
    // `one_meter` 拥有虚类型参数 `Mm`。
    let one_meter: Length <Mm>   = Length(1000.0, PhantomData);

    // `+` 调用了我们对 `Length <Unit>` 实现的 `add()` 方法。
    //
    // 由于 `Length` 了实现了 `Copy`,`add()` 不会消耗 `one_foot`
    // 和 `one_meter`,而是复制它们作为 `self` 和 `rhs`。
    let two_feet = one_foot + one_foot;
    let two_meters = one_meter + one_meter;

    // 加法正常执行。
    println!("one foot + one_foot = {:?} in", two_feet.0);
    println!("one meter + one_meter = {:?} mm", two_meters.0);

    // 无意义的运算当然会失败:
    // 编译期错误:类型不匹配。
    //let one_feter = one_foot + one_meter;
}