Skip to content

Commit 130921b

Browse files
committed
f - support multiple patterns
1 parent 765f455 commit 130921b

File tree

1 file changed

+9
-7
lines changed

1 file changed

+9
-7
lines changed

lightning/src/ln/peer_handler.rs

+9-7
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,7 @@ macro_rules! composite_custom_message_handler {
6767
}
6868

6969
$message_visibility:vis enum $message:ident {
70-
$($variant:ident($range:pat)),* $(,)*
70+
$($variant:ident($($pattern:pat)|+)),* $(,)*
7171
}
7272
) => {
7373
#[allow(missing_docs)]
@@ -92,10 +92,12 @@ macro_rules! composite_custom_message_handler {
9292
) -> Result<Option<Self::CustomMessage>, msgs::DecodeError> {
9393
match message_type {
9494
$(
95-
$range => match <$type>::read(&self.$field, message_type, buffer)? {
96-
None => unreachable!(),
97-
Some(message) => Ok(Some($message::$variant(message))),
98-
},
95+
$(
96+
$pattern => match <$type>::read(&self.$field, message_type, buffer)? {
97+
None => unreachable!(),
98+
Some(message) => Ok(Some($message::$variant(message))),
99+
},
100+
)*
99101
)*
100102
_ => Ok(None),
101103
}
@@ -131,8 +133,8 @@ composite_custom_message_handler!(
131133
}
132134

133135
pub enum CompositeMessage {
134-
Infallible(0),
135-
Infallible2(1..=10),
136+
Infallible(0 | 2 | 4),
137+
Infallible2(1..=10 | 99),
136138
}
137139
);
138140

0 commit comments

Comments
 (0)