X-Git-Url: http://git.bitcoin.ninja/index.cgi?a=blobdiff_plain;f=lightning%2Fsrc%2Frouting%2Fscorer.rs;h=8b5fae70abdbffa34e6a365f312eb654f5165d84;hb=e3bdfa0585b9900cd732d568169a3778005be523;hp=2aa0b4c145e67ef3c285ec2901d6a92631594ff5;hpb=ae210e7d09758560b0e418b939222e7bc26ee009;p=rust-lightning diff --git a/lightning/src/routing/scorer.rs b/lightning/src/routing/scorer.rs index 2aa0b4c1..8b5fae70 100644 --- a/lightning/src/routing/scorer.rs +++ b/lightning/src/routing/scorer.rs @@ -220,7 +220,7 @@ impl routing::Score for ScorerUsingTime { self.params.base_penalty_msat + failure_penalty_msat } - fn payment_path_failed(&mut self, _path: &Vec, short_channel_id: u64) { + fn payment_path_failed(&mut self, _path: &[&RouteHop], short_channel_id: u64) { let failure_penalty_msat = self.params.failure_penalty_msat; let half_life = self.params.failure_penalty_half_life; self.channel_failures @@ -275,34 +275,48 @@ impl Writeable for ScorerUsingTime { #[inline] fn write(&self, w: &mut W) -> Result<(), io::Error> { self.params.write(w)?; - self.channel_failures.write(w) + self.channel_failures.write(w)?; + write_tlv_fields!(w, {}); + Ok(()) } } impl Readable for ScorerUsingTime { #[inline] fn read(r: &mut R) -> Result { - Ok(Self { + let res = Ok(Self { params: Readable::read(r)?, channel_failures: Readable::read(r)?, - }) + }); + read_tlv_fields!(r, {}); + res } } impl Writeable for ChannelFailure { #[inline] fn write(&self, w: &mut W) -> Result<(), io::Error> { - self.undecayed_penalty_msat.write(w)?; - (T::duration_since_epoch() - self.last_failed.elapsed()).write(w) + let duration_since_epoch = T::duration_since_epoch() - self.last_failed.elapsed(); + write_tlv_fields!(w, { + (0, self.undecayed_penalty_msat, required), + (2, duration_since_epoch, required), + }); + Ok(()) } } impl Readable for ChannelFailure { #[inline] fn read(r: &mut R) -> Result { + let mut undecayed_penalty_msat = 0; + let mut duration_since_epoch = Duration::from_secs(0); + read_tlv_fields!(r, { + (0, undecayed_penalty_msat, required), + (2, duration_since_epoch, required), + }); Ok(Self { - undecayed_penalty_msat: Readable::read(r)?, - last_failed: T::now() - (T::duration_since_epoch() - Readable::read(r)?), + undecayed_penalty_msat, + last_failed: T::now() - (T::duration_since_epoch() - duration_since_epoch), }) } }