如何在Rust中基于编译时常量定义可开关的自定义断言？

需求背景 #

部分断言的计算成本较高，希望通过编译时配置分级启用/禁用断言：

• 当断言级别≥1时，启用assert_simple
• 当断言级别≥2时，额外启用assert_heavy
• 当断言级别=0时，禁用所有断言

示例目标代码：

fn do_something_complicated(a: u32, b: u32) -> bool {
    false
}

fn main() {
    let a = 5;
    let b = 10;
    assert_simple!(a != b);
    assert_heavy!(do_something_complicated(a, b));
}

当前方案的问题 #

之前的实现存在以下不足：

• 断言级别常量定义在主文件，无法和宏逻辑集中管理
• 暴露无业务用途的全局常量
• 其他模块需逐个引入宏，无法批量导入

优雅解决方案 #

方案1：基于编译时Feature控制（推荐） #

通过Cargo Feature实现编译时条件编译，完全移除未启用的断言代码，无需全局常量。

1. 在Cargo.toml中定义分级Feature：

[features]
default = ["assert_level_1"]  # 默认启用简单断言
assert_level_0 = []           # 禁用所有断言
assert_level_1 = []           # 启用简单断言
assert_level_2 = ["assert_level_1"]  # 启用所有断言

2. 创建断言模块（比如src/asserts.rs），实现宏并批量导出：

#[macro_export]
macro_rules! assert_simple {
    ($($arg:tt)*) => {
        #[cfg(feature = "assert_level_1")]
        assert!($($arg)*);
        #[cfg(not(feature = "assert_level_1"))]
        {}
    };
}

#[macro_export]
macro_rules! assert_heavy {
    ($($arg:tt)*) => {
        #[cfg(feature = "assert_level_2")]
        assert!($($arg)*);
        #[cfg(not(feature = "assert_level_2"))]
        {}
    };
}

// 批量导出宏，方便其他模块一次性导入
pub use {assert_simple, assert_heavy};

3. 在主文件或其他模块中使用：

mod asserts;
use asserts::*;

fn do_something_complicated(a: u32, b: u32) -> bool {
    false
}

fn main() {
    let a = 5;
    let b = 10;
    assert_simple!(a != b);
    assert_heavy!(do_something_complicated(a, b));
}

编译时通过Feature指定级别：

# 禁用所有断言
cargo build --features=assert_level_0

# 启用简单断言
cargo build --features=assert_level_1

# 启用所有断言
cargo build --features=assert_level_2

方案2：基于编译时常量控制 #

如果必须使用常量而非Feature，可将常量与宏集中管理，避免全局暴露：

1. 创建断言模块（src/asserts.rs）：

// 私有常量，仅模块内宏可访问
const MY_ASSERT_LEVEL: u32 = 2;

#[macro_export]
macro_rules! assert_simple {
    ($($arg:tt)*) => {
        if $crate::asserts::MY_ASSERT_LEVEL >= 1 {
            assert!($($arg)*);
        }
    };
}

#[macro_export]
macro_rules! assert_heavy {
    ($($arg:tt)*) => {
        if $crate::asserts::MY_ASSERT_LEVEL >= 2 {
            assert!($($arg)*);
        }
    };
}

pub use {assert_simple, assert_heavy};

2. 使用方式同方案1，只需修改常量值即可切换级别。

注意：此方案中，未启用的断言代码会被编译为空分支，但Rust编译器会在优化阶段将其移除，最终效果与Feature方案一致。

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