如何通过Rust过程宏生成变量类型与初始化器并适配多上下文？

问题原因 #

你尝试在pub static FOO后直接调用宏生成: u8 = 42，但Rust语法解析器会先检查当前位置的语法合法性——在static变量名之后，解析器仅接受:、;或=作为下一个token，宏调用foo!()不符合这个语法预期，因此会直接报错，根本不会执行宏展开。

解决方案 #

要实现同一个宏适配static、let等不同上下文，同时拆分生成类型与初始化器，推荐使用属性宏，它可以接收整个变量声明项，解析出变量名和上下文类型，再生成完整的声明代码。

实现示例

extern crate proc_macro;
use proc_macro::TokenStream;
use quote::quote;
use syn::{parse_macro_input, Item, ItemStatic, ItemLet};

#[proc_macro_attribute]
pub fn define(_attr: TokenStream, item: TokenStream) -> TokenStream {
    let input = parse_macro_input!(item as Item);
    
    match input {
        Item::Static(s) => {
            let ident = s.ident;
            TokenStream::from(quote! {
                pub static #ident: u8 = 42;
            })
        }
        Item::Let(l) => {
            let pat = l.pat;
            TokenStream::from(quote! {
                let #pat: u8 = 42;
            })
        }
        _ => panic!("此宏仅支持static变量或let绑定"),
    }
}

使用方式

// 生成static变量
#[define]
pub static FOO;

// 生成本地let绑定
#[define]
let bar;

替代方案（函数式宏） #

如果坚持使用函数式过程宏，可以将类型和初始化器拆分到两个宏中，在合法的语法位置调用：

extern crate proc_macro;
use proc_macro::TokenStream;
use quote::quote;

#[proc_macro]
pub fn var_type(_: TokenStream) -> TokenStream {
    TokenStream::from(quote! { u8 })
}

#[proc_macro]
pub fn var_init(_: TokenStream) -> TokenStream {
    TokenStream::from(quote! { 42 })
}

调用时：

pub static FOO: var_type!() = var_init!();
let bar: var_type!() = var_init!();

内容的提问来源于stack exchange，提问作者Cactus