我正在尝试编写一个宏来重写某些 Rust 控制流,但我在匹配 if
表达式时遇到困难。问题在于谓词是一个表达式,但是 expr
不允许后跟 block
或 {
。
我最好的方法是使用 tt
:
macro_rules! branch {
(
if $pred:tt
$r1:block
else
$r2:block
) => {
if $pred {
$r1
} else {
$r2
}
};
}
这适用于单标记或分组谓词:
branch! {
if (foo == bar) {
1
} else {
2
}
}
但如果谓词未分组则失败:
branch! {
if foo == bar {
1
} else {
2
}
}
error: no rules expected the token `==`
我还尝试在谓词中使用 tt
的重复模式:
macro_rules! branch {
(
if $($pred:tt)+
$r1:block
else
$r2:block
) => {
if $($pred)+ {
$r1
} else {
$r2
}
};
}
但这会产生一个错误,因为现在不清楚后续 block 是否也应与 tt
匹配:
error: local ambiguity: multiple parsing options: built-in NTs tt ('pred') or block ('r1').
有没有办法做到这一点,还是我坚持发明特殊语法以在宏中使用?
最佳答案
你可以使用 TT muncher解析谓词:
macro_rules! branch {
{
if $($rest:tt)*
} => {
branch_parser! {
predicate = ()
rest = ($($rest)*)
}
};
}
macro_rules! branch_parser {
{
predicate = ($($predicate:tt)*)
rest = ({ $($then:tt)* } else { $($else:tt)* })
} => {
println!("predicate: {}", stringify!($($predicate)*));
println!("then: {}", stringify!($($then)*));
println!("else: {}", stringify!($($else)*));
};
{
predicate = ($($predicate:tt)*)
rest = ($next:tt $($rest:tt)*)
} => {
branch_parser! {
predicate = ($($predicate)* $next)
rest = ($($rest)*)
}
};
}
fn main() {
branch! {
if foo == bar {
1
} else {
2
}
}
}
输出:
predicate: foo == bar
then: 1
else: 2
关于macros - 如何在宏中匹配 Rust 的 `if` 表达式?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/51932944/