apollo_storage: implement TxExecutionInfoStorageReader and TxExecutionInfoStorageWriter (#13926)

Co-authored-by: Claude Sonnet 4.6 <noreply@anthropic.com>

Author: yoavGrs

crates/apollo_storage/src/lib.rs

@@ -921,7 +921,6 @@ impl FileHandlers<RW> {
     }
 
     #[cfg(feature = "os_input")]
-    #[expect(dead_code)]
     fn append_tx_execution_infos(&self, tx_execution_infos: &TxExecutionInfos) -> LocationInFile {
         self.clone().tx_execution_infos.append(tx_execution_infos)
     }
@@ -1022,7 +1021,6 @@ impl<Mode: TransactionKind> FileHandlers<Mode> {
     }
 
     #[cfg(feature = "os_input")]
-    #[expect(dead_code)]
     // Returns the transaction execution infos at the given location or an error in case they don't
     // exist.
     pub(crate) fn get_tx_execution_infos_unchecked(

crates/apollo_storage/src/tx_execution_info.rs

@@ -1,9 +1,13 @@
 //! Storage for transaction execution info per block (input to the starknet OS).
 
 use blockifier::transaction::objects::TransactionExecutionInfo;
+use starknet_api::block::BlockNumber;
 
 use crate::compression_utils::{compress, decompress};
 use crate::db::serialization::{StorageSerde, StorageSerdeError};
+use crate::db::table_types::Table;
+use crate::db::{TransactionKind, RW};
+use crate::{OffsetKind, StorageResult, StorageTxn};
 
 /// Per-block container of transaction execution infos, stored as a compressed JSON blob.
 #[derive(Debug)]
@@ -22,3 +26,70 @@ impl StorageSerde for TxExecutionInfos {
         serde_json::from_slice(&data).ok().map(TxExecutionInfos)
     }
 }
+
+/// Interface for reading transaction execution infos from storage.
+pub trait TxExecutionInfoStorageReader<Mode: TransactionKind> {
+    /// Returns the transaction execution infos for the given block, or `None` if not stored.
+    fn get_tx_execution_infos(
+        &self,
+        block_number: BlockNumber,
+    ) -> StorageResult<Option<Vec<TransactionExecutionInfo>>>;
+}
+
+/// Interface for writing transaction execution infos to storage.
+pub trait TxExecutionInfoStorageWriter
+where
+    Self: Sized,
+{
+    /// Appends the transaction execution infos for the given block to storage.
+    fn append_tx_execution_infos(
+        self,
+        block_number: BlockNumber,
+        tx_execution_infos: Vec<TransactionExecutionInfo>,
+    ) -> StorageResult<Self>;
+
+    /// Removes the transaction execution infos for the given block from storage.
+    /// If no entry exists for the block, returns without error.
+    fn revert_tx_execution_infos(self, block_number: BlockNumber) -> StorageResult<Self>;
+}
+
+impl<Mode: TransactionKind> TxExecutionInfoStorageReader<Mode> for StorageTxn<'_, Mode> {
+    fn get_tx_execution_infos(
+        &self,
+        block_number: BlockNumber,
+    ) -> StorageResult<Option<Vec<TransactionExecutionInfo>>> {
+        let table = self.open_table(&self.tables.tx_execution_infos)?;
+        let Some(location) = table.get(&self.txn, &block_number)? else {
+            return Ok(None);
+        };
+        Ok(Some(self.file_handlers.get_tx_execution_infos_unchecked(location)?.0))
+    }
+}
+
+impl TxExecutionInfoStorageWriter for StorageTxn<'_, RW> {
+    fn append_tx_execution_infos(
+        self,
+        block_number: BlockNumber,
+        tx_execution_infos: Vec<TransactionExecutionInfo>,
+    ) -> StorageResult<Self> {
+        let file_offset_table = self.txn.open_table(&self.tables.file_offsets)?;
+        let tx_execution_infos_table = self.open_table(&self.tables.tx_execution_infos)?;
+
+        let infos = TxExecutionInfos(tx_execution_infos);
+        let location = self.file_handlers.append_tx_execution_infos(&infos);
+        tx_execution_infos_table.upsert(&self.txn, &block_number, &location)?;
+        file_offset_table.upsert(
+            &self.txn,
+            &OffsetKind::TxExecutionInfo,
+            &location.next_offset(),
+        )?;
+
+        Ok(self)
+    }
+
+    fn revert_tx_execution_infos(self, block_number: BlockNumber) -> StorageResult<Self> {
+        let tx_execution_infos_table = self.open_table(&self.tables.tx_execution_infos)?;
+        tx_execution_infos_table.delete(&self.txn, &block_number)?;
+        Ok(self)
+    }
+}