fix: Even more refactoring to code and documentation

Thank you to @LLFourn and @danielabrozzoni for these suggestions.
This commit is contained in:
志宇 2023-08-01 18:27:24 +08:00
parent bea8e5aff4
commit b206a985cf
No known key found for this signature in database
GPG Key ID: F6345C9837C2BDE8
6 changed files with 44 additions and 45 deletions

View File

@ -500,7 +500,7 @@ impl<D> Wallet<D> {
// anchor tx to checkpoint with lowest height that is >= position's height // anchor tx to checkpoint with lowest height that is >= position's height
let anchor = self let anchor = self
.chain .chain
.heights() .blocks()
.range(height..) .range(height..)
.next() .next()
.ok_or(InsertTxError::ConfirmationHeightCannotBeGreaterThanTip { .ok_or(InsertTxError::ConfirmationHeightCannotBeGreaterThanTip {
@ -1697,13 +1697,12 @@ impl<D> Wallet<D> {
} }
/// Applies an update to the wallet and stages the changes (but does not [`commit`] them). /// Applies an update to the wallet and stages the changes (but does not [`commit`] them).
/// Returns whether the `update` resulted in any changes.
/// ///
/// Usually you create an `update` by interacting with some blockchain data source and inserting /// Usually you create an `update` by interacting with some blockchain data source and inserting
/// transactions related to your wallet into it. /// transactions related to your wallet into it.
/// ///
/// [`commit`]: Self::commit /// [`commit`]: Self::commit
pub fn apply_update(&mut self, update: Update) -> Result<bool, CannotConnectError> pub fn apply_update(&mut self, update: Update) -> Result<(), CannotConnectError>
where where
D: PersistBackend<ChangeSet>, D: PersistBackend<ChangeSet>,
{ {
@ -1717,9 +1716,8 @@ impl<D> Wallet<D> {
self.indexed_graph.apply_update(update.graph), self.indexed_graph.apply_update(update.graph),
)); ));
let changed = !changeset.is_empty();
self.persist.stage(changeset); self.persist.stage(changeset);
Ok(changed) Ok(())
} }
/// Commits all curently [`staged`] changed to the persistence backend returning and error when /// Commits all curently [`staged`] changed to the persistence backend returning and error when

View File

@ -8,6 +8,9 @@ use alloc::sync::Arc;
use bitcoin::BlockHash; use bitcoin::BlockHash;
/// A structure that represents changes to [`LocalChain`]. /// A structure that represents changes to [`LocalChain`].
///
/// The key represents the block height, and the value either represents added a new [`CheckPoint`]
/// (if [`Some`]), or removing a [`CheckPoint`] (if [`None`]).
pub type ChangeSet = BTreeMap<u32, Option<BlockHash>>; pub type ChangeSet = BTreeMap<u32, Option<BlockHash>>;
/// A [`LocalChain`] checkpoint is used to find the agreement point between two chains and as a /// A [`LocalChain`] checkpoint is used to find the agreement point between two chains and as a
@ -15,8 +18,9 @@ pub type ChangeSet = BTreeMap<u32, Option<BlockHash>>;
/// ///
/// Each checkpoint contains the height and hash of a block ([`BlockId`]). /// Each checkpoint contains the height and hash of a block ([`BlockId`]).
/// ///
/// Internaly, checkpoints are nodes of a linked-list. This allows the caller to view the entire /// Internaly, checkpoints are nodes of a reference-counted linked-list. This allows the caller to
/// chain without holding a lock to [`LocalChain`]. /// cheaply clone a [`CheckPoint`] without copying the whole list and to view the entire chain
/// without holding a lock on [`LocalChain`].
#[derive(Debug, Clone)] #[derive(Debug, Clone)]
pub struct CheckPoint(Arc<CPInner>); pub struct CheckPoint(Arc<CPInner>);
@ -54,10 +58,7 @@ impl CheckPoint {
/// ///
/// Returns an `Err(self)` if there is block which does not have a greater height than the /// Returns an `Err(self)` if there is block which does not have a greater height than the
/// previous one. /// previous one.
pub fn extend_with_blocks( pub fn extend(self, blocks: impl IntoIterator<Item = BlockId>) -> Result<Self, Self> {
self,
blocks: impl IntoIterator<Item = BlockId>,
) -> Result<Self, Self> {
let mut curr = self.clone(); let mut curr = self.clone();
for block in blocks { for block in blocks {
curr = curr.push(block).map_err(|_| self.clone())?; curr = curr.push(block).map_err(|_| self.clone())?;
@ -120,12 +121,15 @@ impl IntoIterator for CheckPoint {
/// A struct to update [`LocalChain`]. /// A struct to update [`LocalChain`].
/// ///
/// This is used as input for [`LocalChain::apply_update`]. It contains the update's chain `tip` and /// This is used as input for [`LocalChain::apply_update`]. It contains the update's chain `tip` and
/// a `bool` which signals whether this update can introduce blocks below the original chain's tip /// a flag `introduce_older_blocks` which signals whether this update intends to introduce missing
/// without invalidating blocks residing on the original chain. Block-by-block syncing mechanisms /// blocks to the original chain.
/// would typically create updates that builds upon the previous tip. In this case, this paramater ///
/// would be `false`. Script-pubkey based syncing mechanisms may not introduce transactions in a /// Block-by-block syncing mechanisms would typically create updates that builds upon the previous
/// chronological order so some updates require introducing older blocks (to anchor older /// tip. In this case, `introduce_older_blocks` would be `false`.
/// transactions). For script-pubkey based syncing, this parameter would typically be `true`. ///
/// Script-pubkey based syncing mechanisms may not introduce transactions in a chronological order
/// so some updates require introducing older blocks (to anchor older transactions). For
/// script-pubkey based syncing, `introduce_older_blocks` would typically be `true`.
#[derive(Debug, Clone)] #[derive(Debug, Clone)]
pub struct Update { pub struct Update {
/// The update chain's new tip. /// The update chain's new tip.
@ -205,13 +209,13 @@ impl LocalChain {
/// Construct a [`LocalChain`] from a given `checkpoint` tip. /// Construct a [`LocalChain`] from a given `checkpoint` tip.
pub fn from_tip(tip: CheckPoint) -> Self { pub fn from_tip(tip: CheckPoint) -> Self {
let mut _self = Self { let mut chain = Self {
tip: Some(tip), tip: Some(tip),
..Default::default() ..Default::default()
}; };
_self.reindex(0); chain.reindex(0);
debug_assert!(_self._check_index_is_consistent_with_tip()); debug_assert!(chain._check_index_is_consistent_with_tip());
_self chain
} }
/// Constructs a [`LocalChain`] from a [`BTreeMap`] of height to [`BlockHash`]. /// Constructs a [`LocalChain`] from a [`BTreeMap`] of height to [`BlockHash`].
@ -247,26 +251,23 @@ impl LocalChain {
/// Returns whether the [`LocalChain`] is empty (has no checkpoints). /// Returns whether the [`LocalChain`] is empty (has no checkpoints).
pub fn is_empty(&self) -> bool { pub fn is_empty(&self) -> bool {
self.tip.is_none() let res = self.tip.is_none();
debug_assert_eq!(res, self.index.is_empty());
res
} }
/// Applies the given `update` to the chain. /// Applies the given `update` to the chain.
/// ///
/// The method returns [`ChangeSet`] on success. This represents the applied changes to `self`. /// The method returns [`ChangeSet`] on success. This represents the applied changes to `self`.
/// ///
/// To update, the `update_tip` must *connect* with `self`. If `self` and `update_tip` has a /// There must be no ambiguity about which of the existing chain's blocks are still valid and
/// mutual checkpoint (same height and hash), it can connect if: /// which are now invalid. That is, the new chain must implicitly connect to a definite block in
/// * The mutual checkpoint is the tip of `self`. /// the existing chain and invalidate the block after it (if it exists) by including a block at
/// * An ancestor of `update_tip` has a height which is of the checkpoint one higher than the /// the same height but with a different hash to explicitly exclude it as a connection point.
/// mutual checkpoint from `self`.
/// ///
/// Additionally: /// Additionally, an empty chain can be updated with any chain, and a chain with a single block
/// * If `self` is empty, `update_tip` will always connect. /// can have it's block invalidated by an update chain with a block at the same height but
/// * If `self` only has one checkpoint, `update_tip` must have an ancestor checkpoint with the /// different hash.
/// same height as it.
///
/// To invalidate from a given checkpoint, `update_tip` must contain an ancestor checkpoint with
/// the same height but different hash.
/// ///
/// # Errors /// # Errors
/// ///
@ -325,7 +326,7 @@ impl LocalChain {
} }
let new_tip = match base { let new_tip = match base {
Some(base) => Some( Some(base) => Some(
base.extend_with_blocks(extension.into_iter().map(BlockId::from)) base.extend(extension.into_iter().map(BlockId::from))
.expect("extension is strictly greater than base"), .expect("extension is strictly greater than base"),
), ),
None => LocalChain::from_blocks(extension).tip(), None => LocalChain::from_blocks(extension).tip(),
@ -379,7 +380,7 @@ impl LocalChain {
self.index.iter().map(|(k, v)| (*k, Some(*v))).collect() self.index.iter().map(|(k, v)| (*k, Some(*v))).collect()
} }
/// Iterate over checkpoints in decending height order. /// Iterate over checkpoints in descending height order.
pub fn iter_checkpoints(&self) -> CheckPointIter { pub fn iter_checkpoints(&self) -> CheckPointIter {
CheckPointIter { CheckPointIter {
current: self.tip.as_ref().map(|tip| tip.0.clone()), current: self.tip.as_ref().map(|tip| tip.0.clone()),
@ -387,7 +388,7 @@ impl LocalChain {
} }
/// Get a reference to the internal index mapping the height to block hash. /// Get a reference to the internal index mapping the height to block hash.
pub fn heights(&self) -> &BTreeMap<u32, BlockHash> { pub fn blocks(&self) -> &BTreeMap<u32, BlockHash> {
&self.index &self.index
} }

View File

@ -627,7 +627,7 @@ impl<A: Anchor> TxGraph<A> {
}; };
let mut has_missing_height = false; let mut has_missing_height = false;
for anchor_block in tx_anchors.iter().map(Anchor::anchor_block) { for anchor_block in tx_anchors.iter().map(Anchor::anchor_block) {
match chain.heights().get(&anchor_block.height) { match chain.blocks().get(&anchor_block.height) {
None => { None => {
has_missing_height = true; has_missing_height = true;
continue; continue;
@ -651,7 +651,7 @@ impl<A: Anchor> TxGraph<A> {
.filter_map(move |(a, _)| { .filter_map(move |(a, _)| {
let anchor_block = a.anchor_block(); let anchor_block = a.anchor_block();
if Some(anchor_block.height) != last_height_emitted if Some(anchor_block.height) != last_height_emitted
&& !chain.heights().contains_key(&anchor_block.height) && !chain.blocks().contains_key(&anchor_block.height)
{ {
last_height_emitted = Some(anchor_block.height); last_height_emitted = Some(anchor_block.height);
Some(anchor_block.height) Some(anchor_block.height)

View File

@ -212,7 +212,7 @@ fn test_list_owned_txouts() {
( (
*tx, *tx,
local_chain local_chain
.heights() .blocks()
.get(&height) .get(&height)
.cloned() .cloned()
.map(|hash| BlockId { height, hash }) .map(|hash| BlockId { height, hash })
@ -232,7 +232,7 @@ fn test_list_owned_txouts() {
|height: u32, |height: u32,
graph: &IndexedTxGraph<ConfirmationHeightAnchor, KeychainTxOutIndex<String>>| { graph: &IndexedTxGraph<ConfirmationHeightAnchor, KeychainTxOutIndex<String>>| {
let chain_tip = local_chain let chain_tip = local_chain
.heights() .blocks()
.get(&height) .get(&height)
.map(|&hash| BlockId { height, hash }) .map(|&hash| BlockId { height, hash })
.unwrap_or_else(|| panic!("block must exist at {}", height)); .unwrap_or_else(|| panic!("block must exist at {}", height));

View File

@ -56,8 +56,8 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
let (update_graph, last_active_indices) = let (update_graph, last_active_indices) =
client.update_tx_graph(keychain_spks, None, None, STOP_GAP, PARALLEL_REQUESTS)?; client.update_tx_graph(keychain_spks, None, None, STOP_GAP, PARALLEL_REQUESTS)?;
let get_heights = wallet.tx_graph().missing_heights(wallet.local_chain()); let missing_heights = wallet.tx_graph().missing_heights(wallet.local_chain());
let chain_update = client.update_local_chain(prev_tip, get_heights)?; let chain_update = client.update_local_chain(prev_tip, missing_heights)?;
let update = LocalUpdate { let update = LocalUpdate {
last_active_indices, last_active_indices,
graph: update_graph, graph: update_graph,

View File

@ -57,8 +57,8 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
let (update_graph, last_active_indices) = client let (update_graph, last_active_indices) = client
.update_tx_graph(keychain_spks, None, None, STOP_GAP, PARALLEL_REQUESTS) .update_tx_graph(keychain_spks, None, None, STOP_GAP, PARALLEL_REQUESTS)
.await?; .await?;
let get_heights = wallet.tx_graph().missing_heights(wallet.local_chain()); let missing_heights = wallet.tx_graph().missing_heights(wallet.local_chain());
let chain_update = client.update_local_chain(prev_tip, get_heights).await?; let chain_update = client.update_local_chain(prev_tip, missing_heights).await?;
let update = LocalUpdate { let update = LocalUpdate {
last_active_indices, last_active_indices,
graph: update_graph, graph: update_graph,