fix: handle token balances on add_contract_changes (#181)

2025-03-21 11:34:37 +02:00
parent 18d8969227
commit aa378e6ef5
1 changed files with 4 additions and 0 deletions
--- a/substreams/crates/tycho-substreams/src/models.rs
+++ b/substreams/crates/tycho-substreams/src/models.rs
@@ -52,6 +52,9 @@ impl TransactionChangesBuilder {
                if !change.code.is_empty() {
                    c.set_code(&change.code)
                }
+                if !change.token_balances.is_empty() {
+                    c.upsert_token_balances(&change.token_balances)
+                }
            })
            .or_insert_with(|| {
                let mut c = InterimContractChange::new(
@@ -61,6 +64,7 @@ impl TransactionChangesBuilder {
                c.upsert_slots(&change.slots);
                c.set_code(&change.code);
                c.set_balance(&change.balance);
+                c.upsert_token_balances(&change.token_balances);
                c
            });
    }