如何将枚举作为字符串？

Question

我有一个包含许多值的枚举，我想将其中一个值的名称写入流：

enum Foo {
    Bar = 0x00,
    Baz = 0x01,
    Qux = 0x02,
    // ...
    Quux = 0xFF,
}

我可以推导出Debug 并做

writer.write(format!("I am {:?}", Foo::Quux).as_bytes())

这将输出例如I am Quux。没关系，除了

• 我想为面向用户的输出执行此操作，因此 Debug 不合适
• 将枚举作为字符串获取（而不是直接写入流）会非常有帮助，因为这样我就可以将它的长度合并到我想做的一些不稳定的格式计算中。

Answer 1

可能最简单的方法是通过调用Debug 来实现Display：

impl fmt::Display for Foo {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        write!(f, "{:?}", self)
        // or, alternatively:
        // fmt::Debug::fmt(self, f)
    }
}

然后您可以使用to_string() 来获得String 表示：

let s: String = Foo::Quux.to_string();

如果您有许多要打印的枚举，您可以编写一个简单的宏来为每个枚举生成Display 的上述实现。

不幸的是，在 Rust 中反射式编程有些困难。例如，没有标准的方法来获取类 C 枚举的所有变体的列表。几乎总是你必须使用自定义编写的宏（或在 crates.io 上找到一些东西）来抽象样板。如果有人写一个 RFC 并且它会被接受，这可能会在未来改变。

Answer 2

由于枚举变量的名称是固定的，你不需要分配String，&'static str 就足够了。宏可以删除样板：

macro_rules! enum_str {
    (enum $name:ident {
        $($variant:ident = $val:expr),*,
    }) => {
        enum $name {
            $($variant = $val),*
        }

        impl $name {
            fn name(&self) -> &'static str {
                match self {
                    $($name::$variant => stringify!($variant)),*
                }
            }
        }
    };
}

enum_str! {
    enum Foo {
        Bar = 0x00,
        Baz = 0x01,
        Qux = 0x02,
        //...
        Quux = 0xFF,
    }
}

fn main() {
    assert_eq!(Foo::Baz.name(), "Baz");
}

更好的是，您可以使用 strum_macros 之类的 crate 导出这些。

在 strum 0.10 中，您可以使用 AsStaticRef / AsStaticStr 执行完全相同的代码：

extern crate strum; // 0.10.0
#[macro_use]
extern crate strum_macros; // 0.10.0

use strum::AsStaticRef;

#[derive(AsStaticStr)]
enum Foo {
    Bar = 0x00,
    Baz = 0x01,
    Qux = 0x02,
    //...
    Quux = 0xFF,
}

fn main() {
    assert_eq!(Foo::Baz.as_static(), "Baz");
}

在 strum 0.9 中，字符串切片的生命周期是 not 'static in this case：

#[macro_use]
extern crate strum_macros; // 0.9.0

#[derive(AsRefStr)]
enum Foo {
    Bar = 0x00,
    Baz = 0x01,
    Qux = 0x02,
    //...
    Quux = 0xFF,
}

fn main() {
    assert_eq!(Foo::Baz.as_ref(), "Baz");
}