Rust 提取 struct 生成宏中字段出现的索引

发布于01月18日

有没有办法在这样的宏中提取 struct 字段的"索引"(出现的顺序):

macro_rules! example {
    (struct $name:ident {
        $($field_name:ident: $field_type:ty,)*
    }) => {
        struct $name {
            $($field_name: $field_type,)*
        }
        impl $name {
            fn print_members(&self) {
                $(println!("{} {} {}",
                    stringify!($field_name),
                    stringify!($field_type),
                    stringify!(<<<FIELD INDEX SOMEHOW>>>),
                );)*
            }
        }
    }
}

以至于

example! {
    struct SomeStruct {
        a: String,
        b: String,
        c: usize,
    }
}

将打印:

a String 0
b String 1
c usize 2

我非常肯定proc宏可以做到这一点(并不是我知道如何编写)，但是使用声明性宏能做到这一点吗？

推荐答案

在晚上，这很容易做到，使用#![feature(macro_metavar_expr)]:

#![feature(macro_metavar_expr)]

macro_rules! example {
    (struct $name:ident {
        $($field_name:ident: $field_type:ty,)*
    }) => {
        struct $name {
            $($field_name: $field_type,)*
        }
        impl $name {
            fn print_members(&self) {
                $(println!("{} {} {}",
                    stringify!($field_name),
                    stringify!($field_type),
                    stringify!(${index()}),
                );)*
            }
        }
    }
}

在马厩上，你可以用TT咀嚼来计算.但是，您不能再使用stringify!()，因为这将创建一个未展开的序列0 + 1 + 1 + 1 + ....由于您只是打印，您可以直接使用数字:

macro_rules! example {
    (struct $name:ident {
        $($field_name:ident: $field_type:ty,)*
    }) => {
        struct $name {
            $($field_name: $field_type,)*
        }
        impl $name {
            fn print_members(&self) {
                example! { @generate_println
                    [ 0 ]
                    $( $field_name : $field_type, )*
                }
            }
        }
    };
    
    { @generate_println
        [ $index:expr ]
    } => {
        // Stop condition.
    };
    { @generate_println
        [ $index:expr ]
        $first_field_name:ident : $first_field_type:ty,
        $( $rest_field_name:ident : $rest_field_type:ty, )*
    } => {
        println!("{} {} {}",
            stringify!($first_field_name),
            stringify!($first_field_type),
            $index,
        );
        example! { @generate_println
            [ $index + 1 ]
            $( $rest_field_name : $rest_field_type, )*
        }
    };
}