Add Account and create relationship with Balance

schema.graphql

@@ -264,9 +264,15 @@ type ReserveMetrics @entity {
   allocations: [ReserveAllocationMetrics] @derivedFrom(field: "reserveMetrics")
 }
 
-type Balances @entity {
+type Balance @entity {
   id: ID!
   address: String @index
   balance: BigInt
   denom: String  @index
-}
+  account: Account!
+}
+
+type Account @entity {
+  id: ID!
+  balances: [Balance] @derivedFrom(field: "account")
+}

src/mappings/events/balances.ts

@@ -1,4 +1,4 @@
-import { Balances } from '../../types';
+import { Account, Balance } from '../../types';
 import { BALANCE_FIELDS } from '../constants';
 import { b64decode } from '../utils';
 import { CosmosEvent } from '@subql/types-cosmos';
@@ -72,7 +72,7 @@ export const balancesEventKit = () => {
     address: string,
     denom: string
   ): Promise<boolean> {
-    const balance = await Balances.getByFields([
+    const balance = await Balance.getByFields([
       ['address', '=', address],
       ['denom', '=', denom],
     ]);
@@ -88,13 +88,14 @@ export const balancesEventKit = () => {
     denom: string,
     primaryKey: string
   ) {
-    const newBalance = new Balances(primaryKey);
+    const newBalance = new Balance(primaryKey, address);
     newBalance.address = address;
     newBalance.balance = BigInt(0);
     newBalance.denom = denom;
-
     await newBalance.save();
 
+    await new Account(address).save();
+
     logger.info(`Created new entry for address: ${address}`);
   }
 
@@ -127,7 +128,7 @@ export const balancesEventKit = () => {
     amount: bigint,
     operation: Operation
   ): Promise<void> {
-    const balances = await Balances.getByFields([
+    const balances = await Balance.getByFields([
       ['address', '=', address],
       ['denom', '=', denom],
     ]);

src/mappings/mappingHandlers.ts

@@ -16,7 +16,7 @@ import {
   VaultManagerMetricsDaily,
   PsmMetricDaily,
   ReserveAllocationMetricsDaily,
-  Balances,
+  Balance,
 } from "../types";
 import { CosmosBlock, CosmosEvent } from "@subql/types-cosmos";
 import {
@@ -38,7 +38,7 @@ import {
   STORE_NAME_KEY,
   SUBKEY_KEY,
   UNPROVED_VALUE_KEY,
-  BALANCE_FIELDS
+  BALANCE_FIELDS,
 } from "./constants";
 import { psmEventKit } from "./events/psm";
 import { boardAuxEventKit } from "./events/boardAux";
@@ -245,7 +245,7 @@ export async function initiateBalancesTable(block: CosmosBlock): Promise<void> {
     for (let element of data.balances) {
       let newBalance;
       for (const coin of element.coins) {
-        newBalance = new Balances(`${element.address}-${coin.denom}`);
+        newBalance = new Balance(`${element.address}-${coin.denom}`, element.address);
         newBalance.address = element.address;
         newBalance.balance = BigInt(coin.amount);
         newBalance.denom = coin.denom;