过程宏属性解析出错时，如何同时报错并消除无main函数错误？

核心原因 #

当你用syn::Error生成编译报错后，如果宏返回的TokenStream里没有有效的main函数定义，编译器在处理完属性错误提示后，仍会检查程序入口，进而抛出「找不到main函数」的错误。之前的尝试可能只返回了错误信息，或是合并逻辑有误，导致有效的main函数未被正确注入。

解决方案 #

正确的处理逻辑是：先将编译错误转换为TokenStream，再与空main函数的TokenStream合并后返回。这样编译器会优先报告属性参数的错误，同时因为存在有效的main函数定义，不会触发入口缺失的报错。

代码示例 #

use proc_macro::TokenStream;
use syn::{parse_macro_input, AttributeArgs, ItemFn, Error};
use quote::quote;

#[proc_macro_attribute]
pub fn custom_main_attr(args: TokenStream, input: TokenStream) -> TokenStream {
    let args = parse_macro_input!(args as AttributeArgs);
    let _input_fn = parse_macro_input!(input as ItemFn);

    // 检测属性中的非法字面量（示例：禁止字符串字面量"forbidden"）
    if let Some(invalid_lit) = args.iter().find(|arg| {
        matches!(arg, syn::NestedMeta::Lit(syn::Lit::Str(s)) if s.value() == "forbidden")
    }) {
        // 生成属性错误
        let err = Error::new_spanned(invalid_lit, "该字面量不允许在属性中使用");
        // 将错误转为编译器可识别的TokenStream
        let err_ts = err.into_compile_error();
        // 定义空的main函数
        let empty_main = quote! {
            fn main() {}
        };
        // 合并错误流与空main流并返回
        return (err_ts.into_iter().chain(empty_main.into_iter())).collect();
    }

    // 正常逻辑：返回原main函数或自定义处理后的函数
    input
}

关键细节 #

• Error::into_compile_error()会把错误转换为compile_error!宏调用的TokenStream，确保错误信息被正确输出。
• 用iter().chain()拼接错误和空main的Token流，保证两者都能被编译器处理。
• 即便编译器报告了属性错误，由于空main函数存在，程序入口检查会通过，不会额外抛出「找不到main函数」的错误。

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