Rust过程宏中传递引用与值的机制及实现方法咨询

以编译时计算整数序列中位数为例，分别用C++和Rust实现了相关逻辑：

• C++中可实例化std::array后传入重载的median函数，示例代码：

std::array<int, 5> arr1 = {1, 2, 3, 4, 5};
auto median_seven = median<5>(arr1);

• 但在Rust中尝试用过程宏median!传入向量引用时：

let my_vec = vec![1,2,3,4,5];
let median = median!(&my_vec);

宏仅能识别到my_vec这个syn::Ident，无法区分引用/原值，也无法完成预期的编译时计算。

解决方案与惯用方法 #

1. 解析完整语法树节点，而非仅提取Ident #

过程宏接收的是完整的TokenStream，需通过syn库解析出完整的表达式节点，以此区分引用与原值：

use syn::{parse_macro_input, Expr, ExprReference, ExprPath};
use quote::quote;

#[proc_macro]
pub fn median(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    let expr = parse_macro_input!(input as Expr);
    
    let output = match expr {
        // 匹配引用表达式（&x 形式）
        Expr::Reference(ExprReference { expr, .. }) => {
            let inner_expr = *expr;
            // 生成针对引用的处理代码，比如解引用后访问值
            quote! {
                let target = #inner_expr;
                // 此处插入编译时中位数计算逻辑
            }
        }
        // 匹配直接变量/路径表达式（x 形式）
        Expr::Path(ExprPath { path, .. }) => {
            let ident = path.get_ident().expect("仅支持标识符作为参数");
            // 生成针对直接变量的处理代码
            quote! {
                let target = #ident;
                // 此处插入编译时中位数计算逻辑
            }
        }
        _ => panic!("median! 仅支持变量或变量引用作为参数"),
    };
    
    output.into()
}

2. 明确编译时计算的前提：参数必须是编译时常量 #

Rust过程宏无法获取运行时变量（比如vec!创建的动态向量）的值，因此若要实现编译时计算，传入的参数必须是编译时常量：

• 可以直接传入常量数组：

  let median = median!([1,2,3,4,5]);
  

  此时宏可解析ExprArray节点，直接提取元素进行编译时计算。
• 若要使用变量，需将变量声明为const：

  const MY_ARR: [i32; 5] = [1,2,3,4,5];
  let median = median!(&MY_ARR);
  

  宏解析引用后，可通过常量路径直接访问其值。

3. 惯用设计模式：直接接受常量表达式 #

Rust社区处理编译时计算的常见做法是让过程宏直接接收常量表达式（如字面量数组），而非变量引用，这样既简化宏的解析逻辑，也符合编译时计算的语义要求。

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