如何编写基于str_to_num宏生成match分支的Rust宏？

问题背景 #

已有编译期宏str_to_num，仅能处理字符串字面量，需要扩展为支持运行时变量的版本，通过宏生成match表达式实现。

实现方案 #

方案1：集中管理映射，同时生成编译期宏与运行时match

这种方式将所有键值映射集中定义，避免重复代码，同时生成原有的编译期宏和运行时匹配逻辑：

// 定义公共映射数据源
macro_rules! str_num_mappings {
    ($macro:ident) => {
        $macro! {
            ("one") => { 1 },
            ("two") => { 2 },
            ("three") => { 3 },
        }
    };
}

// 基于公共数据源生成原编译期宏
macro_rules! str_to_num {
    ($($lit:literal)=>$num:expr,) => {
        macro_rules! str_to_num {
            ($lit) => { $num };
            $(($lit) => { $num };)*
        }
    };
}

// 基于公共数据源生成运行时match闭包
macro_rules! str_to_num_match {
    ($($lit:literal)=>$num:expr,) => {
        |s: &str| match s {
            $lit => Some($num),
            $( $lit => Some($num), )*
            _ => None,
        }
    };
}

// 初始化两个宏
str_num_mappings!(str_to_num);
str_num_mappings!(str_to_num_match);

fn main() {
    // 编译期使用原宏
    let compile_val = str_to_num!("one");
    println!("编译期值：{}", compile_val);

    // 运行时使用匹配逻辑
    let runtime_str = "two";
    let runtime_val = str_to_num_match!(runtime_str);
    println!("运行时值：{:?}", runtime_val); // 输出 Some(2)

    let invalid_str = "four";
    let invalid_val = str_to_num_match!(invalid_str);
    println!("无效输入：{:?}", invalid_val); // 输出 None
}

方案2：使用递归宏生成match分支

若希望直接复用原宏的规则，可通过递归宏逐个构建match的分支：

// 原编译期宏
macro_rules! str_to_num {
    ("one") => { 1 };
    ("two") => { 2 };
    ("three") => { 3 };
}

// 递归宏：构建match分支
macro_rules! build_match_branches {
    // 递归终止：生成默认分支
    () => {
        _ => None
    };
    // 处理单个分支，继续递归剩余规则
    (($lit:literal) => $num:expr, $($rest:tt)*) => {
        $lit => Some($num),
        build_match_branches!($($rest)*)
    };
}

// 主宏：生成完整match表达式
macro_rules! str_to_num_match {
    ($input:expr) => {
        match $input {
            build_match_branches!(
                ("one") => str_to_num!("one"),
                ("two") => str_to_num!("two"),
                ("three") => str_to_num!("three"),
            )
        }
    };
}

fn main() {
    let args = vec!["main", "three"];
    let value = str_to_num_match!(&args[1]);
    println!("运行时值：{:?}", value); // 输出 Some(3)
}

核心说明 #

• 原str_to_num宏仅在编译期展开，无法处理运行时变量，因为宏展开阶段无法获取运行时的值。
• 通过生成match表达式，将运行时字符串与预定义字面量做匹配，即可实现运行时转换。
• 递归宏完全可行，它可以遍历所有映射规则，逐步构建完整的match分支列表，适合映射较多或需要动态扩展的场景。
• 方案1的优势是集中管理映射，减少重复定义；方案2直接复用原宏规则，无需额外维护数据源。

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