Add Supply entity and update supply handling logic

- Introduced `supplies` field in `Block` type to reference `Supply` entities.
- Added new fields `denom` and `block` to `Supply` type.
- Replaced `getSupply` with `queryTotalSupply` and implement `getSupplyRecord`.
- Modified `_handleSupply` to create supply records with `getSupplyRecord` and avoid duplicates.

Author: jorgecuesta

schema.graphql

@@ -23,6 +23,7 @@ type Block @entity {
   messages: [Message] @derivedFrom(field: "block")
   events: [Event] @derivedFrom(field: "block")
   balancesOfAccountByDenom: [Balance] @derivedFrom(field: "lastUpdatedBlock")
+  supplies: [Supply] @derivedFrom(field: "block")
 }
 
 type Transaction @entity {
@@ -124,5 +125,7 @@ type GenesisFile @entity {
 type Supply @entity {
   # coin denom
   id: ID!
+  denom: String! @index
   amount: BigInt!
+  block: Block!
 }

src/mappings/bank/supply.ts

@@ -1,8 +1,19 @@
+import type {
+  Coin,
+  CosmosBlock,
+} from "@subql/types-cosmos";
 import type { QueryTotalSupplyResponse } from "cosmjs-types/cosmos/bank/v1beta1/query";
-import { Coin } from "cosmjs-types/cosmos/base/v1beta1/coin";
+import { Supply } from "../../types";
 import { stringify } from "../utils";
 
-export async function getSupply(): Promise<Coin[]> {
+export const getSupplyRecord = (supply: Coin, block: CosmosBlock): Supply => Supply.create({
+  id: `${supply.denom}@${block.block.id}`,
+  denom: supply.denom,
+  amount: BigInt(supply.amount),
+  blockId: block.block.id,
+});
+
+export async function queryTotalSupply(): Promise<Coin[]> {
   let finalSupply: Coin[] = [];
   let paginationKey: Uint8Array | undefined;
 

src/mappings/primitives.ts

@@ -20,11 +20,13 @@ import {
   GenesisFile as GenesisEntity,
   Message,
   NativeBalanceChange,
-  Supply,
   Transaction,
   TxStatus,
 } from "../types";
-import { getSupply } from "./bank/supply";
+import {
+  getSupplyRecord,
+  queryTotalSupply,
+} from "./bank/supply";
 import { PREFIX } from "./constants";
 import type { Genesis } from "./types/genesis";
 import {
@@ -57,12 +59,6 @@ export async function handleGenesis(block: CosmosBlock): Promise<void> {
           chainId: block.block.header.chainId,
         };
       })),
-      store.bulkCreate("Supply", genesis.app_state.bank.supply.map(supply => {
-        return {
-          id: supply.denom,
-          amount: BigInt(supply.amount),
-        };
-      })),
       Event.create({
         id: "genesis",
         type: "genesis",
@@ -173,7 +169,7 @@ async function _handleBlock(block: CosmosBlock): Promise<void> {
 
   await blockEntity.save();
 
-  await _handleSupply();
+  await _handleSupply(block);
 }
 
 async function _handleTransaction(tx: CosmosTransaction): Promise<void> {
@@ -285,29 +281,15 @@ async function _handleEvent(event: CosmosEvent): Promise<void> {
   }
 }
 
-async function _handleSupply(): Promise<void> {
-  // we need to check if the amount is different to avoid duplicate records because subquery does not check that
-  const totalSupply = await getSupply();
-  if (totalSupply.length > 0) {
-    logger.info(`[handleSupply]: indexing total supply for ${totalSupply.length} coins`);
-    for (const supply of totalSupply) {
-      logger.debug(`[handleSupply] (totalSupply.coin: total supply of ${supply.denom}`);
-      // we need to check if the amount is different to avoid duplicate records because subquery does not check that
-      const s = await Supply.get(supply.denom);
-      if (!s) {
-        await Supply.create({
-          id: supply.denom,
-          amount: BigInt(supply.amount),
-        }).save();
-        continue;
-      } else if (s.amount.toString() === supply.amount) {
-        continue;
-      }
-
-      s.amount = BigInt(supply.amount);
-      await s.save();
-    }
+async function _handleSupply(block: CosmosBlock): Promise<void> {
+  const totalSupply = await queryTotalSupply();
+  if (totalSupply.length === 0) {
+    logger.warn(`[handleSupply]: no total supply found`);
+    return;
   }
+
+  logger.info(`[handleSupply]: indexing total supply for ${totalSupply.length} coins`);
+  await store.bulkCreate("Supply", totalSupply.map(supply => getSupplyRecord(supply, block)));
 }
 
 // eslint-disable-next-line @typescript-eslint/no-unused-vars