example: small fixes, improvements to example-crates

bitcoindevkit · Jul 9, 2024 · 31f8b95 · 31f8b95
1 parent 0f4ab48
commit 31f8b95
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/example-crates/wallet_esplora_async/src/main.rs b/example-crates/wallet_esplora_async/src/main.rs
@@ -44,7 +44,7 @@ async fn main() -> Result<(), anyhow::Error> {
         let mut stdout = std::io::stdout();
         move |spk_i, _| {
             match once.take() {
-                Some(_) => print!("\nScanning keychain [{:?}]", kind),
+                Some(_) => print!("\nScanning keychain [{:?}] {:<3}", kind, spk_i),
                 None => print!(" {:<3}", spk_i),
             };
             stdout.flush().expect("must flush");

diff --git a/example-crates/wallet_rpc/src/main.rs b/example-crates/wallet_rpc/src/main.rs
@@ -143,6 +143,9 @@ fn main() -> anyhow::Result<()> {
                 let connected_to = block_emission.connected_to();
                 let start_apply_block = Instant::now();
                 wallet.apply_block_connected_to(&block_emission.block, height, connected_to)?;
+                if let Some(changeset) = wallet.take_staged() {
+                    db.write(&changeset)?;
+                }
                 let elapsed = start_apply_block.elapsed().as_secs_f32();
                 println!(
                     "Applied block {} at height {} in {}s",
@@ -152,6 +155,9 @@ fn main() -> anyhow::Result<()> {
             Emission::Mempool(mempool_emission) => {
                 let start_apply_mempool = Instant::now();
                 wallet.apply_unconfirmed_txs(mempool_emission.iter().map(|(tx, time)| (tx, *time)));
+                if let Some(changeset) = wallet.take_staged() {
+                    db.write(&changeset)?;
+                }
                 println!(
                     "Applied unconfirmed transactions in {}s",
                     start_apply_mempool.elapsed().as_secs_f32()