面试题：Rust中自定义derive宏基础实现

关键步骤：
- 定义一个过程宏（Procedural Macro），在Rust中，用于生成代码的宏分为声明式宏（macro_rules!）和过程宏。这里需要使用过程宏中的派生宏（Derive Macro）。
- 解析输入的结构体定义，提取出结构体的字段信息。
- 根据提取的字段信息，生成Debug trait实现的代码。

相关语法：

定义过程宏：

use proc_macro::TokenStream;
use quote::quote;
use syn::{parse_macro_input, DeriveInput};

#[proc_macro_derive(SimpleDebug)]
pub fn simple_debug_derive(input: TokenStream) -> TokenStream {
    let ast = parse_macro_input!(input as DeriveInput);
    let struct_name = &ast.ident;
    let fields = match &ast.data {
        syn::Data::Struct(s) => &s.fields,
        _ => panic!("Only structs are supported"),
    };

    let field_debugs = fields.iter().map(|field| {
        let field_name = &field.ident;
        quote! {
            write!(f, "{}: {:?}, ", stringify!(#field_name), self.#field_name)?;
        }
    });

    let gen = quote! {
        impl std::fmt::Debug for #struct_name {
            fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
                f.write_str(concat!(stringify!(#struct_name), " {"))?;
                #(#field_debugs)*
                f.write_char('}')
            }
        }
    };

    gen.into()
}

解释：
- 首先引入必要的库，proc_macro 是Rust标准库中用于过程宏的模块，quote 用于生成Rust代码，syn 用于解析Rust语法树。
- #[proc_macro_derive(SimpleDebug)] 声明这是一个派生宏，名为 SimpleDebug。
- parse_macro_input!(input as DeriveInput) 将输入的 TokenStream 解析为 DeriveInput 结构体，这个结构体包含了被派生宏修饰的结构体的信息。
- 提取结构体名称 struct_name 和字段信息 fields。
- 使用 map 遍历每个字段，生成每个字段在 fmt 方法中的调试输出代码。
- 最后使用 quote! 宏生成完整的 Debug trait实现代码，并将其转换为 TokenStream 返回。

使用示例：
```
#[derive(SimpleDebug)]
struct Point {
    x: i32,
    y: i32,
}

fn main() {
    let p = Point { x: 10, y: 20 };
    println!("{:?}", p);
}
```
上述代码定义了一个 Point 结构体，并使用自定义的 SimpleDebug 派生宏，在 main 函数中打印 Point 实例时会输出类似 Point { x: 10, y: 20 } 的格式。

星途面试题库

面试题：Rust中自定义derive宏基础实现

知识考点

面试题答案