我正在尝试编写一个将重写某些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
}
};
}
但这会产生错误,因为现在对于后续块是否也应与tt
相匹配尚不明确:
error: local ambiguity: multiple parsing options: built-in NTs tt ('pred') or block ('r1').
有没有办法做到这一点,还是我坚持发明在宏中使用特殊语法?
答案 0 :(得分:1)
您可以使用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