Liquidity.Party/tycho-protocol-sdk
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #53 from propeller-heads/fp/fix-vm-indexing

Browse Source 
fix(substreams): Fix storage changes indexing for DELEGATECALL
This commit is contained in:
Zizou
2024-07-22 10:25:14 +02:00
committed by GitHub
parent 52c644a4ed e6edcd1fa2
commit ead8e17082
1 changed files with 18 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
substreams/crates/tycho-substreams/src/contract.rs
View File

@@ -10,9 +10,9 @@
/// more [here](https://streamingfastio.medium.com/new-block-model-to-accelerate-chain-integration-9f65126e5425)
use std::collections::HashMap;
use substreams_ethereum::pb::{
eth,
eth::v2::{block::DetailLevel, StorageChange},
use substreams_ethereum::pb::eth::{
self,
v2::{block::DetailLevel, CallType, StorageChange},
};
use crate::pb::tycho::evm::v1::{self as tycho};
@@ -128,15 +128,21 @@ pub fn extract_contract_changes<F: Fn(&[u8]) -> bool>(
let mut balance_changes = Vec::new();
let mut code_changes = Vec::new();
block_tx
.calls
.iter()
.filter(|call| !call.state_reverted && inclusion_predicate(&call.address))
.for_each(|call| {
storage_changes.extend(call.storage_changes.iter());
balance_changes.extend(call.balance_changes.iter());
code_changes.extend(call.code_changes.iter());
});
let filtered_calls = block_tx.calls.iter().filter(|call| {
let address_included = inclusion_predicate(&call.address);
let caller_included = inclusion_predicate(&call.caller);
let is_delegate_or_callcode = call.call_type() == CallType::Delegate ||
call.call_type() == CallType::Callcode;
!call.state_reverted &&
(address_included || (caller_included && is_delegate_or_callcode))
});
filtered_calls.for_each(|call| {
storage_changes.extend(call.storage_changes.iter());
balance_changes.extend(call.balance_changes.iter());
code_changes.extend(call.code_changes.iter());
});
storage_changes.sort_unstable_by_key(|change| change.ordinal);
balance_changes.sort_unstable_by_key(|change| change.ordinal);