When the before/until signatures can't be found, throw `Signature…

…NotFound` instead of `RowNotFound`
anza-xyz · Nov 19, 2024 · f2ca0d0 · f2ca0d0
1 parent 3dd5947
commit f2ca0d0
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/storage-bigtable/src/lib.rs b/storage-bigtable/src/lib.rs
@@ -796,7 +796,11 @@ impl LedgerStorage {
             Some(before_signature) => {
                 let TransactionInfo { slot, index, .. } = bigtable
                     .get_bincode_cell("tx", before_signature.to_string())
-                    .await?;
+                    .await
+                    .map_err(|err| match err {
+                        bigtable::Error::RowNotFound => Error::SignatureNotFound,
+                        _ => err.into(),
+                    })?;
 
                 (slot, index)
             }
@@ -808,7 +812,11 @@ impl LedgerStorage {
             Some(until_signature) => {
                 let TransactionInfo { slot, index, .. } = bigtable
                     .get_bincode_cell("tx", until_signature.to_string())
-                    .await?;
+                    .await
+                    .map_err(|err| match err {
+                        bigtable::Error::RowNotFound => Error::SignatureNotFound,
+                        _ => err.into(),
+                    })?;
 
                 (slot, index)
             }