From 012673929345fde6f4093c3b7f03d1b279a299a7 Mon Sep 17 00:00:00 2001
From: Chris Guimaraes <cguimaraes@hiro.so>
Date: Wed, 6 Nov 2024 16:43:16 +0000
Subject: [PATCH] fix: warn when not able to query mock_proposals

---
 components/chainhook-cli/src/storage/signers.rs | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/components/chainhook-cli/src/storage/signers.rs b/components/chainhook-cli/src/storage/signers.rs
index 2a48836a..9729178d 100644
--- a/components/chainhook-cli/src/storage/signers.rs
+++ b/components/chainhook-cli/src/storage/signers.rs
@@ -108,7 +108,7 @@ pub fn initialize_signers_db(base_dir: &PathBuf, ctx: &Context) -> Result<Connec
     conn.execute(
         "CREATE TABLE IF NOT EXISTS mock_proposals (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
-            message_id INTEGER NOT NULL,
+            message_id INTEGER,
             burn_block_height INTEGER NOT NULL,
             stacks_tip_consensus_hash TEXT NOT NULL,
             stacks_tip TEXT NOT NULL,
@@ -143,7 +143,7 @@ pub fn initialize_signers_db(base_dir: &PathBuf, ctx: &Context) -> Result<Connec
         "CREATE TABLE IF NOT EXISTS mock_signatures (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
             mock_proposal_id INTEGER NOT NULL,
-            message_id INTEGER NOT NULL,
+            message_id INTEGER,
             mock_block_id INTEGER,
             server_version TEXT NOT NULL,
             signature TEXT NOT NULL,
@@ -699,8 +699,13 @@ pub fn get_signer_db_messages_received_at_block(
                                 index_block_hash: proposal_row.get(7).unwrap(),
                             }))
                         },
-                    )
-                    .map_err(|e| format!("unable to query mock proposal: {e}"))?,
+                    ).unwrap_or_else(|e| {
+                        warn!(
+                            ctx.expect_logger(),
+                            "unable to query mock_proposals table for message_id {}: {}", message_id, e
+                        );
+                        return vec![];
+                    }),
                 "mock_block" => db_tx
                     .query_row(
                         "SELECT b.id, p.burn_block_height, p.stacks_tip_consensus_hash, p.stacks_tip, p.stacks_tip_height,