Separate sweeper persistent state

Prepare for adding runtime state.

Author: joostjager

lightning/src/util/sweep.rs

@@ -382,7 +382,8 @@ where
 		output_spender: O, change_destination_source: D, kv_store: K, logger: L,
 	) -> Self {
 		let outputs = Vec::new();
-		let sweeper_state = Mutex::new(SweeperState { outputs, best_block });
+		let sweeper_state =
+			Mutex::new(SweeperState { persistent: PersistentSweeperState { outputs, best_block } });
 		Self {
 			sweeper_state,
 			pending_sweep: AtomicBool::new(false),
@@ -437,27 +438,32 @@ where
 				},
 			};
 
-			if state_lock.outputs.iter().find(|o| o.descriptor == output_info.descriptor).is_some()
+			if state_lock
+				.persistent
+				.outputs
+				.iter()
+				.find(|o| o.descriptor == output_info.descriptor)
+				.is_some()
 			{
 				continue;
 			}
 
-			state_lock.outputs.push(output_info);
+			state_lock.persistent.outputs.push(output_info);
 		}
-		self.persist_state(&*state_lock).map_err(|e| {
+		self.persist_state(&state_lock).map_err(|e| {
 			log_error!(self.logger, "Error persisting OutputSweeper: {:?}", e);
 		})
 	}
 
 	/// Returns a list of the currently tracked spendable outputs.
 	pub fn tracked_spendable_outputs(&self) -> Vec<TrackedSpendableOutput> {
-		self.sweeper_state.lock().unwrap().outputs.clone()
+		self.sweeper_state.lock().unwrap().persistent.outputs.clone()
 	}
 
 	/// Gets the latest best block which was connected either via the [`Listen`] or
 	/// [`Confirm`] interfaces.
 	pub fn current_best_block(&self) -> BestBlock {
-		self.sweeper_state.lock().unwrap().best_block
+		self.sweeper_state.lock().unwrap().persistent.best_block
 	}
 
 	/// Regenerates and broadcasts the spending transaction for any outputs that are pending. This method will be a
@@ -503,7 +509,7 @@ where
 
 		// See if there is anything to sweep before requesting a change address.
 		{
-			let sweeper_state = self.sweeper_state.lock().unwrap();
+			let sweeper_state = &self.sweeper_state.lock().unwrap().persistent;
 
 			let cur_height = sweeper_state.best_block.height;
 			let has_respends = sweeper_state.outputs.iter().any(|o| filter_fn(o, cur_height));
@@ -518,12 +524,13 @@ where
 
 		// Sweep the outputs.
 		{
-			let mut sweeper_state = self.sweeper_state.lock().unwrap();
+			let mut sweeper_state = &mut self.sweeper_state.lock().unwrap();
 
-			let cur_height = sweeper_state.best_block.height;
-			let cur_hash = sweeper_state.best_block.block_hash;
+			let cur_height = sweeper_state.persistent.best_block.height;
+			let cur_hash = sweeper_state.persistent.best_block.block_hash;
 
 			let respend_descriptors: Vec<&SpendableOutputDescriptor> = sweeper_state
+				.persistent
 				.outputs
 				.iter()
 				.filter(|o| filter_fn(*o, cur_height))
@@ -536,7 +543,11 @@ where
 			}
 
 			let spending_tx = self
-				.spend_outputs(&sweeper_state, &respend_descriptors, change_destination_script)
+				.spend_outputs(
+					&sweeper_state.persistent,
+					&respend_descriptors,
+					change_destination_script,
+				)
 				.map_err(|e| {
 					log_error!(self.logger, "Error spending outputs: {:?}", e);
 				})?;
@@ -550,7 +561,7 @@ where
 			// As we didn't modify the state so far, the same filter_fn yields the same elements as
 			// above.
 			let respend_outputs =
-				sweeper_state.outputs.iter_mut().filter(|o| filter_fn(&**o, cur_height));
+				sweeper_state.persistent.outputs.iter_mut().filter(|o| filter_fn(&**o, cur_height));
 			for output_info in respend_outputs {
 				if let Some(filter) = self.chain_data_source.as_ref() {
 					let watched_output = output_info.to_watched_output(cur_hash);
@@ -571,10 +582,10 @@ where
 	}
 
 	fn prune_confirmed_outputs(&self, sweeper_state: &mut SweeperState) {
-		let cur_height = sweeper_state.best_block.height;
+		let cur_height = sweeper_state.persistent.best_block.height;
 
 		// Prune all outputs that have sufficient depth by now.
-		sweeper_state.outputs.retain(|o| {
+		sweeper_state.persistent.outputs.retain(|o| {
 			if let Some(confirmation_height) = o.status.confirmation_height() {
 				// We wait at least `PRUNE_DELAY_BLOCKS` as before that
 				// `Event::SpendableOutputs` from lingering monitors might get replayed.
@@ -596,7 +607,7 @@ where
 				OUTPUT_SWEEPER_PERSISTENCE_PRIMARY_NAMESPACE,
 				OUTPUT_SWEEPER_PERSISTENCE_SECONDARY_NAMESPACE,
 				OUTPUT_SWEEPER_PERSISTENCE_KEY,
-				&sweeper_state.encode(),
+				&sweeper_state.persistent.encode(),
 			)
 			.map_err(|e| {
 				log_error!(
@@ -612,7 +623,7 @@ where
 	}
 
 	fn spend_outputs(
-		&self, sweeper_state: &SweeperState, descriptors: &[&SpendableOutputDescriptor],
+		&self, sweeper_state: &PersistentSweeperState, descriptors: &[&SpendableOutputDescriptor],
 		change_destination_script: ScriptBuf,
 	) -> Result<Transaction, ()> {
 		let tx_feerate =
@@ -635,7 +646,7 @@ where
 	) {
 		let confirmation_hash = header.block_hash();
 		for (_, tx) in txdata {
-			for output_info in sweeper_state.outputs.iter_mut() {
+			for output_info in sweeper_state.persistent.outputs.iter_mut() {
 				if output_info.is_spent_in(*tx) {
 					output_info.status.confirmed(confirmation_hash, height, (*tx).clone())
 				}
@@ -646,7 +657,7 @@ where
 	fn best_block_updated_internal(
 		&self, sweeper_state: &mut SweeperState, header: &Header, height: u32,
 	) {
-		sweeper_state.best_block = BestBlock::new(header.block_hash(), height);
+		sweeper_state.persistent.best_block = BestBlock::new(header.block_hash(), height);
 		self.prune_confirmed_outputs(sweeper_state);
 	}
 }
@@ -666,9 +677,9 @@ where
 		&self, header: &Header, txdata: &chain::transaction::TransactionData, height: u32,
 	) {
 		let mut state_lock = self.sweeper_state.lock().unwrap();
-		assert_eq!(state_lock.best_block.block_hash, header.prev_blockhash,
+		assert_eq!(state_lock.persistent.best_block.block_hash, header.prev_blockhash,
 			"Blocks must be connected in chain-order - the connected header must build on the last connected header");
-		assert_eq!(state_lock.best_block.height, height - 1,
+		assert_eq!(state_lock.persistent.best_block.height, height - 1,
 			"Blocks must be connected in chain-order - the connected block height must be one greater than the previous height");
 
 		self.transactions_confirmed_internal(&mut *state_lock, header, txdata, height);
@@ -685,13 +696,13 @@ where
 		let new_height = height - 1;
 		let block_hash = header.block_hash();
 
-		assert_eq!(state_lock.best_block.block_hash, block_hash,
+		assert_eq!(state_lock.persistent.best_block.block_hash, block_hash,
 		"Blocks must be disconnected in chain-order - the disconnected header must be the last connected header");
-		assert_eq!(state_lock.best_block.height, height,
+		assert_eq!(state_lock.persistent.best_block.height, height,
 			"Blocks must be disconnected in chain-order - the disconnected block must have the correct height");
-		state_lock.best_block = BestBlock::new(header.prev_blockhash, new_height);
+		state_lock.persistent.best_block = BestBlock::new(header.prev_blockhash, new_height);
 
-		for output_info in state_lock.outputs.iter_mut() {
+		for output_info in state_lock.persistent.outputs.iter_mut() {
 			if output_info.status.confirmation_hash() == Some(block_hash) {
 				debug_assert_eq!(output_info.status.confirmation_height(), Some(height));
 				output_info.status.unconfirmed();
@@ -730,6 +741,7 @@ where
 
 		// Get what height was unconfirmed.
 		let unconf_height = state_lock
+			.persistent
 			.outputs
 			.iter()
 			.find(|o| o.status.latest_spending_tx().map(|tx| tx.compute_txid()) == Some(*txid))
@@ -738,6 +750,7 @@ where
 		if let Some(unconf_height) = unconf_height {
 			// Unconfirm all >= this height.
 			state_lock
+				.persistent
 				.outputs
 				.iter_mut()
 				.filter(|o| o.status.confirmation_height() >= Some(unconf_height))
@@ -760,6 +773,7 @@ where
 	fn get_relevant_txids(&self) -> Vec<(Txid, u32, Option<BlockHash>)> {
 		let state_lock = self.sweeper_state.lock().unwrap();
 		state_lock
+			.persistent
 			.outputs
 			.iter()
 			.filter_map(|o| match o.status {
@@ -779,13 +793,18 @@ where
 	}
 }
 
-#[derive(Debug, Clone)]
+#[derive(Debug)]
 struct SweeperState {
+	persistent: PersistentSweeperState,
+}
+
+#[derive(Debug, Clone)]
+struct PersistentSweeperState {
 	outputs: Vec<TrackedSpendableOutput>,
 	best_block: BestBlock,
 }
 
-impl_writeable_tlv_based!(SweeperState, {
+impl_writeable_tlv_based!(PersistentSweeperState, {
 	(0, outputs, required_vec),
 	(2, best_block, required),
 });
@@ -831,7 +850,7 @@ where
 			kv_store,
 			logger,
 		) = args;
-		let state = SweeperState::read(reader)?;
+		let state = PersistentSweeperState::read(reader)?;
 		let best_block = state.best_block;
 
 		if let Some(filter) = chain_data_source.as_ref() {
@@ -841,7 +860,7 @@ where
 			}
 		}
 
-		let sweeper_state = Mutex::new(state);
+		let sweeper_state = Mutex::new(SweeperState { persistent: state });
 		Ok(Self {
 			sweeper_state,
 			pending_sweep: AtomicBool::new(false),
@@ -880,7 +899,7 @@ where
 			kv_store,
 			logger,
 		) = args;
-		let state = SweeperState::read(reader)?;
+		let state = PersistentSweeperState::read(reader)?;
 		let best_block = state.best_block;
 
 		if let Some(filter) = chain_data_source.as_ref() {
@@ -890,7 +909,7 @@ where
 			}
 		}
 
-		let sweeper_state = Mutex::new(state);
+		let sweeper_state = Mutex::new(SweeperState { persistent: state });
 		Ok((
 			best_block,
 			OutputSweeper {