Question

我需要一个宏，它将调用具有不同数量参数的函数或一个宏，它将从其（重复）参数生成一个有效的参数列表。

我很好地明确地给出了关于宏的参数数量的信息，但是我无法弄清楚如何为函数生成参数列表 - 我总是偶然发现宏返回表达式而不是令牌树。

macro_rules! call (
    ($f: expr, $($params:tt)*) => {
        $f(make_params!($($params:tt)*))
    };
);

macro_rules! make_params {
    () => {};
    (I $($params: tt)*) => {
        1, make_params!($($params:tt)*)
    };
}


fn foo(a: i32, b: i32, c: i32) {
    println!("foo: {} {} {}", a, b, c);
}

fn bar(a: i32, b: i32) {
    println!("bar: {} {}", a, b);
}

fn main() {
    call!(foo, I I I);
    call!(bar, I I);
}

编译器抱怨如下：

error: macro expansion ignores token `,` and any following
  --> src/main.rs:10:10
   |
10 |         1, make_params!($($params:tt)*)
   |          ^
   |
note: caused by the macro expansion here; the usage of `make_params!` is likely invalid in expression context
  --> src/main.rs:3:12
   |
3  |         $f(make_params!($($params:tt)*))
   |            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^
...

如何将make_params!作为令牌流（或类似）返回而不是表达？

我的真实用例比这个玩具示例更具参与性。我的函数有多种参数类型，它们以不同的方式构造。在我的情况下，只是制作宏call1，call2!，......似乎不是一个好的解决方案，因为我需要call_IIOOI，call_IIIO等等。

Answer 1

你需要在你去的时候逐步建立函数调用，最后只在一次发出它：

macro_rules! call (
    ($f: expr, $($params:tt)*) => {
        make_call!($f, () $($params)*)
    };
);

macro_rules! make_call {
    ($f: expr, ($($args:tt)*)) => { $f($($args)*) };
    ($f: expr, () I $($params:tt)*) => {
        make_call!($f, (1) $($params)*)
    };
    ($f: expr, ($($args:tt)*) I $($params:tt)*) => {
        make_call!($f, ($($args)*, 1) $($params)*)
    };
}

playground

在Rust宏中调用具有不同数量参数的函数

1 个答案: