所以我定义了以下过程宏:
#[proc_macro_attribute]
pub fn hello(attr: TokenStream, item: TokenStream) -> TokenStream {
println!("attr: {}", attr);
println!("item: {}", item);
item
}
然后我将此 proc 宏应用于非内联模块:
// In file some_mod.rs
#![hello]
fn foo() {}
fn bar() {}
编译器的输出表明传递给 proc 宏 hello
的项目是 mod some_mod;
,模块中没有任何项目。但是我想对模块 some_mod
的内容做一些修改。
我发现内联模块有效:
mod some_mod {
#![hello]
fn foo() { }
fo bar() { }
}
传递给我的 proc 宏的项目是 mod some_mod { fn foo() { } fn bar() { } }
。
但我会在复杂的模块层次结构中使用我的 proc 宏,我不想将所有这些模块放在一个文件中。
有什么办法可以让我的proc宏获取非内联模块的内容吗?
最佳答案
您描述的问题是由于目前在内部属性位置使用过程宏是不稳定的。有一个tracking issue对于这个问题大约 1.5 年了,this comment明确提及您当前面临的问题,将宏应用于内联模块会产生与将其应用于文件模块不同的结果。
据我所知,目前没有针对此问题的现成解决方案。
或者,由于您基本上只是修改文件的标记流,您可以尝试通过 build.rs
将您的程序 Hook 到编译过程中(您可以加载您想要的任何库构建过程),但这将涉及手动筛选所有文件,并且必须从输入 token 流中动态生成新文件。不幸的是,我目前无法想到任何更好的选择,只要您真的需要宏将整个模块作为输入即可。
关于rust - 如何获取传递给内部属性宏的模块内容?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/60133410/