feat(rpc): v0.8.0 getMessagesStatus method

CHANGELOG.md

@@ -2,6 +2,7 @@
 
 ## Next release
 
+- feat(rpc): added `getMessagesStatus` method
 - fix(db): fix number of files in db, startup hang, ram issues and flushing issues
 - fix: FeePayment conversion
 - fix(block_production): get l2-to-l1 messages recursively from the call tree

crates/client/db/Cargo.toml

@@ -33,6 +33,7 @@ starknet-types-core = { workspace = true }
 starknet_api = { workspace = true }
 
 # Other
+alloy = { workspace = true }
 anyhow.workspace = true
 bincode = { workspace = true }
 rayon = { workspace = true }

crates/client/db/src/l1_db.rs

@@ -1,9 +1,10 @@
+use crate::error::DbError;
+use crate::{Column, DatabaseExt, MadaraBackend, MadaraStorageError};
+use alloy::primitives::TxHash;
 use rocksdb::WriteOptions;
 use serde::{Deserialize, Serialize};
 use starknet_api::core::Nonce;
-
-use crate::error::DbError;
-use crate::{Column, DatabaseExt, MadaraBackend, MadaraStorageError};
+use starknet_types_core::felt::Felt;
 
 type Result<T, E = MadaraStorageError> = std::result::Result<T, E>;
 
@@ -128,4 +129,32 @@ impl MadaraBackend {
         self.db.put_cf_opt(&nonce_column, bincode::serialize(&nonce)?, /* empty value */ [], &writeopts)?;
         Ok(())
     }
+
+    pub fn get_l1_handler_tx_hashes(&self, l1_tx_hash: TxHash) -> Result<Vec<Felt>, DbError> {
+        let l1_l2_mappings_column = self.db.get_column(Column::L1MessagingHandlerTxHashes);
+        let mut l1_handler_tx_hashes = vec![];
+        for kv_bytes in self.db.prefix_iterator_cf(&l1_l2_mappings_column, l1_tx_hash.as_slice()) {
+            let l1_handler_tx_hash = Felt::from_bytes_be_slice(&kv_bytes?.1);
+            l1_handler_tx_hashes.push(l1_handler_tx_hash);
+        }
+        Ok(l1_handler_tx_hashes)
+    }
+
+    /// Store mapping from L1 transaction to L1 handler transaction (on the L2). A unique order
+    /// value is required to ensure the handler transactions are retreived in the correct order.
+    pub fn add_l1_handler_tx_hash_mapping(
+        &self,
+        l1_tx_hash: TxHash,
+        l1_handler_tx_hash: Felt,
+        order: u64,
+    ) -> Result<(), DbError> {
+        let l1_l2_mappings_column = self.db.get_column(Column::L1MessagingHandlerTxHashes);
+        let mut key = [0u8; 40];
+        key[..32].copy_from_slice(l1_tx_hash.as_slice());
+        key[32..].copy_from_slice(&order.to_be_bytes()); // BE is important for the lexographic sorting
+        let mut writeopts = WriteOptions::default();
+        writeopts.disable_wal(true);
+        self.db.put_cf_opt(&l1_l2_mappings_column, key, l1_handler_tx_hash.to_bytes_be(), &writeopts)?;
+        Ok(())
+    }
 }

crates/client/db/src/lib.rs

@@ -143,6 +143,7 @@ pub enum Column {
 
     L1Messaging,
     L1MessagingNonce,
+    L1MessagingHandlerTxHashes,
 
     /// Devnet: stores the private keys for the devnet predeployed contracts
     Devnet,
@@ -190,6 +191,7 @@ impl Column {
             BonsaiClassesLog,
             L1Messaging,
             L1MessagingNonce,
+            L1MessagingHandlerTxHashes,
             PendingContractToClassHashes,
             PendingContractToNonces,
             PendingContractStorage,
@@ -227,6 +229,7 @@ impl Column {
             ContractStorage => "contract_storage",
             L1Messaging => "l1_messaging",
             L1MessagingNonce => "l1_messaging_nonce",
+            L1MessagingHandlerTxHashes => "l1_messaging_handler_tx_hashes",
             PendingContractToClassHashes => "pending_contract_to_class_hashes",
             PendingContractToNonces => "pending_contract_to_nonces",
             PendingContractStorage => "pending_contract_storage",

crates/client/db/src/rocksdb_options.rs

@@ -2,6 +2,8 @@
 #![allow(non_upper_case_globals)] // allow KiB/MiB/GiB names
 
 use crate::{contract_db, Column};
+use alloy::primitives::private::alloy_rlp::MaxEncodedLenAssoc;
+use alloy::primitives::TxHash;
 use anyhow::{Context, Result};
 use rocksdb::{DBCompressionType, Env, Options, SliceTransform};
 
@@ -56,6 +58,9 @@ impl Column {
                     contract_db::CONTRACT_NONCES_PREFIX_EXTRACTOR,
                 ));
             }
+            Column::L1MessagingHandlerTxHashes => {
+                options.set_prefix_extractor(SliceTransform::create_fixed_prefix(TxHash::LEN));
+            }
             _ => {}
         }