Rust中如何消除实现同一Trait的枚举变体匹配代码重复？

我用clap写了一个CLI程序，里面包含多个结构体，还有一个嵌套这些结构体的枚举用于定义子命令。目前遇到了代码重复问题：枚举的每个变体都仅包含一个实现了Runner trait的结构体，导致为枚举实现Runner时，match分支里的代码全是重复的runner.run()调用。实际代码已有七个变体，手动编写这些分支过于繁琐，想知道Rust有没有办法利用所有变体的共性消除这类重复。

以下是最小可复现示例（MRE）：

pub trait Runner {
    fn run(&self);
}

pub struct Foo;

impl Runner for Foo {
    fn run(&self) {
        println!("Fooing");
    }
}

pub struct Bar;

impl Runner for Bar {
    fn run(&self) {
        println!("Baring");
    }
}

pub enum Action {
    DoFoo(Foo),
    DoBar(Bar),
}

impl Runner for Action {
    fn run(&self) {
        match self {
            Self::DoFoo(runner) => runner.run(),
            Self::DoBar(runner) => runner.run(),
        }
    }
}

解决方案 #

1. 自定义宏自动生成匹配分支

最直接的方式是编写一个自定义宏，让它帮你自动生成所有枚举变体的match分支，无需手动重复编写代码。

// 定义宏：接收枚举类型和变体列表，自动实现Runner trait
macro_rules! impl_runner_for_enum {
    ($enum_type:ty, $($variant:ident),*) => {
        impl Runner for $enum_type {
            fn run(&self) {
                match self {
                    $(
                        Self::$variant(runner) => runner.run(),
                    )*
                }
            }
        }
    };
}

// 为Action枚举应用宏，传入所有变体
impl_runner_for_enum!(Action, DoFoo, DoBar);

后续新增枚举变体时，只需要在宏调用的变体列表中添加新的变体名称即可，完全不用修改match逻辑。

2. 使用第三方库简化枚举操作

如果你的枚举需要频繁修改或需要更多枚举相关的功能，可以使用strum库来简化代码。它提供了枚举遍历、派生等工具，避免手动编写宏。

首先在Cargo.toml中添加依赖：

strum = { version = "0.25", features = ["derive", "enum_iter"] }

然后为枚举派生EnumIter trait，并实现Runner：

use strum::{EnumIter, IntoEnumIterator};

#[derive(EnumIter)]
pub enum Action {
    DoFoo(Foo),
    DoBar(Bar),
}

impl Runner for Action {
    fn run(&self) {
        match self {
            Action::DoFoo(r) => r.run(),
            Action::DoBar(r) => r.run(),
            // 新增变体时，strum的lint可以提醒你遗漏的分支
        }
    }
}

不过对于单纯消除run方法的重复来说，自定义宏是更轻量的选择，无需引入额外依赖。

内容的提问来源于stack exchange，提问作者user3515670