From 5bdf4ba0a2aac9aa4a1e21a0dccb6c785c862e4b Mon Sep 17 00:00:00 2001
From: colindickson <colin.dickson@gmail.com>
Date: Fri, 17 Nov 2023 16:16:28 -0500
Subject: [PATCH] fix entity for graph_out

---
 schema.graphql  | 14 +++-----------
 src/sinks.rs    | 18 +++++++-----------
 subgraph.yaml   |  2 +-
 substreams.yaml |  2 +-
 4 files changed, 12 insertions(+), 24 deletions(-)

diff --git a/schema.graphql b/schema.graphql
index f704501..ec5b068 100644
--- a/schema.graphql
+++ b/schema.graphql
@@ -1,12 +1,4 @@
-# Log
-
-enum BalanceChangeType {
-    TYPE_UNKNOWN
-    TYPE_1
-    TYPE_2
-}
-
-type BalanceChange @entity(immutable: true) {
+type BalanceChange @entity {
     id: ID!
 
     contract: String
@@ -15,7 +7,7 @@ type BalanceChange @entity(immutable: true) {
     newBalance: String
     transaction: String
     storageKey: String
-    callIndex: Int
+    callIndex: BigInt
     transferValue: String
-    changeType: BalanceChangeType
+    changeType: Int
 }
diff --git a/src/sinks.rs b/src/sinks.rs
index 0f192cc..fc72c8c 100644
--- a/src/sinks.rs
+++ b/src/sinks.rs
@@ -6,8 +6,6 @@ use substreams_database_change::pb::database::DatabaseChanges;
 #[substreams::handlers::map]
 pub fn graph_out(clock: Clock, balance_changes: BalanceChanges) -> Result<EntityChanges, Error> {
     let mut tables = substreams_entity_change::tables::Tables::new();
-    let block_num = clock.number.to_string();
-    let timestamp = clock.timestamp.unwrap().seconds.to_string();
 
     for balance_change in balance_changes.balance_changes {
         if balance_change.change_type == 0 {
@@ -17,18 +15,16 @@ pub fn graph_out(clock: Clock, balance_changes: BalanceChanges) -> Result<Entity
         let key = format!("{}:{}:{}", balance_change.contract, balance_change.owner, balance_change.transaction);
 
         tables
-            .create_row("balance_change", key)
+            .create_row("BalanceChange", key)
             .set("contract", balance_change.contract)
             .set("owner", balance_change.owner)
-            .set("transfer_value", balance_change.transfer_value)
-            .set("old_balance", balance_change.old_balance)
-            .set("new_balance", balance_change.new_balance)
-            .set("storage_key", balance_change.storage_key)
-            .set("call_index", balance_change.call_index)
+            .set("transferValue", balance_change.transfer_value)
+            .set("oldBalance", balance_change.old_balance)
+            .set("newBalance", balance_change.new_balance)
+            .set("storageKey", balance_change.storage_key)
+            .set("callIndex", balance_change.call_index)
             .set("transaction", balance_change.transaction)
-            .set("change_type", balance_change.change_type)
-            .set_bigint("block_num", &block_num)
-            .set_bigint("timestamp", &timestamp);
+            .set("changeType", balance_change.change_type);
     }
 
     Ok(tables.to_entity_changes())
diff --git a/subgraph.yaml b/subgraph.yaml
index b1fccb9..b51b9b4 100644
--- a/subgraph.yaml
+++ b/subgraph.yaml
@@ -11,7 +11,7 @@ dataSources:
     source:
       package:
         moduleName: graph_out
-        file: ./erc20-balance-changes-v1.0.0.spkg
+        file: ./erc20-balance-changes-v1.1.0.spkg
     mapping:
       kind: substreams/graph-entities
       apiVersion: 0.0.7
diff --git a/substreams.yaml b/substreams.yaml
index 43ac874..7ba1e1d 100644
--- a/substreams.yaml
+++ b/substreams.yaml
@@ -1,7 +1,7 @@
 specVersion: v0.1.0
 package:
   name: erc20_balance_changes
-  version: v1.0.0
+  version: v1.1.0
   url: https://github.com/streamingfast/erc20-balance-changes
   image: logo.png
   doc: |