Add seeded startup operations benchmark

Add a startup benchmark that restarts a node whose store already contains
channel and payment data, so startup cost reflects persisted node state.

AI-assisted-by: OpenAI Codex

Author: benthecarman

benches/operations.rs

@@ -11,20 +11,67 @@ mod common;
 use std::sync::Arc;
 use std::time::{Duration, Instant};
 
+use bitcoin::secp256k1::PublicKey;
 use bitcoin::Amount;
 use common::{
 	expect_channel_pending_event, expect_channel_ready_event, expect_event,
 	generate_blocks_and_wait, premine_and_distribute_funds, random_chain_source, random_config,
-	setup_bitcoind_and_electrsd, setup_node, setup_two_nodes_with_store,
+	random_storage_path, setup_bitcoind_and_electrsd, setup_node, setup_two_nodes_with_store,
 };
 use criterion::{criterion_group, criterion_main, Criterion};
 use electrsd::corepc_node::{Client as BitcoindClient, Node as BitcoinD};
+use ldk_node::io::sqlite_store::SqliteStore;
 use ldk_node::{Event, Node};
 use lightning::ln::channelmanager::PaymentId;
 use lightning::routing::router::RouteParametersConfig;
+use lightning::util::persist::migrate_kv_store_data_async;
 use lightning_invoice::{Bolt11InvoiceDescription, Description};
+use lightning_persister::fs_store::v2::FilesystemStoreV2;
+
+use crate::common::{open_channel_push_amt, TestChainSource, TestConfig, TestStoreType};
+
+#[cfg(feature = "postgres")]
+use ldk_node::io::postgres_store::{PostgresStore, POSTGRES_TEST_URL_ENV_VAR};
+
+const STARTUP_SEED_SCENARIOS: [StartupSeedScenario; 6] = [
+	StartupSeedScenario { channel_count: 1, payment_count: 2 },
+	StartupSeedScenario { channel_count: 1, payment_count: 100 },
+	StartupSeedScenario { channel_count: 1, payment_count: 1_000 },
+	StartupSeedScenario { channel_count: 10, payment_count: 2 },
+	StartupSeedScenario { channel_count: 100, payment_count: 2 },
+	StartupSeedScenario { channel_count: 100, payment_count: 1_000 },
+];
+const STARTUP_SEED_PAYMENT_AMOUNT_MSAT: u64 = 1_000_000;
+const STARTUP_SEED_MIN_CHANNEL_FUNDING_SAT: u64 = 100_000;
+const STARTUP_SEED_CHANNEL_BUFFER_SAT: u64 = 1_000_000;
+const STARTUP_SEED_CHANNEL_BATCH_SIZE: u64 = 2;
 
