Rust中未知具体枚举变体时如何访问值并实现枚举比较

核心思路 #

不要依赖PartialOrd默认派生的隐式规则（默认按变体定义顺序判定优先级，后续调整枚举变体位置会导致逻辑错误），手动实现比较逻辑：

1. 先给每个变体分配固定优先级，不同变体直接比较优先级，固定遵循Orange < Banana < Pear规则
2. 同变体情况下，提取内部weight字段比较大小
3. 提取weight不需要提前判断变体，用match多模式匹配即可覆盖所有持有weight字段的变体

完整实现代码 #

use std::cmp::Ordering;

#[derive(Debug, PartialEq)]
enum Fruits {
    Orange { weight: u8 },
    Banana { weight: u8 },
    Pear { weight: u8 },
}

impl Fruits {
    // 获取变体固定优先级，数值越小排序越靠前
    fn priority(&self) -> u8 {
        match self {
            Fruits::Orange { .. } => 0,
            Fruits::Banana { .. } => 1,
            Fruits::Pear { .. } => 2,
        }
    }

    // 不提前判断变体，直接提取任意水果的weight
    fn weight(&self) -> u8 {
        match self {
            // 多模式匹配，覆盖所有变体，统一提取weight
            Fruits::Orange { weight } | Fruits::Banana { weight } | Fruits::Pear { weight } => *weight
        }
    }
}

// 实现PartialOrd支持比较运算符
impl PartialOrd for Fruits {
    fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
        // 第一步：比较变体优先级
        match self.priority().cmp(&other.priority()) {
            Ordering::Equal => {}
            res => return Some(res),
        }
        // 同变体：比较weight
        Some(self.weight().cmp(&other.weight()))
    }
}

// 实现全序比较，支持排序、放入BTreeMap等场景
impl Eq for Fruits {}
impl Ord for Fruits {
    fn cmp(&self, other: &Self) -> Ordering {
        self.partial_cmp(other).unwrap()
    }
}

使用示例 #

fn main() {
    let o1 = Fruits::Orange { weight: 10 };
    let o2 = Fruits::Orange { weight: 20 };
    let b1 = Fruits::Banana { weight: 1 };
    let p1 = Fruits::Pear { weight: 50 };

    // 同变体比较weight：10 < 20
    assert!(o1 < o2);
    // 不同变体比较优先级：Orange < Banana，和weight无关
    assert!(o2 < b1);
    // 不同变体比较优先级：Banana < Pear，和weight无关
    assert!(b1 < p1);
}

关于weight提取的问题 #

之前用if let只能匹配单个变体，所以无法在未知变体的情况下提取值，用match的多模式匹配语法，把所有包含weight字段的变体列在同一个匹配分支中，就可以一次性提取weight，不需要提前判断实例所属变体。

如果你不需要全局实现比较trait，只想单独写compare函数，直接在函数内调用辅助方法即可：

fn compare(fruit_a: Fruits, fruit_b: Fruits) -> bool {
    // 返回fruit_a是否大于fruit_b
    let a_p = fruit_a.priority();
    let b_p = fruit_b.priority();
    if a_p != b_p {
        return a_p > b_p;
    }
    fruit_a.weight() > fruit_b.weight()
}

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