add lightweight update_chain_tip() for fast invoice expiry refresh

Author: Fernando Ledesma

src/chain/esplora.rs

@@ -363,6 +363,34 @@ impl EsploraChainSource {
 		Ok(())
 	}
 
+	pub(super) async fn fetch_chain_tip(
+		&self,
+	) -> Result<(bitcoin::block::Header, u32), Error> {
+		let tip_hash = self.esplora_client.get_tip_hash().await.map_err(|e| {
+			log_error!(self.logger, "Failed to get chain tip hash: {}", e);
+			Error::TxSyncFailed
+		})?;
+
+		let header =
+			self.esplora_client.get_header_by_hash(&tip_hash).await.map_err(|e| {
+				log_error!(self.logger, "Failed to get block header: {}", e);
+				Error::TxSyncFailed
+			})?;
+
+		let status =
+			self.esplora_client.get_block_status(&tip_hash).await.map_err(|e| {
+				log_error!(self.logger, "Failed to get block status: {}", e);
+				Error::TxSyncFailed
+			})?;
+
+		let height = status.height.ok_or_else(|| {
+			log_error!(self.logger, "Block not in best chain");
+			Error::TxSyncFailed
+		})?;
+
+		Ok((header, height))
+	}
+
 	pub(crate) async fn process_broadcast_package(&self, package: Vec<Transaction>) {
 		for tx in &package {
 			let txid = tx.compute_txid();

src/chain/mod.rs

@@ -431,6 +431,17 @@ impl ChainSource {
 		}
 	}
 
+	pub(crate) async fn fetch_chain_tip(
+		&self,
+	) -> Result<(bitcoin::block::Header, u32), Error> {
+		match &self.kind {
+			ChainSourceKind::Esplora(esplora_chain_source) => {
+				esplora_chain_source.fetch_chain_tip().await
+			},
+			_ => Err(Error::TxSyncFailed),
+		}
+	}
+
 	pub(crate) async fn continuously_process_broadcast_queue(
 		&self, mut stop_tx_bcast_receiver: tokio::sync::watch::Receiver<()>,
 	) {

src/io/utils.rs

@@ -45,7 +45,7 @@ use crate::fee_estimator::OnchainFeeEstimator;
 use crate::io::{
 	NODE_METRICS_KEY, NODE_METRICS_PRIMARY_NAMESPACE, NODE_METRICS_SECONDARY_NAMESPACE,
 };
-use crate::logger::{log_error, LdkLogger, Logger};
+use crate::logger::{log_error, log_info, LdkLogger, Logger};
 use crate::peer_store::PeerStore;
 use crate::types::{Broadcaster, DynStore, KeysManager, Sweeper, WordCount};
 use crate::wallet::ser::{ChangeSetDeserWrapper, ChangeSetSerWrapper};
@@ -161,21 +161,24 @@ pub(crate) async fn read_scorer<G: Deref<Target = NetworkGraph<L>>, L: Deref + C
 where
 	L::Target: LdkLogger,
 {
+	log_info!(logger, "[SCORER-LDK] read_scorer() - Attempting to read scorer from VSS store");
 	let params = ProbabilisticScoringDecayParameters::default();
-	let mut reader = Cursor::new(
-		KVStore::read(
-			&*kv_store,
-			SCORER_PERSISTENCE_PRIMARY_NAMESPACE,
-			SCORER_PERSISTENCE_SECONDARY_NAMESPACE,
-			SCORER_PERSISTENCE_KEY,
-		)
-		.await?,
-	);
+	let data = KVStore::read(
+		&*kv_store,
+		SCORER_PERSISTENCE_PRIMARY_NAMESPACE,
+		SCORER_PERSISTENCE_SECONDARY_NAMESPACE,
+		SCORER_PERSISTENCE_KEY,
+	)
+	.await?;
+	log_info!(logger, "[SCORER-LDK] read_scorer() - Read {} bytes from VSS store", data.len());
+	let mut reader = Cursor::new(data);
 	let args = (params, network_graph, logger.clone());
-	ProbabilisticScorer::read(&mut reader, args).map_err(|e| {
-		log_error!(logger, "Failed to deserialize scorer: {}", e);
+	let scorer = ProbabilisticScorer::read(&mut reader, args).map_err(|e| {
+		log_error!(logger, "[SCORER-LDK] read_scorer() - Failed to deserialize scorer: {}", e);
 		std::io::Error::new(std::io::ErrorKind::InvalidData, "Failed to deserialize Scorer")
-	})
+	})?;
+	log_info!(logger, "[SCORER-LDK] read_scorer() - Successfully deserialized scorer from VSS");
+	Ok(scorer)
 }
 
 /// Read previously persisted external pathfinding scores from the cache.

src/lib.rs

@@ -133,7 +133,7 @@ use gossip::GossipSource;
 use graph::NetworkGraph;
 pub use io::utils::generate_entropy_mnemonic;
 use io::utils::write_node_metrics;
-use lightning::chain::BestBlock;
+use lightning::chain::{BestBlock, Confirm};
 use lightning::events::bump_transaction::{Input, Wallet as LdkWallet};
 use lightning::impl_writeable_tlv_based;
 use lightning::ln::chan_utils::{make_funding_redeemscript, FUNDING_TRANSACTION_WITNESS_WEIGHT};
@@ -1505,6 +1505,20 @@ impl Node {
 		})
 	}
 
+	/// Fetch the latest block header and update the ChannelManager's `highest_seen_timestamp`.
+	pub fn update_chain_tip(&self) -> Result<(), Error> {
+		let chain_source = Arc::clone(&self.chain_source);
+		let channel_manager = Arc::clone(&self.channel_manager);
+		let logger = Arc::clone(&self.logger);
+
+		self.runtime.block_on(async move {
+			let (header, height) = chain_source.fetch_chain_tip().await?;
+			channel_manager.best_block_updated(&header, height);
+			log_info!(logger, "Chain tip updated to height {}", height);
+			Ok(())
+		})
+	}
+
 	/// Manually sync the RGS snapshot.
 	///
 	/// If `do_full_sync` is true, the RGS snapshot will be updated from scratch. Otherwise, the