如何向创建函数的Rust宏传递format!字符串及参数？

问题根源 #

原来的宏调用失败，核心原因是你在宏参数里直接写了arg.id——宏是在调用它的作用域中展开的，此时arg还不存在（arg是宏生成的函数的参数，只有在函数内部的作用域里才有效）。编译器会认为arg是未定义变量，因此报错。

解决方案：将URL构建逻辑延迟到函数内部执行 #

我们需要把从请求参数提取URL片段的逻辑，放到宏生成的函数内部（也就是arg存在的作用域里）。下面提供几种实用的实现方式：

方案1：直接传入完整的URL构建表达式 #

调整宏，让它接受一个在函数内部执行的表达式，用来生成URL路径部分：

macro_rules! api_request_with_path {
    ($name: ident, $method: ident, $request_type: ty, $return_type: ty, $path_expr: expr) => {
        pub fn $name(arg: $request_type) -> $return_type {
            // $path_expr会在函数内部执行，此时arg已经是有效参数
            let action_url = format!("https://example.com/api{}", $path_expr);
            let action_result = make_request($method, action_url);
            action_result.into()
        }
    };
}

// 调用时直接传入使用arg的format表达式
api_request_with_path!(
    get_thing_status,
    GET,
    SomeThing,
    SomeThingStatus,
    format!("/things/{}/status", arg.id)
);

方案2：拆分格式字符串与参数表达式 #

如果想把格式字符串和参数分开，让宏结构更清晰，可以使用可变参数接收多个参数表达式：

macro_rules! api_request_with_path {
    ($name: ident, $method: ident, $request_type: ty, $return_type: ty, $fmt: expr, $($arg_expr: expr),*) => {
        pub fn $name(arg: $request_type) -> $return_type {
            let path_fragment = format!($fmt, $($arg_expr),*);
            let action_url = format!("https://example.com/api{}", path_fragment);
            let action_result = make_request($method, action_url);
            action_result.into()
        }
    };
}

// 调用时传入格式字符串和基于arg的参数表达式
api_request_with_path!(
    get_thing_status,
    GET,
    SomeThing,
    SomeThingStatus,
    "/things/{}/status",
    arg.id
);

方案3：使用闭包处理复杂URL逻辑 #

如果URL需要更复杂的构建逻辑（比如多参数拼接、条件判断），可以传入一个闭包，接收请求参数并返回路径字符串：

macro_rules! api_request_with_path {
    ($name: ident, $method: ident, $request_type: ty, $return_type: ty, $path_builder: expr) => {
        pub fn $name(arg: $request_type) -> $return_type {
            // 调用传入的闭包生成路径片段
            let path = ($path_builder)(&arg);
            let action_url = format!("https://example.com/api{}", path);
            let action_result = make_request($method, action_url);
            action_result.into()
        }
    };
}

// 调用时传入闭包处理复杂路径构建
api_request_with_path!(
    get_thing_status,
    GET,
    SomeThing,
    SomeThingStatus,
    |thing: &SomeThing| format!("/things/{}/status", thing.id)
);

验证效果 #

以上三种方案都能解决原问题，因为所有涉及arg的表达式都会在宏生成的函数内部执行，此时arg是有效的函数参数，编译器不会再报错。你可以根据API的复杂度选择合适的方案。

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