时间戳

首页 > 解决方案 > 如何为具有特殊情况的枚举实现“哈希”？

rust - 如何为具有特殊情况的枚举实现“哈希”？

问题描述

我有这个枚举：

enum MyEnum {
    Var1,
    Var2(u32),
    Var3(u32, u32),
}

我想实现以下行为：

let mut set = HashSet::new();
set.insert(MyEnum::Var1);
set.insert(MyEnum::Var1);
set.insert(MyEnum::Var2(1));
set.insert(MyEnum::Var2(2));
set.insert(MyEnum::Var3(1, 1));
set.insert(MyEnum::Var3(1, 1));
set.insert(MyEnum::Var3(2, 1));
set.insert(MyEnum::Var3(2, 2));

println!("set = {:?}", set);
// set = {Var1, Var2(1), Var2(2), Var3(1, 1), Var3(2, 1)}

也就是说，我想将变体的Hash行为Var3 专门更改为仅依赖于它的第一个u32。

重要的是，我不想详尽地列出match每个枚举变体，因为我想保留所有其他枚举变体的默认行为。因此，据我所知，这些 答案不足以满足我的要求。

我想出的解决方案是这样的：

#[derive(Debug, Eq)]
enum MyEnum {
    Var1,
    Var2(u32),
    Var3(u32, u32),
}

impl PartialEq for MyEnum {
    fn eq(&self, other: &Self) -> bool {
        match (self, other) {
            (MyEnum::Var3(a, _), MyEnum::Var3(b, _)) => a == b,
            _ => format!("{:?}", self) == format!("{:?}", other),
        }
    }
}

impl Hash for MyEnum {
    fn hash<H: Hasher>(&self, state: &mut H) {
        match self {
            MyEnum::Var3(a, _) => a.hash(state),
            _ => format!("{:?}", self).hash(state)
        }
    }
}

...我正在寻找反馈/更好的解决方案。

标签： rust

解决方案

衍生板条箱的构建旨在为标准特征提供自动实现，并支持常见的自定义。有了这个板条箱，你可以得到你想要的东西，看起来像这样：

use std::collections::HashSet;
use derivative::Derivative; // 2.2.0

#[derive(Debug, Eq, Derivative)]
#[derivative(PartialEq, Hash)]
enum MyEnum {
    Var1,
    Var2(u32),
    Var3(
        u32, 
        #[derivative(PartialEq="ignore")]
        #[derivative(Hash="ignore")]
        u32
    ),
}

fn main() {
    let mut set = HashSet::new();
    set.insert(MyEnum::Var1);
    set.insert(MyEnum::Var1);
    set.insert(MyEnum::Var2(1));
    set.insert(MyEnum::Var2(2));
    set.insert(MyEnum::Var3(1, 1));
    set.insert(MyEnum::Var3(1, 1));
    set.insert(MyEnum::Var3(2, 1));
    set.insert(MyEnum::Var3(2, 2));
    
    println!("set = {:?}", set);
}

set = {Var1, Var3(1, 1), Var3(2, 1), Var2(2), Var2(1)}

在操场上看到它。

推荐阅读