将所有变体实现相同特征的枚举转换为 Rust 中的框？

Question

我有一个特征Foo，有一些实现，还有一个枚举Foos，每个实现都有一个变体。我希望能够将我的枚举转换为Box<dyn Foo>。

这是我目前的解决方案：

trait Foo {}

struct FooA {}
impl Foo for FooA {}

struct FooB {}
impl Foo for FooB {} 

struct FooC {}
impl Foo for FooC {}

enum Foos {
    A(FooA),
    B(FooB),
    C(FooC),
}

impl Foos {
    fn into_box(self) -> Box<dyn Foo> {
        match self {
            Foos::A(foo) => Box::new(foo),
            Foos::B(foo) => Box::new(foo),
            Foos::C(foo) => Box::new(foo),
        }
    }
}

它有效，但into_enum 中有很多样板。随着变体数量的增加，功能也会增加。有没有更简单的方法来做到这一点？感觉应该是单排！

Answer 1

用enum_dispatch crate，你可以写

#[macro_use]
extern crate enum_dispatch;

#[enum_dispatch]
trait Foo {}

struct FooA {}
impl Foo for FooA {}

struct FooB {}
impl Foo for FooB {}

struct FooC {}
impl Foo for FooC {}

#[enum_dispatch(Foo)]
enum Foos {
    A(FooA),
    B(FooB),
    C(FooC),
}

获取生成的impl Foo for Foos。然后，您可以使用简单的Box::new 将Foos 转换为Box<dyn Foo>。

这种方法有一个潜在的缺点：Box::new(Foos::A(FooA)) 包含Foos，而不是FooA，因此它会产生从dyn Foo 到Foos 的动态调度和@987654333 的开销@从Foos调度到FooA。

另一方面，现在您有了impl Foo for Foos：在您本来可以使用Box<dyn Foo> 的任何地方，您都可以直接使用Foos，这在各个方面都应该更有效。

Answer 2

我最近想要类似的东西。我不能为您提供单行代码，而是自动生成相应的 match 臂以及随后的 enum 变体的宏：

macro_rules! impl_foos{($($enumvariant: ident($foo: ty),)*) => {
    enum Foos {
        $($enumvariant($foo),)*
    }
    impl Foos {
        fn into_enum(self) -> Box<dyn Foo> {
            match self {
                $(Foos::$enumvariant(foo) => Box::new(foo),)*
            }
        }
    }
}}

impl_foos!(
    A(FooA),
    B(FooB),
    C(FooC),
);

这样，只有一个地方来维护所有的可能性，其他的一切都是生成的。甚至crate enum_dispatch 也有帮助。

题外话：真的应该是into_enum(self)->Box<dyn Foo>吗？不应该是as_foo(&self)->&dyn Foo之类的吗？