Rust过程宏：如何检测已启用的特性

核心思路 #

过程宏确实运行在编译阶段，但CARGO_FEATURE_*环境变量仅在构建脚本中生效，直接在宏代码中读取无效。我们需要利用Rust编译期的cfg相关机制来实现特性感知的解析逻辑，同时解决LSP报错问题。

解决方案1：在宏内部使用cfg!编译期宏 #

cfg!是编译期求值的宏，可以直接在过程宏代码中检测特性是否启用，根据结果分支处理语法解析。

use proc_macro::TokenStream;
use syn::{parse_macro_input, DeriveInput};

#[proc_macro_derive(MyDerive)]
pub fn my_derive(input: TokenStream) -> TokenStream {
    let input = parse_macro_input!(input as DeriveInput);
    
    let expanded = if cfg!(feature = "special_parsing") {
        // 仅在special_parsing特性启用时执行的解析/生成逻辑
        generate_special_code(&input)
    } else {
        // 默认逻辑
        generate_default_code(&input)
    };
    
    expanded.into()
}

注意：如果LSP报错，是因为它默认可能未加载目标特性，需要手动配置LSP的特性集（见方案3）。

解决方案2：让用户在宏调用处添加#[cfg]属性 #

将特性判断的控制权交给宏的调用者，只有当特性启用时，才会调用对应的宏逻辑，从根源避免LSP解析未启用特性的代码。

用户侧代码：

#[cfg(feature = "special_syntax")]
my_macro! {
    // 仅在特性启用时才会被解析的语法内容
    special_item: "value"
}

宏侧代码：

#[proc_macro]
pub fn my_macro(input: TokenStream) -> TokenStream {
    // 直接处理特定语法，因为只有特性启用时才会进入此分支
    let parsed = parse_macro_input!(input as SpecialSyntax);
    generate_special_output(parsed).into()
}

解决方案3：配置LSP以加载目标特性 #

如果使用方案1后LSP仍报错，需要让Rust Analyzer（主流Rust LSP）加载指定特性：

• 在项目根目录创建.rust-analyzer/cargo.toml：

  [features]
  default = ["special_parsing"]
  

• 若使用VS Code，可在.vscode/settings.json中添加：

  "rust-analyzer.cargo.features": ["special_parsing"]
  

配置后，LSP会按照指定特性集解析代码，宏中的cfg!分支会被正确识别。

解决方案4：使用Span::def_site().cfg()获取编译配置（高级用法） #

通过过程宏的SpanAPI，可以获取宏定义位置的编译配置，精确匹配特性是否启用：

use proc_macro::{Span, TokenStream};
use syn::parse::Parse;

#[proc_macro]
pub fn my_macro(input: TokenStream) -> TokenStream {
    let cfg = Span::def_site().cfg();
    let has_special_feature = cfg.iter().any(|item| {
        matches!(item, proc_macro::Cfg::Feature { name, .. } if name == "special_parsing")
    });

    let parsed = if has_special_feature {
        parse_macro_input!(input as SpecialInput)
    } else {
        parse_macro_input!(input as DefaultInput)
    };

    // 生成对应输出
    TokenStream::new()
}

此方法更底层，适合需要精确控制编译配置的场景，一般推荐优先使用cfg!。

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