0

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

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

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并为我需要的每种类型手动添加自定义属性,但这显然远非完美......

编辑:

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

4

1 回答 1

0

正如@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它。

这个 repo有许多可以用 proc 宏完成的例子,非常有帮助!

于 2019-07-24T09:15:02.783 回答