类似的匹配臂如何编写宏？

Question

我有一些重复的代码

match *x {
    A(ref a) => "special",
    B(ref a) => "B foo",
    C(ref a) => "C foo",
    D(ref a) => "D foo",
    // ...
}

我想要一个像这样的宏

macro_rules! generic_fmt {
    ($T:ident) => {
        $T(ref a) => {"$T foo"},
    }
}

这样我就可以简化我的匹配

match *x {
    A(ref a) => "special",
    generic_fmt!(B),
    generic_fmt!(C),
    generic_fmt!(D),
    // ...
}

最好的方法是什么？我正在使用 rustc 1.19.0-nightly。

Answer 1

你不能完全做到这一点。宏不能扩展为match臂 ( <Variant> => <Expression>)。

你能得到的最接近的可能是这样的：

enum Foo {
    A(u32),
    B(u32),
    C(u32),
    D(u32),
}

macro_rules! gen1 {
    ($Variant:ident) => {
        Foo::$Variant(ref a)
    }
}

macro_rules! gen2 {
    ($Variant:ident) => {
        concat!(stringify!($Variant), " foo")
    }
}

fn print(x: Foo) {
    println!("{}", match x {
        Foo::A(ref a) => "special",
        gen1!(B) => gen2!(B),
        gen1!(C) => gen2!(C),
        gen1!(D) => gen2!(D),
    });
}

fn main() {
    print(Foo::A(42));
    print(Foo::B(42));
    print(Foo::C(42));
    print(Foo::D(42));
}

游乐场链接。

Answer 2

如果稍微改变你的输出是可以接受的，你可以对所有相同的手臂使用一个包罗万象的：

match x {
    Foo::A(ref a) => println!("special"),
    _ => println!("{:?} Foo", x),
}

游乐场链接

但这会打印类型及其参数。如果您使用 nightly，并且不怕实验，则可以使用std::intrinsics::type_name仅显示类型名称。

或者你可以使用一个宏来完成所有的比赛：

macro_rules! gen_match {
    ($x:ident, $Special:ident, [$( $Foo:ident ),*]) => {
        match $x {
            Foo::$Special(ref a) => println!("special"),
            $(Foo::$Foo(ref a) => println!("{} foo", stringify!($Foo)),)*
        }
    }
}

并称其为：

gen_match!(x, A, [B, C, D]);

游乐场链接

将采用通用格式的变体周围的括号是为了便于阅读，可以将它们从宏定义中删除