enumをキーとしたハッシュマップを作成したい

Question

enumの値を無視して型だけで判別するハッシュマップを作成したいのですが、構造体でラッピングしても値まで完全一致でなければヒットしてくれません。どのようにすればこれが実現できますか。

use std::collections::HashMap;

#[derive(Debug, Hash, Eq, PartialEq)]
enum Animal {
    Dog,
    Cat(i32),
    Bird {
        name: String,
        age: i32,
    },
}

fn main() {
    // やりたいこと
    let foo: HashMap<Animal, i32> = vec![
        (Animal::Dog, 3), (Animal::Cat, 2), (Animal::Bird, 1)]
        .into_iter().collect();
    let bar = Animal::Cat(10);
    assert_eq!(2, foo[&bar])
}

Answer 1

HashMapはキーに完全に一致したものを取り出すためのデータ構造ですので、ご質問のような部分一致したものを取り出すようなことはできません。また、Animal型についても、コードの「やりたいこと」にあるようなフィールド値を省略してCatのようなバリアントだけの値を作ることもできません。

たとえば、以下のように、フィールドを持たないenumと、フィールドを持つenumの両方を定義するのはどうでしょうか？

use std::collections::HashMap;

/// 動物の種類（バリアントだけでフィールドを持たない）
#[derive(Debug, Hash, Eq, PartialEq)]
enum AnimalKind {
    Dog,
    Cat,
    Bird,
}

/// 動物（いくつかのバリアントはフィールドを持つ）
#[derive(Debug, Eq, PartialEq)]
enum Animal {
    Dog,
    Cat(i32),
    Bird { name: String, age: i32 },
}

fn main() {
    let foo: HashMap<AnimalKind, (Animal, i32)> = vec![
        (AnimalKind::Dog, (Animal::Dog, 3)),
        (AnimalKind::Cat, (Animal::Cat(10), 2)),
        (AnimalKind::Bird,
            (Animal::Bird {name: "Tweety".to_string(), age: 3}, 1),
        ),
    ]
    .into_iter()
    .collect();
    let bar = AnimalKind::Cat;
    assert_eq!(foo[&bar], (Animal::Cat(10), 2));
}

Answer 2

キーをすべてAnimalの値から取るのであればstd::mem::discriminantが使えると思います。

他の手段としてはenum AnimalKind { Dog, Cat, Bird }とAnimal::kind: fn(&self) -> AnimalKindを作るという方法でしょうか。AnimalKindについてはenum-mapというクレートを使えば扱いやすくなり、パフォーマンスも向上するかと思います。(直接Animalには使えないので注意)

-use std::collections::HashMap;
+use std::{
+    collections::HashMap,
+    mem::{self, Discriminant},
+};

 #[derive(Debug, Hash, Eq, PartialEq)]
 enum Animal {
     Dog,
     Cat(i32),
     Bird { name: String, age: i32 },
 }

 fn main() {
     // やりたいこと
-    let foo: HashMap<Animal, i32> = vec![(Animal::Dog, 3), (Animal::Cat, 2), (Animal::Bird, 1)]
-        .into_iter()
-        .collect();
+    let foo: HashMap<Discriminant<Animal>, i32> = vec![
+        (mem::discriminant(&Animal::Dog), 3),
+        (mem::discriminant(&Animal::Cat(0)), 2),
+        (
+            mem::discriminant(&Animal::Bird {
+                name: "".to_owned(),
+                age: 0,
+            }),
+            1,
+        ),
+    ]
+    .into_iter()
+    .collect();
     let bar = Animal::Cat(10);
-    assert_eq!(2, foo[&bar])
+    assert_eq!(2, foo[&mem::discriminant(&bar)])
 }

Comments

[asobinin]

私の求めていたものがstd::memの中にあったのですね・・・
盲点でした