apollo_gateway: move get_block_info to be async

Author: itamar-starkware

crates/apollo_gateway/src/rpc_state_reader.rs

@@ -94,15 +94,20 @@ impl RpcStateReader {
     }
 }
 
+#[async_trait]
 impl MempoolStateReader for RpcStateReader {
-    fn get_block_info(&self) -> StateResult<BlockInfo> {
+    async fn get_block_info(&self) -> StateResult<BlockInfo> {
         let get_block_params = GetBlockWithTxHashesParams { block_id: self.block_id };
+        let reader = self.clone();
+
+        let rpc_request_value = tokio::task::spawn_blocking(move || {
+            reader.send_rpc_request("starknet_getBlockWithTxHashes", get_block_params)
+        })
+        .await
+        .map_err(|e| StateError::StateReadError(format!("JoinError: {e}")))??;
 
-        // The response from the rpc is a full block but we only deserialize the header.
-        let block_header: BlockHeader = serde_json::from_value(
-            self.send_rpc_request("starknet_getBlockWithTxHashes", get_block_params)?,
-        )
-        .map_err(serde_err_to_state_err)?;
+        let block_header: BlockHeader =
+            serde_json::from_value(rpc_request_value).map_err(serde_err_to_state_err)?;
         let block_info = block_header.try_into()?;
         Ok(block_info)
     }

crates/apollo_gateway/src/rpc_state_reader_test.rs

@@ -92,8 +92,7 @@ async fn test_get_block_info() {
     );
 
     let client = RpcStateReader::from_latest(&config);
-    let result =
-        tokio::task::spawn_blocking(move || client.get_block_info()).await.unwrap().unwrap();
+    let result = client.get_block_info().await.unwrap();
     assert_eq!(result, expected_result);
     mock.assert_async().await;
 }

crates/apollo_gateway/src/state_reader.rs

@@ -10,8 +10,9 @@ use starknet_api::core::{ClassHash, CompiledClassHash, ContractAddress, Nonce};
 use starknet_api::state::StorageKey;
 use starknet_types_core::felt::Felt;
 
+#[async_trait]
 pub trait MempoolStateReader: BlockifierStateReader + Send + Sync {
-    fn get_block_info(&self) -> Result<BlockInfo, StateError>;
+    async fn get_block_info(&self) -> Result<BlockInfo, StateError>;
 }
 
 #[cfg_attr(test, automock)]
@@ -25,9 +26,10 @@ pub trait StateReaderFactory: Send + Sync {
 // By default, a Box<dyn Trait> does not implement the trait of the object it contains.
 // Therefore, for using the Box<dyn MempoolStateReader>, that the StateReaderFactory creates,
 // we need to implement the MempoolStateReader trait for Box<dyn MempoolStateReader>.
+#[async_trait]
 impl MempoolStateReader for Box<dyn MempoolStateReader> {
-    fn get_block_info(&self) -> Result<BlockInfo, StateError> {
-        self.as_ref().get_block_info()
+    async fn get_block_info(&self) -> Result<BlockInfo, StateError> {
+        self.as_ref().get_block_info().await
     }
 }
 

crates/apollo_gateway/src/state_reader_test_utils.rs

@@ -23,8 +23,9 @@ pub struct TestStateReader {
     pub blockifier_state_reader: DictStateReader,
 }
 
+#[async_trait]
 impl MempoolStateReader for TestStateReader {
-    fn get_block_info(&self) -> Result<BlockInfo, StateError> {
+    async fn get_block_info(&self) -> Result<BlockInfo, StateError> {
         Ok(self.block_info.clone())
     }
 }

crates/apollo_gateway/src/stateful_transaction_validator.rs

@@ -333,7 +333,7 @@ fn skip_stateful_validations(
 pub fn get_latest_block_info(
     state_reader: &dyn MempoolStateReader,
 ) -> StatefulTransactionValidatorResult<BlockInfo> {
-    state_reader
-        .get_block_info()
+    tokio::runtime::Handle::current()
+        .block_on(state_reader.get_block_info())
         .map_err(|e| StarknetError::internal_with_logging("Failed to get latest block info", e))
 }

crates/apollo_gateway/src/sync_state_reader.rs

@@ -16,7 +16,6 @@ use async_trait::async_trait;
 use blockifier::execution::contract_class::RunnableCompiledClass;
 use blockifier::state::errors::StateError;
 use blockifier::state::state_api::{StateReader as BlockifierStateReader, StateResult};
-use futures::executor::block_on;
 use starknet_api::block::{BlockHash, BlockInfo, BlockNumber, GasPriceVector, GasPrices};
 use starknet_api::contract_class::ContractClass;
 use starknet_api::core::{ClassHash, CompiledClassHash, ContractAddress, Nonce};
@@ -55,9 +54,13 @@ impl SyncStateReader {
     }
 }
 
+#[async_trait]
 impl MempoolStateReader for SyncStateReader {
-    fn get_block_info(&self) -> StateResult<BlockInfo> {
-        let block = block_on(self.state_sync_client.get_block(self.block_number))
+    async fn get_block_info(&self) -> StateResult<BlockInfo> {
+        let block = self
+            .state_sync_client
+            .get_block(self.block_number)
+            .await
             .map_err(|e| StateError::StateReadError(e.to_string()))?;
 
         let block_header = block.block_header_without_hash;

crates/apollo_gateway/src/sync_state_reader_test.rs

@@ -71,7 +71,7 @@ async fn test_get_block_info() {
         block_number,
         tokio::runtime::Handle::current(),
     );
-    let result = state_sync_reader.get_block_info().unwrap();
+    let result = state_sync_reader.get_block_info().await.unwrap();
 
     assert_eq!(
         result,