rust - 在 macro_rules 中存储状态

Question

我想创建一个对给定类型列表进行操作的宏，但我需要能够存储正在处理的其他类型。

我想做的事情的一个简单例子:

struct Foo;

struct Bar {
   foo: Foo,
   data: u32,
}

baz!(Foo, Bar);

// outputs
struct OptFoo;

struct OptBar {
   foo: OptFoo,
   data: u32
}

问题是 macro_rules 似乎不允许我存储临时状态(即 HashSet，我会在其中标记哪些类型是宏的一部分调用)。我想到的唯一解决方法是将我想要的内容编写为 proc_macro_derive 并为我需要的每种类型手动添加自定义属性，但这显然远非完美......

编辑:

问题类似于this one ，但在这里我试图在单个宏调用中临时保存一个状态(基本上是在存储有关参数的同时对参数进行两次传递)。然而，这似乎也是不可能的。

Answer 1

正如@trentcl 所指出的，我想要实现的目标确实可以通过 proc 宏实现(我认为 proc 宏仅限于 Derive 和属性...)

#[proc_macro]
pub fn generate(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    let input = proc_macro2::TokenStream::from(input);

    println!("{:?}", input);

    proc_macro::TokenStream::from(input)
}

generate!(struct Foo;);

// outputs its argument without changing anything, i.e:
// struct Foo ;

前面的示例演示了一个简单的宏，它打印到 sdout 已解析的输入:TokenStream [Ident { ident: "struct", span: #0 bytes(330..336) }, Ident { ident: "Foo ", span: #0 bytes(337..340) }, Punct { ch: ';', spacing: Alone, span: #0 bytes(340..341) }]

请注意，它会解析标记但不会创建 AST；为此，我们必须使用 syn。

This repo有许多可以使用 proc 宏完成的示例，非常有帮助!