如何用Rust宏简化直方图统计中的重复match分支？

你的代码里所有区间都是固定2000步长，手动写几十条match分支确实冗余。用宏自动生成这些分支是可行的，同时还有更简洁的非宏方案，下面分别说明：

一、用宏生成match分支 #

可以写一个递归宏来自动生成所有固定步长的区间分支，避免手动重复编写：

use std::collections::HashMap;

// 递归宏：生成区间匹配逻辑
macro_rules! process_histogram_interval {
    ($value:expr, $counts:expr, $start:expr, $step:expr, $remaining:expr, $default_key:expr) => {
        if $remaining == 0 {
            // 处理超出最大区间的情况
            *$counts.entry($default_key.to_string()).or_insert(0) += 1;
        } else {
            let current_end = $start + $step - 1;
            if $value >= $start && $value <= current_end {
                *$counts.entry(format!("{}-{}", $start, current_end)).or_insert(0) += 1;
            } else {
                // 递归处理下一个区间
                process_histogram_interval!($value, $counts, $start + $step, $step, $remaining - 1, $default_key);
            }
        }
    };
}

pub fn make_histogram(array: &[u64]) {
    let mut counts = HashMap::new();
    const STEP: u64 = 2000;
    // 总区间数：从1到60000共30个2000步长的区间
    const INTERVAL_COUNT: usize = 30;

    for &value in array {
        process_histogram_interval!(value, counts, 1, STEP, INTERVAL_COUNT, "60001-inf");
    }
}

这个宏通过递归逐步生成每个区间的判断逻辑，只需要指定起始值、步长、区间总数和默认键，就能自动处理所有分支。

如果想直接展开成原生的match分支（编译期生成，性能和手写一致），可以用另一种声明宏：

use std::collections::HashMap;

// 生成match分支的宏
macro_rules! histogram_match {
    ($value:expr, $counts:expr, $default:expr) => {
        match $value {
            _ => *$counts.entry($default).or_insert(0) += 1,
        }
    };
    ($value:expr, $counts:expr, $default:expr, $start:expr, $end:expr, $($rest:tt)*) => {
        match $value {
            $start..=$end => *$counts.entry(concat!($start, "-", $end)).or_insert(0) += 1,
            _ => histogram_match!($value, $counts, $default, $($rest)*),
        }
    };
}

pub fn make_histogram(array: &[u64]) {
    let mut counts = HashMap::new();

    for &value in array {
        histogram_match!(
            value,
            counts,
            "60001-inf",
            1, 2000,
            2001, 4000,
            4001, 6000,
            // ... 这里可以继续添加区间，直到58001, 60000
            58001, 60000
        );
    }
}

这种方式会在编译期展开成和你手写一模一样的match分支，性能没有损耗，但还是需要手动列出每个区间的起止（适合步长不固定的场景）。

二、更简洁的非宏方案（推荐） #

因为你的区间是严格固定步长的，完全可以通过数学计算直接推导对应的区间键，不需要宏也不需要match：

use std::collections::HashMap;

pub fn make_histogram(array: &[u64]) {
    let mut counts = HashMap::new();
    const STEP: u64 = 2000;
    const MAX_INTERVAL_END: u64 = 60000;

    for &value in array {
        let key = if value > MAX_INTERVAL_END {
            "60001-inf".to_string()
        } else {
            // 计算当前值所属的区间起始和结束
            let interval_idx = (value - 1) / STEP;
            let start = interval_idx * STEP + 1;
            let end = start + STEP - 1;
            format!("{}-{}", start, end)
        };
        *counts.entry(key).or_insert(0) += 1;
    }
}

这个方案代码量极少，逻辑清晰，维护成本远低于宏或手写match，而且性能和手写分支几乎无差别，更推荐使用。

内容的提问来源于stack exchange，提问作者Wouter De Coster