如何创建可扩展的Rust宏以限制仅接受特定整数字面量？

如何创建一个可扩展的Rust宏，使其仅接受特定的整数字面量？

理想情况下，该解决方案需满足以下要求：

• 可扩展至数百个允许的数字
• 当数字字面量无效时，在宏调用处触发编译错误

多分支写法 #

可以创建一个包含多个分支的宏，每个分支对应一个特定数字，但如果要处理数百个数字，且需要对允许的数字执行统一计算，这种写法会导致大量代码重复，维护成本极高。

macro_rules! my_macro {
    (1) => {
        1
    };
    (2) => {
        2
    };
    (3) => {
        3
    };
}

使用compile_error! #

直接在宏内部使用compile_error!会将错误定位到宏定义的位置，而非调用宏的代码行，错误提示非常不直观，开发者很难快速找到问题所在。

macro_rules! my_macro {
    ($num: literal) => {
        // 若数字无效
        compile_error!("number is invalid")
        ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ // 错误会指向这里

        // 合法数字逻辑
        $num
    };
}

// 假设34是无效数字
my_macro!(34)

我想要模拟TypeScript类型系统中的行为：创建一个类似联合类型的限制，让函数仅接受特定的数字字面量。

type AllowedNumbers = 1 | 2 | 3

const myFunc = (num: AllowedNumbers) => {
    // ...
}

要满足可扩展且错误提示友好的需求，最佳方式是通过宏生成宏的方式自动生成所有允许数字的匹配分支，既避免重复代码，又能在调用处精准报错。

实现方式 #

1. 定义辅助宏生成目标宏分支
   创建一个辅助宏，根据你提供的允许数字列表，自动生成目标宏的所有合法分支，同时为非法数字添加精准的错误提示：

// 辅助宏：根据允许数字列表生成目标宏的匹配规则
macro_rules! generate_allowed_macro {
    ($target_macro:ident, [$($allowed:literal),*]) => {
        macro_rules! $target_macro {
            // 为每个允许的数字生成统一处理逻辑
            ($allowed) => {
                // 这里可以编写对合法数字的通用计算或逻辑
                $allowed + 10 // 示例：给输入数字加10
            };
            // 匹配所有非法字面量，在调用位置触发错误
            ($invalid:literal) => {
                compile_error!(concat!(
                    "无效数字: ", stringify!($invalid), 
                    "，仅允许以下数字: ", stringify!($($allowed),*)
                ));
            };
        }
    };
}

// 生成目标宏，传入允许的数字列表（可扩展至数百个）
generate_allowed_macro!(my_macro, [1, 2, 3, 5, 8, 13]);

2. 使用目标宏
   调用宏时，合法数字会执行预设逻辑，非法数字则会在调用行直接报错，错误信息清晰明了：

fn main() {
    // 合法调用，结果为11
    let valid_result = my_macro!(1);
    println!("{}", valid_result);

    // 非法调用，编译错误会指向这一行，提示无效数字34及允许的数字列表
    // let invalid_result = my_macro!(34);
}

方案优势 #

• 高可扩展性：只需修改generate_allowed_macro调用中的数字列表，即可快速添加或删除允许的数字，无需手动编写大量分支。
• 精准错误提示：非法数字的错误会直接定位到宏调用的代码行，同时显示允许的数字列表，便于开发者快速修正。
• 统一逻辑处理：所有合法数字共享同一套处理逻辑，避免代码重复，维护更轻松。

进阶：复杂规则验证（可选） #

如果需要基于范围、奇偶性等复杂规则验证数字，可以结合const fn在编译时验证，再配合宏确保错误位置准确：

// 编译时验证函数：允许1-100之间的奇数
const fn is_valid(num: u32) -> bool {
    num % 2 == 1 && num >= 1 && num <= 100
}

macro_rules! my_macro {
    ($num:literal) => {
        {
            const VALID: bool = is_valid($num);
            // 编译时断言，错误会指向宏调用处
            assert!(VALID, concat!("数字", stringify!($num), "不符合要求：仅允许1-100的奇数"));
            $num
        }
    };
}

这种方式适合规则灵活的场景，但错误提示的友好度略低于固定数字列表的宏生成方案，可根据需求选择。

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