X-Git-Url: http://git.bitcoin.ninja/index.cgi?a=blobdiff_plain;ds=sidebyside;f=lightning%2Fsrc%2Frouting%2Fscoring.rs;h=0093513ee8982628ad5626a0beb14490b1231ae6;hb=9fbafd4b6cc7a290fbd952d8baf58d6017fc3078;hp=0e04c63903392b8ccfc36d72c3fa87109b86c17b;hpb=62edee568985e3362bd1609c6089d05428023925;p=rust-lightning

diff --git a/lightning/src/routing/scoring.rs b/lightning/src/routing/scoring.rs
index 0e04c639..0093513e 100644
--- a/lightning/src/routing/scoring.rs
+++ b/lightning/src/routing/scoring.rs
@@ -137,6 +137,14 @@ pub trait LockableScore<'a> {
 	fn lock(&'a self) -> Self::Locked;
 }
 
+/// Refers to a scorer that is accessible under lock and also writeable to disk
+///
+/// We need this trait to be able to pass in a scorer to `lightning-background-processor` that will enable us to
+/// use the Persister to persist it.
+pub trait WriteableScore<'a>: LockableScore<'a> + Writeable {}
+
+impl<'a, T> WriteableScore<'a> for T where T: LockableScore<'a> + Writeable {}
+
 /// (C-not exported)
 impl<'a, T: 'a + Score> LockableScore<'a> for Mutex<T> {
 	type Locked = MutexGuard<'a, T>;
@@ -623,6 +631,33 @@ impl<G: Deref<Target = NetworkGraph>, L: Deref, T: Time> ProbabilisticScorerUsin
 		assert!(self.channel_liquidities.insert(short_channel_id, liquidity).is_none());
 		self
 	}
+
+	/// Dump the contents of this scorer into the configured logger.
+	///
+	/// Note that this writes roughly one line per channel for which we have a liquidity estimate,
+	/// which may be a substantial amount of log output.
+	pub fn debug_log_liquidity_stats(&self) {
+		let graph = self.network_graph.read_only();
+		for (scid, liq) in self.channel_liquidities.iter() {
+			if let Some(chan_debug) = graph.channels().get(scid) {
+				let log_direction = |source, target| {
+					if let Some((directed_info, _)) = chan_debug.as_directed_to(target) {
+						let amt = directed_info.effective_capacity().as_msat();
+						let dir_liq = liq.as_directed(source, target, amt, self.params.liquidity_offset_half_life);
+						log_debug!(self.logger, "Liquidity from {:?} to {:?} via {} is in the range ({}, {})",
+							source, target, scid, dir_liq.min_liquidity_msat(), dir_liq.max_liquidity_msat());
+					} else {
+						log_debug!(self.logger, "No amount known for SCID {} from {:?} to {:?}", scid, source, target);
+					}
+				};
+
+				log_direction(&chan_debug.node_one, &chan_debug.node_two);
+				log_direction(&chan_debug.node_two, &chan_debug.node_one);
+			} else {
+				log_debug!(self.logger, "No network graph entry for SCID {}", scid);
+			}
+		}
+	}
 }
 
 impl ProbabilisticScoringParameters {