fix(rpc/v0.4): align trace updates

eqlabs · Feb 2, 2024 · ed04033 · ed04033
1 parent b28f94f
commit ed04033
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/crates/rpc/src/v04/method/trace_block_transactions.rs b/crates/rpc/src/v04/method/trace_block_transactions.rs
@@ -102,7 +102,7 @@ pub async fn trace_block_transactions(
         let hash = header.hash;
         let state = ExecutionState::trace(&db, context.chain_id, header, None);
         let traces =
-            pathfinder_executor::trace(state, &context.cache, hash, transactions, true, true)?;
+            pathfinder_executor::trace(state, context.cache, hash, transactions, true, true)?;
 
         let result = traces
             .into_iter()

diff --git a/crates/rpc/src/v04/method/trace_transaction.rs b/crates/rpc/src/v04/method/trace_transaction.rs
@@ -104,7 +104,7 @@ pub async fn trace_transaction(
             .get(&db)
             .context("Querying pending data")?;
 
-        let (header, transactions) = if let Some(pending_tx) = pending
+        let (header, transactions, cache) = if let Some(pending_tx) = pending
             .block
             .transactions
             .iter()
@@ -123,7 +123,11 @@ pub async fn trace_transaction(
                 return Ok(LocalExecution::Unsupported(pending_tx.clone()));
             }
 
-            (header, pending.block.transactions.clone())
+            (
+                header,
+                pending.block.transactions.clone(),
+                Default::default(),
+            )
         } else {
             let block_hash = db
                 .transaction_block_hash(input.transaction_hash)?
@@ -155,7 +159,7 @@ pub async fn trace_transaction(
                 .context("Fetching block transactions")?
                 .context("Block transactions missing")?;
 
-            (header, transactions.clone())
+            (header, transactions.clone(), context.cache.clone())
         };
 
         let hash = header.hash;
@@ -166,7 +170,7 @@ pub async fn trace_transaction(
             .map(|transaction| compose_executor_transaction(transaction, &db))
             .collect::<Result<Vec<_>, _>>()?;
 
-        pathfinder_executor::trace(state, &context.cache, hash, transactions, true, true)
+        pathfinder_executor::trace(state, cache, hash, transactions, true, true)
             .map_err(TraceTransactionError::from)
             .and_then(|txs| {
                 txs.into_iter()