apollo_gateway: add large storage writer

Author: einat-starkware

crates/apollo_gateway/src/gateway.rs

@@ -55,6 +55,7 @@ use crate::stateless_transaction_validator::{
     StatelessTransactionValidatorTrait,
 };
 use crate::sync_state_reader::SyncStateReaderFactory;
+use crate::large_storage_writer::{LargeStorageWriter, LargeStorageWriterTrait};
 
 #[cfg(test)]
 #[path = "gateway_test.rs"]
@@ -68,6 +69,7 @@ pub struct Gateway {
     pub mempool_client: SharedMempoolClient,
     pub transaction_converter: Arc<dyn TransactionConverterTrait>,
     pub proof_manager_client: SharedProofManagerClient,
+    pub large_storage_writer: Arc<dyn LargeStorageWriterTrait>,
 }
 
 impl Gateway {
@@ -78,6 +80,7 @@ impl Gateway {
         transaction_converter: Arc<dyn TransactionConverterTrait>,
         stateless_tx_validator: Arc<dyn StatelessTransactionValidatorTrait>,
         proof_manager_client: SharedProofManagerClient,
+        large_storage_writer: Arc<dyn LargeStorageWriterTrait>,
     ) -> Self {
         Self {
             config: Arc::new(config.clone()),
@@ -93,6 +96,7 @@ impl Gateway {
             mempool_client,
             transaction_converter,
             proof_manager_client,
+            large_storage_writer,
         }
     }
 
@@ -258,14 +262,15 @@ pub fn create_gateway(
     let stateless_tx_validator = Arc::new(StatelessTransactionValidator {
         config: config.stateless_tx_validator_config.clone(),
     });
-
+    let large_storage_writer = Arc::new(LargeStorageWriter::new());
     Gateway::new(
         config,
         state_reader_factory,
         mempool_client,
         transaction_converter,
         stateless_tx_validator,
         proof_manager_client,
+        large_storage_writer,
     )
 }
 

crates/apollo_gateway/src/gateway_test.rs

@@ -101,6 +101,7 @@ use crate::stateful_transaction_validator::{
     MockStatefulTransactionValidatorTrait,
 };
 use crate::stateless_transaction_validator::MockStatelessTransactionValidatorTrait;
+use crate::large_storage_writer::MockLargeStorageWriterTrait;
 
 #[fixture]
 fn mock_stateful_transaction_validator() -> MockStatefulTransactionValidatorTrait {
@@ -135,13 +136,15 @@ fn mock_dependencies() -> MockDependencies {
     let mock_transaction_converter = MockTransactionConverterTrait::new();
     let mock_stateless_transaction_validator = mock_stateless_transaction_validator();
     let mock_proof_manager_client = MockProofManagerClient::new();
+    let mock_large_storage_writer = MockLargeStorageWriterTrait::new();
     MockDependencies {
         config,
         state_reader_factory,
         mock_mempool_client,
         mock_transaction_converter,
         mock_stateless_transaction_validator,
         mock_proof_manager_client,
+        mock_large_storage_writer,
     }
 }
 
@@ -152,6 +155,7 @@ struct MockDependencies {
     mock_transaction_converter: MockTransactionConverterTrait,
     mock_stateless_transaction_validator: MockStatelessTransactionValidatorTrait,
     mock_proof_manager_client: MockProofManagerClient,
+    mock_large_storage_writer: MockLargeStorageWriterTrait,
 }
 
 impl MockDependencies {
@@ -164,6 +168,7 @@ impl MockDependencies {
             Arc::new(self.mock_transaction_converter),
             Arc::new(self.mock_stateless_transaction_validator),
             Arc::new(self.mock_proof_manager_client),
+            Arc::new(self.mock_large_storage_writer),
         )
     }
 
@@ -601,6 +606,7 @@ async fn add_tx_returns_error_when_extract_state_nonce_and_run_validations_fails
         mempool_client: Arc::new(mock_dependencies.mock_mempool_client),
         transaction_converter: Arc::new(mock_dependencies.mock_transaction_converter),
         proof_manager_client: Arc::new(mock_dependencies.mock_proof_manager_client),
+        large_storage_writer: Arc::new(mock_dependencies.mock_large_storage_writer),
     };
 
     let result = gateway.add_tx(input_tx, None).await;
@@ -667,6 +673,7 @@ async fn add_tx_returns_error_when_instantiating_validator_fails(
         mempool_client: Arc::new(mock_dependencies.mock_mempool_client),
         transaction_converter: Arc::new(mock_dependencies.mock_transaction_converter),
         proof_manager_client: Arc::new(mock_dependencies.mock_proof_manager_client),
+        large_storage_writer: Arc::new(mock_dependencies.mock_large_storage_writer),
     };
 
     let result = gateway.add_tx(input_tx, None).await;

crates/apollo_gateway/src/test_utils.rs

@@ -33,6 +33,7 @@ use starknet_types_core::felt::Felt;
 use crate::gateway::Gateway;
 use crate::state_reader_test_utils::local_test_state_reader_factory;
 use crate::stateless_transaction_validator::StatelessTransactionValidator;
+use crate::large_storage_writer::MockLargeStorageWriterTrait;
 
 pub const NON_EMPTY_RESOURCE_BOUNDS: ResourceBounds =
     ResourceBounds { max_amount: GasAmount(1), max_price_per_unit: GasPrice(1) };
@@ -165,6 +166,7 @@ pub fn gateway_for_benchmark(gateway_config: GatewayConfig) -> Gateway {
     let mut mempool_client = MockMempoolClient::new();
     let class_manager_client = Arc::new(MockClassManagerClient::new());
     let proof_manager_client = Arc::new(MockProofManagerClient::new());
+    let large_storage_writer = Arc::new(MockLargeStorageWriterTrait::new());
     let transaction_converter = TransactionConverter::new(
         class_manager_client.clone(),
         gateway_config.chain_info.chain_id.clone(),
@@ -182,5 +184,6 @@ pub fn gateway_for_benchmark(gateway_config: GatewayConfig) -> Gateway {
         Arc::new(transaction_converter),
         stateless_tx_validator,
         proof_manager_client,
+        large_storage_writer,
     )
 }