如何在Rust派生宏中使用AST？含TokenStream与AST互转等问题

问题解答 #

1. 如何将TokenStream转换为AST？ #

使用syn库提供的parse_macro_input!宏完成转换，这是处理宏输入的标准方式。它会把输入的TokenStream解析为指定的AST类型，同时自动处理解析错误，生成符合Rust风格的编译报错。示例中的parse_macro_input!(input as DeriveInput)就是正确的转换写法。

2. 转换后的AST是什么类型？ #

示例中的ast是syn::DeriveInput类型，该类型定义在syn库中，对应被derive宏标记的Rust类型（结构体、枚举、联合体）的完整元信息，包含：

• 类型的标识符（名称）
• 类型上的属性列表
• 泛型参数定义
• 具体的数据结构（比如结构体的字段、枚举的变体等）

3. 如何将AST转换回TokenStream？ #

只要AST类型实现了Into<TokenStream>或syn::ToTokenStream trait（syn库提供的所有AST节点都已实现这些trait），就可以通过TokenStream::from(ast)或者ast.to_token_stream()完成转换。

示例代码验证 #

给出的代码逻辑在语法层面是正确的，但功能上只是原样返回输入的AST，没有实现derive宏的实际功能：

#[proc_macro_derive(Serialize)]
pub fn serialize(input: TokenStream) -> TokenStream {
    // 这行是正确的：将输入TokenStream解析为DeriveInput AST
    let ast = parse_macro_input!(input as DeriveInput);

    // ast 的类型是 syn::DeriveInput

    // 这行是正确的：将AST转换回TokenStream
    TokenStream::from(ast)
}

需要注意：实际开发derive宏时，你需要基于ast的信息生成新的AST节点（比如序列化相关的代码逻辑），再将新AST转换为TokenStream返回，而不是直接返回原AST。

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