有没有办法在 Rust 宏中引用局部变量？

Question

我想使用宏来生成函数体，但这样做需要访问局部范围内的变量：

macro_rules! generate_func {
    ($name:ident) => {
        fn $name(param: i32) {
            generate_body!()
        }
    };
}

macro_rules! generate_body {
    () => {
        println!("{}", &param);
    }
}

generate_func!(foo);

error[E0425]: cannot find value `param` in this scope
  --> src/lib.rs:11:25
   |
11 |         println!("{}", &param);
   |                         ^^^^^ not found in this scope
...
15 | generate_func!(foo);
   | -------------------- in this macro invocation
   |
   = note: this error originates in a macro (in Nightly builds, run with -Z macro-backtrace for more info)

Playground link

令人沮丧的是这不起作用，因为我可以运行 cargo-expand 并查看生成的代码是否有效：

fn foo(param: i32) {

    {
        ::std::io::_print(::core::fmt::Arguments::new_v1(&["", "\n"],
                                                         &match (&&param,) {
                                                              (arg0,) =>
                                                              [::core::fmt::ArgumentV1::new(arg0,
                                                                                            ::core::fmt::Display::fmt)],
                                                          }));
    }
}

（因为 println! 有点乱，但是你可以在那里看到有效的引用）

我什至可以将扩展复制并粘贴到我的源代码中，并且编译器会接受它。肯定有一些方法可以达到预期的结果吗？

好像大致相关：

"cannot find value `a` in this scope" in Rust macro
https://github.com/dtolnay/proc-macro-hack/issues/15

但我不能确定我是否遇到了完全相同的问题；我的情况似乎更基本。

Answer 1

Rust 宏是 hygienic，它可以防止标识符从一个作用域泄漏到另一个作用域并导致各种形式的混乱。但是您可以通过将 param 从 generate_func 传递到 generate_body:

来解决这个特殊问题

macro_rules! generate_func {
    ($name:ident) => {
        fn $name(param: i32) {
            generate_body!(param)
        }
    };
}

macro_rules! generate_body {
    ($param:ident) => {
        println!("{}", &$param);
    }
}

generate_func!(foo);

在 Playground 上查看。

有没有办法在 Rust 宏中引用局部变量？

Is there a way to reference a local variable within a Rust macro?

macros

rust