From 10f2e04c02d1acf202b3ead9c2be09989fc5cc4b Mon Sep 17 00:00:00 2001
From: 0xMochan <mochanbot@gmail.com>
Date: Mon, 5 Feb 2024 13:58:25 -0500
Subject: [PATCH] fix: only emit balance changes if the component is tracked

---
 substreams/ethereum-balancer/src/modules.rs  | 38 +++++++++++++-------
 substreams/ethereum-balancer/substreams.yaml |  1 +
 2 files changed, 27 insertions(+), 12 deletions(-)

diff --git a/substreams/ethereum-balancer/src/modules.rs b/substreams/ethereum-balancer/src/modules.rs
index 0bc9aed..2ac0f0e 100644
--- a/substreams/ethereum-balancer/src/modules.rs
+++ b/substreams/ethereum-balancer/src/modules.rs
@@ -89,7 +89,10 @@ pub fn store_pools_created(map: tycho::GroupedTransactionProtocolComponents, sto
 /// Since the `PoolBalanceChanged` events administer only deltas, we need to leverage a map and a
 ///  store to be able to tally up final balances for tokens in a pool.
 #[substreams::handlers::map]
-pub fn map_balance_deltas(block: eth::v2::Block) -> Result<tycho::BalanceDeltas, anyhow::Error> {
+pub fn map_balance_deltas(
+    block: eth::v2::Block,
+    store: StoreGetInt64,
+) -> Result<tycho::BalanceDeltas, anyhow::Error> {
     Ok(tycho::BalanceDeltas {
         balance_deltas: block
             .events::<abi::vault::events::PoolBalanceChanged>(&[&VAULT_ADDRESS])
@@ -98,17 +101,28 @@ pub fn map_balance_deltas(block: eth::v2::Block) -> Result<tycho::BalanceDeltas,
                     .tokens
                     .iter()
                     .zip(event.deltas.iter())
-                    .map(|(token, delta)| tycho::BalanceDelta {
-                        ord: log.log.ordinal,
-                        tx: Some(tycho::Transaction {
-                            hash: log.receipt.transaction.hash.clone(),
-                            from: log.receipt.transaction.from.clone(),
-                            to: log.receipt.transaction.to.clone(),
-                            index: Into::<u64>::into(log.receipt.transaction.index),
-                        }),
-                        token: token.clone(),
-                        delta: delta.to_signed_bytes_be(),
-                        component_id: event.pool_id.into(),
+                    .filter_map(|(token, delta)| {
+                        let component_id: Vec<_> = event.pool_id.into();
+
+                        if store
+                            .get_last(format!("pool:{0}", hex::encode(&component_id)))
+                            .is_none()
+                        {
+                            return None;
+                        }
+
+                        Some(tycho::BalanceDelta {
+                            ord: log.log.ordinal,
+                            tx: Some(tycho::Transaction {
+                                hash: log.receipt.transaction.hash.clone(),
+                                from: log.receipt.transaction.from.clone(),
+                                to: log.receipt.transaction.to.clone(),
+                                index: Into::<u64>::into(log.receipt.transaction.index),
+                            }),
+                            token: token.clone(),
+                            delta: delta.to_signed_bytes_be(),
+                            component_id: component_id.clone(),
+                        })
                     })
                     .collect::<Vec<_>>()
             })
diff --git a/substreams/ethereum-balancer/substreams.yaml b/substreams/ethereum-balancer/substreams.yaml
index 7006218..6324d2b 100644
--- a/substreams/ethereum-balancer/substreams.yaml
+++ b/substreams/ethereum-balancer/substreams.yaml
@@ -38,6 +38,7 @@ modules:
     initialBlock: 12369300  # An arbitrary block that should change based on your requirements
     inputs:
       - source: sf.ethereum.type.v2.Block
+      - store: store_pools_created
     output:
       type: proto:tycho.evm.v1.BalanceDeltas