如何将多套相似嵌套循环的Rust宏规则简化为单规则？

完全没问题，Rust的宏重复语法$(...)*和$(...)+就是为这种场景设计的——能把你那三套高度相似的规则合并成一个，还能完美实现嵌套循环逻辑。我给你一步步演示怎么做：

先看原来的问题场景（假设的三套规则） #

比如你原来的宏大概是这样，分别处理1、2、3层for循环：

macro_rules! old_nested_loop {
    // 1层循环
    (for $i:ident in $iter:expr, $body:expr) => {
        for $i in $iter {
            $body
        }
    };
    // 2层嵌套
    (for $i:ident in $iter:expr, for $j:ident in $j_iter:expr, $body:expr) => {
        for $i in $iter {
            for $j in $j_iter {
                $body
            }
        }
    };
    // 3层嵌套
    (for $i:ident in $iter:expr, for $j:ident in $j_iter:expr, for $k:ident in $k_iter:expr, $body:expr) => {
        for $i in $iter {
            for $j in $j_iter {
                for $k in $k_iter {
                    $body
                }
            }
        }
    };
}

合并成单套规则的宏 #

利用宏重复语法，我们可以把这三套规则压缩成一个，而且支持任意层数的嵌套循环：

macro_rules! nested_loop {
    // 匹配任意数量的`for var in iter,`模式，最后跟上代码块
    ($(for $var:ident in $iter:expr,)* $body:expr) => {
        // 先重复生成每一层循环的开头
        $(
            for $var in $iter {
        )*
            // 插入要执行的核心代码
            $body
        // 再对应重复生成每一层循环的结尾
        $(
            }
        )*
    };
}

原理说明 #

这里的核心是宏重复的对称展开特性：

• 匹配阶段：$(for $var:ident in $iter:expr,)*会自动捕获所有for 变量 in 迭代器,的片段，不管有多少个；
• 展开阶段：$(for $var in $iter {)*会把捕获到的每一个循环片段展开成对应的循环开头，然后插入核心代码$body，最后用$(})*对应补上每个循环的闭合大括号——完美形成嵌套结构。

用法示例 #

现在你可以用这个宏处理任意层数的嵌套循环了：

fn main() {
    // 1层循环
    nested_loop!(
        for x in 0..3,
        println!("x: {}", x);
    );

    // 2层嵌套循环
    nested_loop!(
        for x in 0..2,
        for y in 0..2,
        println!("x: {}, y: {}", x, y);
    );

    // 3层嵌套循环
    nested_loop!(
        for x in 0..2,
        for y in 0..2,
        for z in 0..2,
        println!("x: {}, y: {}, z: {}", x, y, z);
    );
}

进阶：避免空循环的情况 #

如果你想确保用户至少传入一个循环（防止空循环无意义的展开），可以把*换成+，同时调整匹配语法让它更严谨：

macro_rules! nested_loop {
    // 用`=>`分隔循环列表和代码块，确保至少有一个循环
    ($(for $var:ident in $iter:expr),+ => $body:expr) => {
        $(
            for $var in $iter {
        )*
            $body
        $(
            }
        )*
    };
}

使用的时候改成这样：

nested_loop!(
    for x in 0..3,
    for y in 0..3
    => println!("x: {}, y: {}", x, y);
);

这样不仅合并了重复的规则，还让宏的扩展性更强——以后要支持4层、5层循环完全不用改宏的代码，直接用就行！

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