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

Question

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

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

它的输出是这样的:

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

(人工示例)

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

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

我的 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() {}