]> git.bitcoin.ninja Git - rust-lightning/commitdiff
Add `source` and `target` to `DirectedChannelInfo`
authorjbesraa <jbesraa@gmail.com>
Tue, 5 Dec 2023 19:00:56 +0000 (21:00 +0200)
committerjbesraa <jbesraa@gmail.com>
Tue, 5 Dec 2023 19:07:45 +0000 (21:07 +0200)
  `DirectedChannelInfo::source` return the `node_id`
  of the forwarding hop.
  `DirectedChannelInfo::target` return the `node_id`
  of the destination hop.

lightning/src/routing/gossip.rs

index 68bc7945acd9bb659bf530a6752e351ab14e0681..edb372616972770fc1481cd576d99718c32002b0 100644 (file)
@@ -1043,6 +1043,16 @@ impl<'a> DirectedChannelInfo<'a> {
        /// Returns information for the direction.
        #[inline]
        pub(super) fn direction(&self) -> &'a ChannelUpdateInfo { self.direction }
+
+       /// Returns the `node_id` of the source hop.
+       ///
+       /// Refers to the `node_id` forwarding the payment to the next hop.
+       pub(super) fn source(&self) -> &'a NodeId { if self.outbound { &self.channel.node_one } else { &self.channel.node_two } }
+
+       /// Returns the `node_id` of the target hop.
+       ///
+       /// Refers to the `node_id` receiving the payment from the previous hop.
+       pub(super) fn target(&self) -> &'a NodeId { if self.outbound { &self.channel.node_two } else { &self.channel.node_one } }
 }
 
 impl<'a> fmt::Debug for DirectedChannelInfo<'a> {