使用宏，如何获得结构字段的唯一名称？

Question

假设我这样调用了一些宏：

my_macro!(Blah, (a, b, c));

它输出如下内容：

struct Blah {
    a: i32,
    b: i32,
    c: i32
}
impl Blah {
    fn foo() -> i32 {
        a + b + c
    }
}

（人工示例）

这些字段将是结构私有的，但我需要允许重新定义。所以，输入

my_macro!(Blah, (a, b, c, a));

将生成如下内容：

struct Blah {
    a1: i32,
    b: i32,
    c: i32,
    a2: i32
}
impl Blah {
    fn foo() -> i32 {
        a1 + b + c + a2
    }
}

命名方案不需要遵循任何逻辑模式。

这可能吗？

Answer 1

如果不使用编译器插件，不，我认为这是不可能的。两个原因：

您不能构造标识符。有concat_idents!，但由于宏展开的方式，在这种情况下没用。
不能进行非文字比较。也就是说，宏无法计算出它之前已经看过 a 一次。

你能得到的最接近的是直接用固定的标识符列表替换所有提供的标识符，但这可能不是你想要的；在这种情况下，只指定您想要 4 个字段并生成一个固定大小的数组 [i32; 4].

会更容易

Answer 2

我的 mashup crate 为您提供了一种将 my_macro!(Blah, (a, b, c, a)) 扩展到字段 x_a、xx_b、xxx_c、xxxx_d 的方法，如果该命名约定对您有用。我们为每个字段添加一个额外的 x ，后跟一个下划线，然后是原始字段名称，这样任何字段都不会以名称冲突告终。此方法适用于 >= 1.15.0.

的任何 Rust 版本

#[macro_use]
extern crate mashup;

macro_rules! my_macro {
    ($name:ident, ($($field:ident),*)) => {
        my_macro_helper!($name (x) () $($field)*);
    };
}

macro_rules! my_macro_helper {
    // In the recursive case: append another `x` into our prefix.
    ($name:ident ($($prefix:tt)*) ($($past:tt)*) $next:ident $($rest:ident)*) => {
        my_macro_helper!($name ($($prefix)* x) ($($past)* [$($prefix)* _ $next]) $($rest)*);
    };

    // When there are no fields remaining.
    ($name:ident ($($prefix:tt)*) ($([$($field:tt)*])*)) => {
        // Use mashup to define a substitution macro `m!` that replaces every
        // occurrence of the tokens `"concat" $($field)*` in its input with the
        // resulting concatenated identifier.
        mashup! {
            $(
                m["concat" $($field)*] = $($field)*;
            )*
        }

        // Invoke the substitution macro to build a struct and foo method.
        // This expands to:
        //
        //     pub struct Blah {
        //         x_a: i32,
        //         xx_b: i32,
        //         xxx_c: i32,
        //         xxxx_a: i32,
        //     }
        //
        //     impl Blah {
        //         pub fn foo(&self) -> i32 {
        //             0 + self.x_a + self.xx_b + self.xxx_c + self.xxxx_a
        //         }
        //     }
        m! {
            pub struct $name {
                $(
                    "concat" $($field)*: i32,
                )*
            }

            impl $name {
                pub fn foo(&self) -> i32 {
                    0 $(
                        + self."concat" $($field)*
                    )*
                }
            }
        }
    };
}

my_macro!(Blah, (a, b, c, a));

fn main() {}

使用宏，如何获得结构字段的唯一名称？

Using macros, how to get unique names for struct fields?

macros

rust