第3章过程宏入门

过程宏入门：过程宏的分类与声明宏的区别

好，咱们今天聊聊过程宏。说实话，我刚接触 Rust 宏的时候，也被这玩意儿绕得够呛。声明宏还没玩明白呢，又冒出来个过程宏。但后来我在一个序列化库的项目里真正用上了过程宏，才体会到它的威力。

过程宏和声明宏最大的区别是什么？我打个比方你就懂了。声明宏就像是一个文本替换器，你给它什么模式，它就怎么替换。而过程宏呢？它是在编译期运行的 Rust 代码，可以操作语法树。说白了，声明宏是「文本级」的，过程宏是「语法级」的。

过程宏的三种类型

过程宏分三类：派生宏、属性宏、函数宏。咱们一个一个来看。

1. 派生宏（Derive Macros）

这个你应该不陌生。最常见的 #[derive(Debug, Clone)] 就是派生宏。它给结构体或枚举自动实现 trait。

我在项目中遇到过这样一个场景：有个业务模型有几十个字段，手动实现 Default trait 简直要命。后来我写了个自定义派生宏，一行 #[derive(MyDefault)] 就搞定了。

// 定义一个派生宏
use proc_macro::TokenStream;
use quote::quote;
use syn;

#[proc_macro_derive(MyDefault)]
pub fn my_default_derive(input: TokenStream) -> TokenStream {
    let ast = syn::parse_macro_input!(input as syn::DeriveInput);
    let name = &ast.ident;
    
    let gen = quote! {
        impl Default for #name {
            fn default() -> Self {
                // 这里可以自定义默认值逻辑
                Self {
                    // 字段默认值
                }
            }
        }
    };
    gen.into()
}

小技巧：写派生宏时，我习惯先用 syn 解析输入，再用 quote 生成代码。这两个库几乎是过程宏开发的标配。

2. 属性宏（Attribute Macros）

属性宏比派生宏更灵活。它可以加在函数、结构体、模块上，做任意代码转换。你想想看，派生宏只能给结构体或枚举加 trait 实现，但属性宏能干的事就多了。

我记得有个项目需要给某些函数自动添加日志。手动在每个函数开头写 println! 太蠢了。用属性宏就优雅得多：

#[proc_macro_attribute]
pub fn log_call(attr: TokenStream, item: TokenStream) -> TokenStream {
    // attr 是属性参数，item 是被修饰的项
    let input_fn = syn::parse_macro_input!(item as syn::ItemFn);
    let fn_name = &input_fn.sig.ident;
    
    let gen = quote! {
        fn #fn_name() {
            println!("调用函数: {}", stringify!(#fn_name));
            #input_fn
        }
    };
    gen.into()
}

// 使用
#[log_call]
fn my_function() {
    println!("执行中...");
}

注意：属性宏的参数 attr 可能为空。我曾经踩过这个坑——没处理空参数的情况，结果编译直接崩了。记得做空值检查。

3. 函数宏（Function-like Macros）

函数宏长得像函数调用，但它在编译期执行。比如 println! 其实就是一个函数宏。不过我们这里说的是过程宏版本的函数宏，它比声明宏版本的更强大。

#[proc_macro]
pub fn make_answer(input: TokenStream) -> TokenStream {
    // input 是宏调用时传入的参数
    let gen = quote! {
        fn answer() -> u32 {
            42
        }
    };
    gen.into()
}

// 使用
make_answer!();
println!("{}", answer()); // 输出 42

函数宏的应用场景其实挺多的。我做过一个 SQL 查询构建器，就是用函数宏把类似 sql!("SELECT * FROM users WHERE id = ?") 的调用在编译期解析成类型安全的代码。

与声明宏的核心区别

咱们用表格对比一下，这样更直观：

对比维度	声明宏	过程宏
处理层级	Token 流匹配替换	语法树操作
表达能力	模式匹配，有限	任意 Rust 代码，无限
定义位置	同一 crate 或 `macro_use`	独立的 proc-macro crate
编译速度	快	较慢（需要编译宏代码）
调试难度	低	高（需要 `eprintln!` 或 `cargo expand`）
典型用途	`vec!`, `println!`	`#[derive(Serialize)]`, `#[tokio::main]`

为什么会这样？说白了，声明宏是在编译器展开宏之前就做文本匹配。而过程宏是在编译器拿到语法树之后，用 Rust 代码去操作这棵树。你想想看，一个是在「字符串」层面工作，一个是在「数据结构」层面工作，能力自然不一样。

过程宏的工作流程

我画了个图，帮你理解过程宏在编译期是怎么运作的：

实际项目中的选择建议

嗯，这里要注意。不是所有场景都需要上过程宏。我个人的经验是：

简单的代码生成：用声明宏就够了。比如 vec! 这种，声明宏简洁高效。
需要操作类型信息：用派生宏。比如自动实现 Serialize、Deserialize。
需要修改函数或模块：用属性宏。比如 #[tokio::main] 把普通函数变成异步运行时入口。
需要自定义语法：用函数宏。比如 sql!、html! 这种 DSL。

核心要点：过程宏的本质是「在编译期运行 Rust 代码来生成 Rust 代码」。它让 Rust 的元编程能力达到了一个新高度。但能力越大，责任越大——调试过程宏比普通代码难得多，我建议你多配合 cargo expand 查看展开结果。

最后说一句，过程宏的学习曲线确实陡。但一旦你掌握了，很多以前觉得「这得靠代码生成器才能搞定」的问题，用过程宏就能优雅解决。我在生产项目里用过程宏重构过一个配置解析模块，代码量减少了 60%，而且类型安全更强了。

好了，过程宏的分类和区别就聊到这儿。记住：声明宏是文本游戏，过程宏是语法树游戏。选对工具，事半功倍。

📚 课程目录

过程宏入门：过程宏的分类与声明宏的区别

过程宏的三种类型

1. 派生宏（Derive Macros）

2. 属性宏（Attribute Macros）

3. 函数宏（Function-like Macros）

与声明宏的核心区别

过程宏的工作流程

实际项目中的选择建议