Start storing payments using PaginatedKVStore.

Author: G8XSU

ldk-server/src/io/mod.rs

@@ -5,3 +5,7 @@ pub(crate) mod utils;
 /// The forwarded payments will be persisted under this prefix.
 pub(crate) const FORWARDED_PAYMENTS_PERSISTENCE_PRIMARY_NAMESPACE: &str = "forwarded_payments";
 pub(crate) const FORWARDED_PAYMENTS_PERSISTENCE_SECONDARY_NAMESPACE: &str = "";
+
+/// The payments will be persisted under this prefix.
+pub(crate) const PAYMENTS_PERSISTENCE_PRIMARY_NAMESPACE: &str = "payments";
+pub(crate) const PAYMENTS_PERSISTENCE_SECONDARY_NAMESPACE: &str = "";

ldk-server/src/main.rs

@@ -17,12 +17,14 @@ use crate::io::paginated_kv_store::PaginatedKVStore;
 use crate::io::sqlite_store::SqliteStore;
 use crate::io::{
 	FORWARDED_PAYMENTS_PERSISTENCE_PRIMARY_NAMESPACE,
-	FORWARDED_PAYMENTS_PERSISTENCE_SECONDARY_NAMESPACE,
+	FORWARDED_PAYMENTS_PERSISTENCE_SECONDARY_NAMESPACE, PAYMENTS_PERSISTENCE_PRIMARY_NAMESPACE,
+	PAYMENTS_PERSISTENCE_SECONDARY_NAMESPACE,
 };
 use crate::util::config::load_config;
-use crate::util::proto_adapter::forwarded_payment_to_proto;
+use crate::util::proto_adapter::{forwarded_payment_to_proto, payment_to_proto};
 use hex::DisplayHex;
 use ldk_node::config::Config;
+use ldk_node::lightning::util::persist::KVStore;
 use ldk_node::logger::LogLevel;
 use prost::Message;
 use rand::Rng;
@@ -33,6 +35,38 @@ use std::time::{SystemTime, UNIX_EPOCH};
 
 const USAGE_GUIDE: &str = "Usage: ldk-server <config_path>";
 
+macro_rules! upsert_payment_details {
+	($event_node:expr, $paginated_store:expr, $payment_id:expr) => {{
+		if let Some(payment_details) = $event_node.payment($payment_id) {
+			let payment = payment_to_proto(payment_details);
+			let time = SystemTime::now()
+				.duration_since(UNIX_EPOCH)
+				.expect("Time must be > 1970")
+				.as_secs() as i64;
+
+			match $paginated_store.write(
+				PAYMENTS_PERSISTENCE_PRIMARY_NAMESPACE,
+				PAYMENTS_PERSISTENCE_SECONDARY_NAMESPACE,
+				&$payment_id.0.to_lower_hex_string(),
+				time,
+				&payment.encode_to_vec(),
+			) {
+				Ok(_) => {
+					$event_node.event_handled();
+				},
+				Err(e) => {
+					eprintln!("Failed to write payment to persistence: {}", e);
+				},
+			}
+		} else {
+			eprintln!(
+				"Unable to find payment with paymentId: {}",
+				$payment_id.0.to_lower_hex_string()
+			);
+		}
+	}};
+}
+
 fn main() {
 	let args: Vec<String> = std::env::args().collect();
 
@@ -146,7 +180,19 @@ fn main() {
 								"PAYMENT_RECEIVED: with id {:?}, hash {}, amount_msat {}",
 								payment_id, payment_hash, amount_msat
 							);
-							event_node.event_handled();
+							let payment_id = payment_id.expect("PaymentId expected for ldk-server >=0.1");
+							upsert_payment_details!(event_node, &paginated_store, &payment_id);
+						},
+						Event::PaymentSuccessful {payment_id, ..} => {
+							let payment_id = payment_id.expect("PaymentId expected for ldk-server >=0.1");
+							upsert_payment_details!(event_node, &paginated_store, &payment_id);
+						},
+						Event::PaymentFailed {payment_id, ..} => {
+							let payment_id = payment_id.expect("PaymentId expected for ldk-server >=0.1");
+							upsert_payment_details!(event_node, &paginated_store, &payment_id);
+						},
+						Event::PaymentClaimable {payment_id, ..} => {
+							upsert_payment_details!(event_node, &paginated_store, &payment_id);
 						},
 						Event::PaymentForwarded {
 							prev_channel_id,