fix: tsv conn

hirosystems · Aug 7, 2024 · fb9304f · fb9304f
1 parent a3aeecf
commit fb9304f
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/components/chainhook-cli/src/cli/mod.rs b/components/chainhook-cli/src/cli/mod.rs
@@ -11,8 +11,8 @@ use crate::storage::{
     delete_confirmed_entry_from_stacks_blocks, delete_unconfirmed_entry_from_stacks_blocks,
     get_last_block_height_inserted, get_last_unconfirmed_block_height_inserted,
     get_stacks_block_at_block_height, insert_unconfirmed_entry_in_stacks_blocks,
-    is_stacks_block_present, open_readonly_stacks_db_conn, open_readwrite_stacks_db_conn,
-    set_last_confirmed_insert_key,
+    is_stacks_block_present, open_readonly_stacks_db_conn, open_readonly_stacks_db_conn_with_retry,
+    open_readwrite_stacks_db_conn, set_last_confirmed_insert_key,
 };
 use chainhook_sdk::chainhooks::bitcoin::BitcoinChainhookSpecification;
 use chainhook_sdk::chainhooks::bitcoin::BitcoinChainhookSpecificationNetworkMap;
@@ -543,16 +543,22 @@ async fn handle_command(opts: Opts, ctx: Context) -> Result<(), String> {
                             }
                         };
                         match open_readonly_stacks_db_conn(&config.expected_cache_path(), &ctx) {
-                            Ok(db_conn) => {
+                            Ok(_) => {
                                 let _ = consolidate_local_stacks_chainstate_using_csv(
                                     &mut config,
                                     &ctx,
                                 )
                                 .await;
+                                // Refresh DB connection so it picks up recent changes made by TSV consolidation.
+                                let new_conn = open_readonly_stacks_db_conn_with_retry(
+                                    &config.expected_cache_path(),
+                                    5,
+                                    &ctx,
+                                )?;
                                 scan_stacks_chainstate_via_rocksdb_using_predicate(
                                     &predicate_spec,
                                     None,
-                                    &db_conn,
+                                    &new_conn,
                                     &config,
                                     None,
                                     &ctx,