-use crate::common::{open_channel_push_amt, TestChainSource, TestStoreType};
+#[derive(Clone, Copy)]
+struct StartupSeedScenario {
+	channel_count: u64,
+	payment_count: u64,
+}
+
+impl StartupSeedScenario {
+	fn bench_name(self, store_name: &str) -> String {
+		format!("{}/channels_{}_payments_{}", store_name, self.channel_count, self.payment_count)
+	}
+
+	fn runs_in_ci(self) -> bool {
+		self.channel_count == 1 && self.payment_count == 2
+	}
+
+	fn channel_funding_sat(self) -> u64 {
+		let payment_amount_sat = STARTUP_SEED_PAYMENT_AMOUNT_MSAT / 1_000;
+		let payment_funding_sat =
+			self.payment_count * payment_amount_sat + STARTUP_SEED_CHANNEL_BUFFER_SAT;
+		payment_funding_sat.max(STARTUP_SEED_MIN_CHANNEL_FUNDING_SAT)
+	}
+
+	fn premine_amount_sat(self) -> u64 {
+		self.channel_count * self.channel_funding_sat() + STARTUP_SEED_CHANNEL_BUFFER_SAT
+	}
+}
 
 #[derive(Clone, Copy)]
 struct StoreBenchConfig {
@@ -37,6 +84,7 @@ fn operations_benchmark(c: &mut Criterion) {
 
 	forwarding_benchmark(c);
 	channel_open_benchmark(c);
+	startup_benchmark(c);
 }
 
 fn forwarding_benchmark(c: &mut Criterion) {
@@ -133,6 +181,214 @@ fn channel_open_benchmark(c: &mut Criterion) {
 	}
 }
 
+fn startup_benchmark(c: &mut Criterion) {
+	let (bitcoind, electrsd) = setup_bitcoind_and_electrsd();
+	let chain_source = random_chain_source(&bitcoind, &electrsd);
+	let runtime =
+		tokio::runtime::Builder::new_multi_thread().worker_threads(4).enable_all().build().unwrap();
+
+	let mut group = c.benchmark_group("startup");
+	group.sample_size(10);
+
+	for startup_seed_scenario in STARTUP_SEED_SCENARIOS {
+		if is_ci() && !startup_seed_scenario.runs_in_ci() {
+			continue;
+		}
+
+		let matching_store_configs: Vec<_> = store_bench_configs()
+			.into_iter()
+			.filter(|store_config| {
+				let bench_name = startup_seed_scenario.bench_name(store_config.name);
+				should_register_bench("startup", &bench_name)
+			})
+			.collect();
+		if matching_store_configs.is_empty() {
+			continue;
+		}
+
+		let seeded_config = setup_startup_seed_node(
+			&chain_source,
+			&bitcoind,
+			&electrsd,
+			startup_seed_scenario,
+			&runtime,
+		);
+		let startup_configs = migrate_startup_seed_configs(
+			&seeded_config,
+			startup_seed_scenario,
+			matching_store_configs,
+			&runtime,
+		);
+
+		for (bench_name, config) in startup_configs {
+			group.bench_function(bench_name, |b| {
+				b.iter_custom(|iter| {
+					let mut total = Duration::ZERO;
+					for _ in 0..iter {
+						let start = Instant::now();
+						let node = setup_node(&chain_source, config.clone());
+						total += start.elapsed();
+						node.stop().unwrap();
+					}
+					total
+				});
+			});
+		}
+	}
+}
+
+fn setup_startup_seed_node(
+	chain_source: &TestChainSource, bitcoind: &BitcoinD, electrsd: &electrsd::ElectrsD,
+	seed_scenario: StartupSeedScenario, runtime: &tokio::runtime::Runtime,
+) -> TestConfig {
+	let mut config_a = random_config(true);
+	config_a.store_type = TestStoreType::Sqlite;
+	let node_a = Arc::new(setup_node(chain_source, config_a.clone()));
+
+	let mut config_b = random_config(true);
+	config_b.store_type = TestStoreType::Sqlite;
+	let node_b = Arc::new(setup_node(chain_source, config_b));
+
+	runtime.block_on(async {
+		let address_a = node_a.onchain_payment().new_address().unwrap();
+		premine_and_distribute_funds(
+			&bitcoind.client,
+			&electrsd.client,
+			vec![address_a],
+			Amount::from_sat(seed_scenario.premine_amount_sat()),
+		)
+		.await;
+		node_a.sync_wallets().unwrap();
+		node_b.sync_wallets().unwrap();
+
+		let funding_amount_sat = seed_scenario.channel_funding_sat();
+		let mut remaining_channel_count = seed_scenario.channel_count;
+		while remaining_channel_count > 0 {
+			let channel_batch_size = remaining_channel_count.min(STARTUP_SEED_CHANNEL_BATCH_SIZE);
+			for _ in 0..channel_batch_size {
+				node_a
+					.open_channel(
+						node_b.node_id(),
+						node_b.listening_addresses().unwrap().first().unwrap().clone(),
+						funding_amount_sat,
+						None,
+						None,
+					)
+					.unwrap();
+				assert!(node_a.list_peers().iter().any(|peer| peer.node_id == node_b.node_id()));
+
+				let funding_txo_a = expect_channel_pending_event!(node_a, node_b.node_id());
+				let funding_txo_b = expect_channel_pending_event!(node_b, node_a.node_id());
+				assert_eq!(funding_txo_a, funding_txo_b);
+				node_a.sync_wallets().unwrap();
+			}
+			generate_blocks_and_wait(&bitcoind.client, &electrsd.client, 6).await;
+
+			for _ in 0..channel_batch_size {
+				node_a.sync_wallets().unwrap();
+				node_b.sync_wallets().unwrap();
+				wait_for_channel_ready_events(&node_a, node_b.node_id(), 1).await;
+				wait_for_channel_ready_events(&node_b, node_a.node_id(), 1).await;
+			}
+			remaining_channel_count -= channel_batch_size;
+		}
+
+		for idx in 0..seed_scenario.payment_count {
+			let invoice_description = Bolt11InvoiceDescription::Direct(
+				Description::new(format!("startup seed {}", idx + 1)).unwrap(),
+			);
+			let invoice = node_b
+				.bolt11_payment()
+				.receive(STARTUP_SEED_PAYMENT_AMOUNT_MSAT, &invoice_description.into(), 9217)
+				.unwrap();
+			let payment_id = node_a.bolt11_payment().send(&invoice, None).unwrap();
+			wait_for_payment_success(&node_a, payment_id).await;
+		}
+
+		drain_events(&node_a);
+		drain_events(&node_b);
+	});
+
+	node_a.stop().unwrap();
+	node_b.stop().unwrap();
+
+	config_a
+}
+
+fn migrate_startup_seed_configs(
+	source_config: &TestConfig, seed_scenario: StartupSeedScenario,
+	store_configs: Vec<StoreBenchConfig>, runtime: &tokio::runtime::Runtime,
+) -> Vec<(String, TestConfig)> {
+	let source_store =
+		SqliteStore::new(source_config.node_config.storage_dir_path.clone().into(), None, None)
+			.unwrap();
+
+	store_configs
+		.into_iter()
+		.map(|store_config| {
+			let mut config = source_config.clone();
+			config.store_type = store_config.store_type;
+			if !matches!(store_config.store_type, TestStoreType::Sqlite) {
+				config.node_config.storage_dir_path =
+					random_storage_path().to_str().unwrap().to_owned();
+				migrate_startup_seed_store(&source_store, &config, runtime);
+			}
+
+			(seed_scenario.bench_name(store_config.name), config)
+		})
+		.collect()
+}
+
+fn migrate_startup_seed_store(
+	source_store: &SqliteStore, destination_config: &TestConfig, runtime: &tokio::runtime::Runtime,
+) {
+	runtime.block_on(async {
+		match destination_config.store_type {
+			TestStoreType::Sqlite => {},
+			TestStoreType::FilesystemStore => {
+				let destination_store = FilesystemStoreV2::new(
+					destination_config.node_config.storage_dir_path.clone().into(),
+				)
+				.unwrap();
+				migrate_kv_store_data_async(source_store, &destination_store).await.unwrap();
+			},
+			#[cfg(feature = "postgres")]
+			TestStoreType::Postgres => {
+				let connection_string = postgres_connection_string();
+				let table_name = postgres_table_name(destination_config);
+				let destination_store =
+					PostgresStore::new(connection_string, None, Some(table_name), None)
+						.await
+						.unwrap();
+				migrate_kv_store_data_async(source_store, &destination_store).await.unwrap();
+			},
+			TestStoreType::TestSyncStore => {
+				unreachable!("startup benches do not use TestSyncStore")
+			},
+		}
+	});
+}
+
+#[cfg(feature = "postgres")]
+fn postgres_connection_string() -> String {
+	dotenvy::dotenv().ok();
+	std::env::var(POSTGRES_TEST_URL_ENV_VAR)
+		.unwrap_or_else(|_| "host=localhost user=postgres password=postgres".to_string())
+}
+
+#[cfg(feature = "postgres")]
+fn postgres_table_name(config: &TestConfig) -> String {
+	format!(
+		"test_{}",
+		config
+			.node_config
+			.storage_dir_path
+			.chars()
+			.filter(|c| c.is_ascii_alphanumeric())
+			.collect::<String>()
+	)
+}
+
 fn should_register_bench(group: &str, name: &str) -> bool {
 	let target = format!("{}/{}", group, name);
 	let filters: Vec<String> =
@@ -143,6 +399,10 @@ fn should_register_bench(group: &str, name: &str) -> bool {
 		})
 }
 
