如何修复Rust中遍历枚举生成const的宏编译错误？

我有一个枚举，想在match语句里把它的变体当作const来用，于是写了这样的宏：

#[macro_export]
macro_rules! enum_consts {
    ($name:ident) => {
        // 尝试用iter方法获取迭代器
        for variant in $name::iter() {
            // 用paste从字面量生成标识符
            paste::paste! {
                // 为每个变体声明常量
                const [< $name _ $variant >]: u8 = variant as u8;
            }
        }
    };
}

enum Color {
    Red,
    Green
}

enum_consts!{ Color }

但编译时报错：

error: macro expansion ignores token `for` and any following
 --> abc.rs:1951:9
     |
1951 |         for variant in $name::iter() {
     |         ^^^

解决方案 #

问题根源 #

Rust的声明宏是静态代码展开工具，它在编译期处理代码模板，完全无法执行运行时的逻辑（比如for循环、调用iter()方法）。你写的for循环是运行时代码，宏展开阶段根本无法执行，所以编译器报错。

要实现需求，必须在编译期枚举所有枚举变体，用宏的模式匹配来生成常量，而不是依赖运行时迭代。

方案1：宏同时定义枚举和常量 #

把枚举定义和常量生成放在同一个宏里，让宏直接解析变体列表，静态展开成代码：

use paste::paste;

#[macro_export]
macro_rules! enum_with_consts {
    ($enum_name:ident { $($variant:ident),* $(,)? }) => {
        // 定义枚举
        enum $enum_name {
            $($variant),*
        }

        // 为每个变体生成对应的常量
        $(
            paste! {
                const [<$enum_name _ $variant>]: u8 = $enum_name::$variant as u8;
            }
        )*
    };
}

// 使用示例
enum_with_consts! {
    Color {
        Red,
        Green,
    }
}

fn main() {
    match Color::Red as u8 {
        Color_Red => println!("匹配到红色"),
        Color_Green => println!("匹配到绿色"),
        _ => unreachable!(),
    }
}

方案2：为已有枚举生成常量 #

如果已经定义好枚举，不想改动定义方式，可以写一个宏接收枚举名称和变体列表：

use paste::paste;

// 已有的枚举定义
enum Color {
    Red,
    Green,
}

#[macro_export]
macro_rules! enum_consts {
    ($enum_name:ident, $($variant:ident),* $(,)?) => {
        $(
            paste! {
                const [<$enum_name _ $variant>]: u8 = $enum_name::$variant as u8;
            }
        )*
    };
}

// 生成常量
enum_consts!(Color, Red, Green);

fn main() {
    match Color::Green as u8 {
        Color_Green => println!("匹配到绿色"),
        Color_Red => println!("匹配到红色"),
        _ => unreachable!(),
    }
}

进阶：自动遍历枚举变体（可选） #

如果不想手动写变体列表，可以用strum crate的EnumIter derive宏来自动生成迭代器，再配合过程宏生成常量。不过声明宏做不到自动 introspect 已定义枚举的变体，必须用过程宏或者在宏里定义枚举。

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