Rust/Cargo中类似C/C++ -DPREPROC_VAR的编译时配置方案

要实现类似C/C++预处理器的编译时配置const变量（带默认值），可以用以下几种可靠方案：

方案一：使用option_env!配合const上下文解析 #

env!会在环境变量不存在时直接panic，而option_env!返回Option<&'static str>，可以结合const上下文的解析逻辑设置默认值。注意需使用Rust 1.70+版本，该版本支持usize::from_str的const实现：

const MAX_THREADS: usize = match option_env!("MYLIB_MAX_THREADS") {
    Some(val) => match val.parse() {
        Ok(num) => num,
        Err(_) => 1000, // 解析失败时使用默认值
    },
    None => 1000,
};

编译时通过环境变量覆盖默认值：

MYLIB_MAX_THREADS=2000 cargo build

方案二：使用build.rs构建脚本（推荐） #

如果需要更灵活的配置逻辑（比如验证输入范围、处理复杂类型），用build.rs脚本生成配置代码是更稳妥的方式：

1. 在项目根目录创建build.rs文件：

use std::env;
use std::fs::File;
use std::io::Write;
use std::path::Path;

fn main() {
    // 读取环境变量，解析失败或未设置时 fallback 到默认值1000
    let max_threads = env::var("MYLIB_MAX_THREADS")
        .unwrap_or_else(|_| "1000".to_string())
        .parse::<usize>()
        .unwrap_or(1000);

    // 获取编译输出目录，生成配置文件
    let out_dir = env::var("OUT_DIR").unwrap();
    let dest_path = Path::new(&out_dir).join("config.rs");
    let mut f = File::create(&dest_path).unwrap();

    // 写入const变量定义
    writeln!(f, "pub const MAX_THREADS: usize = {};", max_threads).unwrap();
}

2. 在主 crate 的代码中导入生成的配置：

// 导入build.rs生成的const变量
include!(concat!(env!("OUT_DIR"), "/config.rs"));

fn main() {
    println!("Max threads: {}", MAX_THREADS);
}

3. 编译时覆盖默认值：

MYLIB_MAX_THREADS=2000 cargo build

关于Cargo.toml [env]段无效的说明 #

Cargo.toml中的[env]区块是用来设置构建脚本（build.rs）和依赖构建脚本的环境变量，而非直接传递给主 crate 的编译上下文。如果想通过Cargo.toml设置默认值，可在build.rs中读取该环境变量，或者改用方案一中的option_env!配合默认值逻辑。

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