Rust过程宏接收宏生成字符串报错，求解决方法

问题场景 #

我通过include_str!从文件读取字符串，再传递给自定义的过程宏时，出现了expected string literal错误，无法正常编译运行。

相关代码 #

主调用代码 #

macro_rules! ptcl_layer {
    () => {
        include_str!("tcp_to_msg.layer")
    };
}

pub static TEMPLETE: &'static str = ptcl_layer!();
pub static TEMPLETE_HEADER: &'static str = get_protocol_layer_header!(ptcl_layer!());
pub static TEMPLETE_END: &'static str = get_protocol_layer_end!(ptcl_layer!());

过程宏实现代码 #

#[proc_macro]
pub fn get_protocol_layer_header(_s: TokenStream) -> TokenStream {
    let total_str = parse_macro_input!(_s as LitStr).value();
    let header = total_str.lines().next().unwrap();
    let header = header.replace("${.LEN}\n", "");
    let output = quote::quote! {
        #header
    };
    TokenStream::from(output)
}

#[proc_macro]
pub fn get_protocol_layer_end(_s: TokenStream) -> TokenStream {
    let input = parse_macro_input!(_s as LitStr);
    let input = input.value();
    let last_line = input.lines().last().unwrap_or("");
    let output = quote::quote! {
        #last_line
    };
    
    TokenStream::from(output)
}

tcp_to_msg.layer文件内容 #

<? lines: ${.LEN}
${.CONTENT}
<? end

错误信息 #

error: expected string literal
  --> src/controller/mod.rs:16:71
   |
16 | pub static TEMPLETE_HEADER: &'static str = get_protocol_layer_header!(ptcl_layer!());
   |                                                                       ^^^^^^^^^^

尝试过的无效方案 #

我修改了过程宏实现，直接将TokenStream转为字符串处理，但结果是返回未处理的原始文本，问题依旧：

#[proc_macro]
pub fn get_protocol_layer_header(_s: TokenStream) -> TokenStream {
    let total_str = _s.to_string();
    let header = total_str.lines().next().unwrap();
    let header = header.replace("${.LEN}\n", "");
    header.parse().unwrap()
}

#[proc_macro]
pub fn get_protocol_layer_end(_s: TokenStream) -> TokenStream {
    let total_str = _s.to_string();
    let last_line = total_str.lines().last().unwrap_or("");
    last_line.parse().unwrap()
}

解决方案 #

问题根源：ptcl_layer!()展开后是include_str!(...)表达式，而你的过程宏期望接收字符串字面量（LitStr），编译器无法将表达式直接解析为字面量，因此报错。提供两种可行解决思路：

方案1：直接传递include_str!到过程宏（编译期处理） #

include_str!本身在编译阶段会被替换为字符串字面量，因此可以直接传入过程宏，不需要额外的声明宏封装：

pub static TEMPLETE: &'static str = include_str!("tcp_to_msg.layer");
pub static TEMPLETE_HEADER: &'static str = get_protocol_layer_header!(include_str!("tcp_to_msg.layer"));
pub static TEMPLETE_END: &'static str = get_protocol_layer_end!(include_str!("tcp_to_msg.layer"));

此方案下字符串处理逻辑在编译期完成，运行时无额外开销。

方案2：修改过程宏支持接收任意表达式（运行时处理） #

如果需要保留声明宏的封装，可以修改过程宏，让它接收表达式而非仅字面量，在生成的代码中执行字符串处理：

#[proc_macro]
pub fn get_protocol_layer_header(input: TokenStream) -> TokenStream {
    let expr = parse_macro_input!(input as Expr);
    
    let output = quote::quote! {
        {
            let total_str = #expr;
            let header = total_str.lines().next().unwrap();
            header.replace("${.LEN}", "")
        }
    };
    
    TokenStream::from(output)
}

#[proc_macro]
pub fn get_protocol_layer_end(input: TokenStream) -> TokenStream {
    let expr = parse_macro_input!(input as Expr);
    
    let output = quote::quote! {
        {
            let total_str = #expr;
            total_str.lines().last().unwrap_or("")
        }
    };
    
    TokenStream::from(output)
}

此时可以保留原调用方式get_protocol_layer_header!(ptcl_layer!())，但字符串处理逻辑会在运行时执行。

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