+fn is_ci() -> bool {
+	std::env::var("CI").is_ok_and(|value| !value.is_empty() && value != "0" && value != "false")
+}
+
 fn setup_forwarding_nodes(
 	chain_source: &TestChainSource, bitcoind: &BitcoinD, electrsd: &electrsd::ElectrsD,
 	store_type: TestStoreType, runtime: &tokio::runtime::Runtime,
@@ -408,6 +668,34 @@ async fn wait_for_payment_success(node: &Node, expected_payment_id: PaymentId) {
 	}
 }
 
+async fn wait_for_channel_ready_events(node: &Node, counterparty_node_id: PublicKey, count: u64) {
+	let mut remaining_count = count;
+	while remaining_count > 0 {
+		let event = tokio::time::timeout(
+			Duration::from_secs(common::INTEROP_TIMEOUT_SECS),
+			node.next_event_async(),
+		)
+		.await
+		.unwrap_or_else(|_| {
+			panic!("{} timed out waiting for ChannelReady event after 60s", node.node_id())
+		});
+
+		match event {
+			ref e @ Event::ChannelReady { counterparty_node_id: Some(node_id), .. }
+				if node_id == counterparty_node_id =>
+			{
+				println!("{} got event {:?}", node.node_id(), e);
+				remaining_count -= 1;
+			},
+			ref e @ Event::ChannelReady { .. } => {
+				panic!("{} got unexpected ChannelReady event: {:?}", node.node_id(), e);
+			},
+			_ => {},
+		}
+		node.event_handled().unwrap();
+	}
+}
+
 fn drain_events(node: &Node) {
 	while node.next_event().is_some() {
 		node.event_handled().unwrap();