Skip to content

Commit 6621042

Browse files
committed
f Introduce write_event function
1 parent fdef833 commit 6621042

File tree

1 file changed

+13
-26
lines changed

1 file changed

+13
-26
lines changed

src/event.rs

Lines changed: 13 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,14 @@ trait EventType {
5858
const TYPE: u8;
5959
}
6060

61+
fn write_event<T: EventType + Writeable, W: Writer>(
62+
event: &T, writer: &mut W,
63+
) -> Result<(), lightning::io::Error> {
64+
T::TYPE.write(writer)?;
65+
event.write(writer)?;
66+
Ok(())
67+
}
68+
6169
impl Readable for LdkLiteEvent {
6270
fn read<R: lightning::io::Read>(
6371
reader: &mut R,
@@ -84,31 +92,12 @@ impl Readable for LdkLiteEvent {
8492
impl Writeable for LdkLiteEvent {
8593
fn write<W: Writer>(&self, writer: &mut W) -> Result<(), lightning::io::Error> {
8694
match self {
87-
Self::PaymentSuccessful(event) => {
88-
event.write(writer)?;
89-
Ok(())
90-
}
91-
Self::PaymentFailed(event) => {
92-
event.write(writer)?;
93-
Ok(())
94-
}
95-
Self::PaymentReceived(event) => {
96-
event.write(writer)?;
97-
Ok(())
98-
}
99-
//Self::ChannelOpened { .. } => {
100-
//TODO
101-
//}
102-
Self::ChannelClosed(event) => {
103-
event.write(writer)?;
104-
Ok(())
105-
} //Self::OnChainPaymentSent { .. } => {
106-
//TODO
107-
//}
108-
//Self::OnChainPaymentReceived { .. } => {
109-
//TODO
110-
//}
95+
Self::PaymentSuccessful(event) => write_event(event, writer)?,
96+
Self::PaymentFailed(event) => write_event(event, writer)?,
97+
Self::PaymentReceived(event) => write_event(event, writer)?,
98+
Self::ChannelClosed(event) => write_event(event, writer)?,
11199
}
100+
Ok(())
112101
}
113102
}
114103

@@ -137,7 +126,6 @@ impl Readable for PaymentSuccessfulEvent {
137126

138127
impl Writeable for PaymentSuccessfulEvent {
139128
fn write<W: Writer>(&self, writer: &mut W) -> Result<(), lightning::io::Error> {
140-
Self::TYPE.write(writer)?;
141129
self.payment_hash.write(writer)?;
142130
self.inner.write(writer)?;
143131
Ok(())
@@ -169,7 +157,6 @@ impl Readable for PaymentFailedEvent {
169157

170158
impl Writeable for PaymentFailedEvent {
171159
fn write<W: Writer>(&self, writer: &mut W) -> Result<(), lightning::io::Error> {
172-
Self::TYPE.write(writer)?;
173160
self.payment_hash.write(writer)?;
174161
self.inner.write(writer)?;
175162
Ok(())

0 commit comments

Comments
 (0)