如何以小写形式显示枚举？

Question

我有一个枚举：

pub enum BoxColour {
    Red,
    Blue,
}

我不仅要get this value as a string，还希望将值转换为小写。

这行得通：

impl Display for BoxColour {
    fn fmt(&self, fmt: &mut fmt::Formatter) -> fmt::Result {
        fmt.write_str(match self {
            BoxColour::Red => "red",
            BoxColour::Blue => "blue",
        })?;
        Ok(())
    }
}

当颜色列表增加时，需要更新此列表。

如果我使用write! 宏，似乎无法操纵结果，因为write! 返回() 的实例而不是String：

impl Display for BoxColour {
    fn fmt(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
        write!(formatter, "{:?}", self)
    }
}

这表明这是通过副作用工作的，也许我们可以破解内存中值所在的相同位置，但即使可以，这也可能不是一个好主意...

Answer 1

strum crate 提供了一个派生宏，用于为枚举实现Display，并可选择小写变体名称：

use strum_macros::Display;

#[derive(Display)]
// If we don't care about inner capitals, we don't need to set `serialize_all` 
// and can leave parenthesis empty.
#[strum(serialize_all = "snake_case")]
pub enum BoxColour {
    Red,
    Blue,
    LightGreen,  // example of how inner capitals are treated
}

fn main() {
    for c in [BoxColour::Red, BoxColor::Blue, BoxColor::LightGreen] {
        println!("{}", c);
    }
}

您还需要在Cargo.toml 中对strum 进行相应的依赖：

[dependencies]
strum = { version = "0.21", features = ["derive"] }

这应该打印出来：

red
blue
light_green

strum 将生成类似于您提到的 match 和 BoxColour::Red => "red", 案例的代码，但无需手动更新。

Answer 2

作为@Smitop 答案的扩展，通常可以使用自定义格式化程序以小写形式显示任何值，例如：

use std::fmt::{self, Write};

struct LowercaseFormatter<'a, 'b>(pub &'a mut fmt::Formatter<'b>);

impl<'a, 'b> fmt::Write for LowercaseFormatter<'a, 'b> {
    fn write_str(&mut self, s: &str) -> Result<(), fmt::Error> {
        for ch in s.chars() {
            self.0.write_fmt(format_args!("{}", ch.to_lowercase()))?;
        }
        
        Ok(())
    }
}

然后这样使用它：

#[derive(Debug)]
pub enum BoxColour {
    Red,
    Blue,
}

impl fmt::Display for BoxColour {
    fn fmt(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
        write!(LowercaseFormatter(formatter), "{:?}", self)
    }
}

fn main() {
    println!("{}", BoxColour::Red); // "red"
}

Playground

这个版本并没有为每个格式分配一个字符串，但也不是很优化，为每个字符调用write_fmt相对成本较高。

仅ascii 字符的一种替代方法是调用write_char(ch.to_ascii_lowercase())，但一次写入一个字符的字符串也相对昂贵。

适当的解决方案是以某种方式对字符串进行分区，以便能够一次写入所有已经小写的字符，并且只写入大写字符。

Answer 3

这是一种无需在每次添加新颜色时手动更新Display::fmt 的方法，方法是使用派生的Debug 实现并将其小写：

#[derive(Debug)]
pub enum BoxColour {
    Red,
    Blue,
}

impl fmt::Display for BoxColour {
    fn fmt(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
        write!(formatter, "{}", format!("{:?}", self).to_lowercase())
    }
}

(playground)

注意write! 返回一个Result<(), fmt::Error>（这是一个fmt::Result），不是一个原始的()。

但请考虑手动更新列表或使用宏来指定是否会更好。还要考虑如何将带有多个单词的颜色（例如LightBlue）小写：lightblue 是你想要的吗？