add threshold for persister

Author: Prabhat1308

lightning/src/util/persist.rs

@@ -446,6 +446,13 @@ where
 /// If you have many stale updates stored (such as after a crash with pending lazy deletes), and
 /// would like to get rid of them, consider using the
 /// [`MonitorUpdatingPersister::cleanup_stale_updates`] function.
+///
+/// # Size-based persistence optimization
+///
+/// For small channel monitors (below `minimum_monitor_size_for_updates` bytes when serialized),
+/// this persister will always write the full monitor instead of individual updates. This avoids
+/// the overhead of managing update files and later compaction for tiny monitors that don't benefit
+/// from differential updates.
 pub struct MonitorUpdatingPersister<K: Deref, L: Deref, ES: Deref, SP: Deref, BI: Deref, FE: Deref>
 where
 	K::Target: KVStore,
@@ -458,6 +465,7 @@ where
 	kv_store: K,
 	logger: L,
 	maximum_pending_updates: u64,
+	minimum_monitor_size_for_updates: usize,
 	entropy_source: ES,
 	signer_provider: SP,
 	broadcaster: BI,
@@ -491,21 +499,45 @@ where
 	/// less frequent "waves."
 	///   - [`MonitorUpdatingPersister`] will potentially have more listing to do if you need to run
 	/// [`MonitorUpdatingPersister::cleanup_stale_updates`].
+	///
+	/// The `minimum_monitor_size_for_updates` parameter sets the minimum serialized size (in bytes)
+	/// for a [`ChannelMonitor`] to use update-based persistence. Monitors smaller than this threshold
+	/// will always be persisted in full, avoiding the overhead of managing update files for tiny
+	/// monitors. A reasonable default is 4096 bytes (4 KiB). Set to 0 to always use update-based
+	/// persistence regardless of size.
 	pub fn new(
-		kv_store: K, logger: L, maximum_pending_updates: u64, entropy_source: ES,
+		kv_store: K, logger: L, maximum_pending_updates: u64, minimum_monitor_size_for_updates: usize, entropy_source: ES,
 		signer_provider: SP, broadcaster: BI, fee_estimator: FE,
 	) -> Self {
 		MonitorUpdatingPersister {
 			kv_store,
 			logger,
 			maximum_pending_updates,
+			minimum_monitor_size_for_updates,
 			entropy_source,
 			signer_provider,
 			broadcaster,
 			fee_estimator,
 		}
 	}
 
+	/// Constructs a new [`MonitorUpdatingPersister`] with a default minimum monitor size threshold.
+	///
+	/// This is a convenience method that sets `minimum_monitor_size_for_updates` to 4096 bytes (4 KiB),
+	/// which is a reasonable default for most use cases. Monitors smaller than this will be persisted
+	/// in full rather than using update-based persistence.
+	///
+	/// For other parameters, see [`MonitorUpdatingPersister::new`].
+	pub fn new_with_default_threshold(
+		kv_store: K, logger: L, maximum_pending_updates: u64, entropy_source: ES,
+		signer_provider: SP, broadcaster: BI, fee_estimator: FE,
+	) -> Self {
+		Self::new(
+			kv_store, logger, maximum_pending_updates, 4096, entropy_source,
+			signer_provider, broadcaster, fee_estimator,
+		)
+	}
+
 	/// Reads all stored channel monitors, along with any stored updates for them.
 	///
 	/// It is extremely important that your [`KVStore::read`] implementation uses the
@@ -752,7 +784,12 @@ where
 	) -> chain::ChannelMonitorUpdateStatus {
 		const LEGACY_CLOSED_CHANNEL_UPDATE_ID: u64 = u64::MAX;
 		if let Some(update) = update {
-			let persist_update = update.update_id != LEGACY_CLOSED_CHANNEL_UPDATE_ID
+			// Check if monitor is too small for update-based persistence
+			let monitor_size = monitor.serialized_length();
+			let use_full_persistence = monitor_size < self.minimum_monitor_size_for_updates;
+			
+			let persist_update = !use_full_persistence
+				&& update.update_id != LEGACY_CLOSED_CHANNEL_UPDATE_ID
 				&& update.update_id % self.maximum_pending_updates != 0;
 			if persist_update {
 				let monitor_key = monitor_name.to_string();
@@ -1156,6 +1193,7 @@ mod tests {
 			kv_store: &TestStore::new(false),
 			logger: &TestLogger::new(),
 			maximum_pending_updates: persister_0_max_pending_updates,
+			minimum_monitor_size_for_updates: 0,
 			entropy_source: &chanmon_cfgs[0].keys_manager,
 			signer_provider: &chanmon_cfgs[0].keys_manager,
 			broadcaster: &chanmon_cfgs[0].tx_broadcaster,
@@ -1165,6 +1203,7 @@ mod tests {
 			kv_store: &TestStore::new(false),
 			logger: &TestLogger::new(),
 			maximum_pending_updates: persister_1_max_pending_updates,
+			minimum_monitor_size_for_updates: 0,
 			entropy_source: &chanmon_cfgs[1].keys_manager,
 			signer_provider: &chanmon_cfgs[1].keys_manager,
 			broadcaster: &chanmon_cfgs[1].tx_broadcaster,
@@ -1330,6 +1369,7 @@ mod tests {
 				kv_store: &TestStore::new(true),
 				logger: &TestLogger::new(),
 				maximum_pending_updates: 11,
+				minimum_monitor_size_for_updates: 0,
 				entropy_source: node_cfgs[0].keys_manager,
 				signer_provider: node_cfgs[0].keys_manager,
 				broadcaster: node_cfgs[0].tx_broadcaster,
@@ -1376,6 +1416,7 @@ mod tests {
 			kv_store: &TestStore::new(false),
 			logger: &TestLogger::new(),
 			maximum_pending_updates: test_max_pending_updates,
+			minimum_monitor_size_for_updates: 0,
 			entropy_source: &chanmon_cfgs[0].keys_manager,
 			signer_provider: &chanmon_cfgs[0].keys_manager,
 			broadcaster: &chanmon_cfgs[0].tx_broadcaster,
@@ -1385,6 +1426,7 @@ mod tests {
 			kv_store: &TestStore::new(false),
 			logger: &TestLogger::new(),
 			maximum_pending_updates: test_max_pending_updates,
+			minimum_monitor_size_for_updates: 0,
 			entropy_source: &chanmon_cfgs[1].keys_manager,
 			signer_provider: &chanmon_cfgs[1].keys_manager,
 			broadcaster: &chanmon_cfgs[1].tx_broadcaster,