Add persisted payment metadata reading in store

Issue #425

Author: moisesPompilio

src/builder.rs

@@ -23,7 +23,7 @@ use crate::liquidity::{
 };
 use crate::logger::{log_error, log_info, LdkLogger, LogLevel, LogWriter, Logger};
 use crate::message_handler::NodeCustomMessageHandler;
-use crate::payment::store::PaymentStore;
+use crate::payment::store::{PaymentMetadataStore, PaymentStore};
 use crate::peer_store::PeerStore;
 use crate::tx_broadcaster::TransactionBroadcaster;
 use crate::types::{
@@ -1023,6 +1023,18 @@ fn build_with_store_internal(
 		},
 	};
 
+	let metadata_store = match io::utils::read_metadata(Arc::clone(&kv_store), Arc::clone(&logger))
+	{
+		Ok(metadata) => Arc::new(PaymentMetadataStore::new(
+			metadata,
+			Arc::clone(&kv_store),
+			Arc::clone(&logger),
+		)),
+		Err(_) => {
+			return Err(BuildError::ReadFailed);
+		},
+	};
+
 	let wallet = Arc::new(Wallet::new(
 		bdk_wallet,
 		wallet_persister,
@@ -1513,6 +1525,7 @@ fn build_with_store_internal(
 		scorer,
 		peer_store,
 		payment_store,
+		metadata_store,
 		is_listening,
 		node_metrics,
 	})

src/io/utils.rs

@@ -14,6 +14,7 @@ use crate::io::{
 	NODE_METRICS_KEY, NODE_METRICS_PRIMARY_NAMESPACE, NODE_METRICS_SECONDARY_NAMESPACE,
 };
 use crate::logger::{log_error, LdkLogger, Logger};
+use crate::payment::store::PaymentMetadata;
 use crate::peer_store::PeerStore;
 use crate::sweep::DeprecatedSpendableOutputInfo;
 use crate::types::{Broadcaster, DynStore, KeysManager, Sweeper};
@@ -232,6 +233,36 @@ where
 	Ok(res)
 }
 
+/// Read previously persisted payment metadata information from the store.
+pub(crate) fn read_metadata<L: Deref>(
+	kv_store: Arc<DynStore>, logger: L,
+) -> Result<Vec<PaymentMetadata>, std::io::Error>
+where
+	L::Target: LdkLogger,
+{
+	let mut res = Vec::new();
+
+	for stored_key in kv_store.list(
+		PAYMENT_INFO_PERSISTENCE_PRIMARY_NAMESPACE,
+		PAYMENT_METADATA_INFO_PERSISTENCE_SECONDARY_NAMESPACE,
+	)? {
+		let mut reader = Cursor::new(kv_store.read(
+			PAYMENT_INFO_PERSISTENCE_PRIMARY_NAMESPACE,
+			PAYMENT_METADATA_INFO_PERSISTENCE_SECONDARY_NAMESPACE,
+			&stored_key,
+		)?);
+		let payment = PaymentMetadata::read(&mut reader).map_err(|e| {
+			log_error!(logger, "Failed to deserialize PaymentMetadata: {}", e);
+			std::io::Error::new(
+				std::io::ErrorKind::InvalidData,
+				"Failed to deserialize PaymentMetadata",
+			)
+		})?;
+		res.push(payment);
+	}
+	Ok(res)
+}
+
 /// Read `OutputSweeper` state from the store.
 pub(crate) fn read_output_sweeper(
 	broadcaster: Arc<Broadcaster>, fee_estimator: Arc<OnchainFeeEstimator>,