Merge pull request #435 from hack-a-chain-software/total-count-blocks-from-height

feat: added total count in blocksFromHeight query

Author: mschead

indexer/src/kadena-server/config/graphql-types.ts

@@ -1008,6 +1008,7 @@ export type QueryBlocksFromHeightConnection = {
   __typename?: 'QueryBlocksFromHeightConnection';
   edges: Array<QueryBlocksFromHeightConnectionEdge>;
   pageInfo: PageInfo;
+  totalCount: Scalars['Int']['output'];
 };
 
 export type QueryBlocksFromHeightConnectionEdge = {
@@ -3061,6 +3062,7 @@ export type QueryBlocksFromHeightConnectionResolvers<
     ContextType
   >;
   pageInfo?: Resolver<ResolversTypes['PageInfo'], ParentType, ContextType>;
+  totalCount?: Resolver<ResolversTypes['Int'], ParentType, ContextType>;
   __isTypeOf?: IsTypeOfResolverFn<ParentType, ContextType>;
 };
 

indexer/src/kadena-server/config/schema.graphql

@@ -556,6 +556,7 @@ type QueryEventsConnectionEdge {
 type QueryBlocksFromHeightConnection {
   edges: [QueryBlocksFromHeightConnectionEdge!]!
   pageInfo: PageInfo!
+  totalCount: Int!
 }
 
 type QueryBlocksFromHeightConnectionEdge {

indexer/src/kadena-server/resolvers/fields/query-blocks-from-height-connection/total-count-query-blocks-from-height-connection-resolver.ts

@@ -0,0 +1,63 @@
+/**
+ * Resolver for the totalCount field of the QueryEventsConnection type.
+ * This module counts blockchain events based on various filter criteria.
+ */
+import { ResolverContext } from '../../../config/apollo-server-config';
+import { QueryBlocksFromHeightConnectionResolvers } from '../../../config/graphql-types';
+import zod from 'zod';
+
+/**
+ * Zod schema for validating event query parameters.
+ * Defines the structure and types of filter parameters used to count events.
+ * Note that qualifiedEventName is required, while other filters are optional.
+ */
+const schema = zod.object({
+  chainIds: zod.array(zod.string()).optional(),
+  startHeight: zod.number(),
+  endHeight: zod.number().nullable().optional(),
+});
+
+/**
+ * Resolver function for the totalCount field of the QueryEventsConnection type.
+ * Retrieves the total count of events matching the provided filter criteria.
+ *
+ * @param parent - The parent object containing filter parameters
+ * @param _args - GraphQL arguments (unused in this resolver)
+ * @param context - The resolver context containing repositories and services
+ * @returns The total count of events matching the criteria
+ */
+export const totalCountQueryBlocksFromHeightConnectionResolver: QueryBlocksFromHeightConnectionResolvers<ResolverContext>['totalCount'] =
+  async (parent, _args, context) => {
+    const { startHeight, endHeight, chainIds: chainIdsParam } = schema.parse(parent);
+
+    if (endHeight && startHeight > endHeight) {
+      throw new Error('Start height cannot be greater than end height');
+    }
+
+    let chainIds = [];
+    if (chainIdsParam?.length) {
+      chainIds = [...chainIdsParam];
+    } else {
+      const networkInfo = await context.networkRepository.getAllInfo();
+      chainIds = [...networkInfo.nodeChains];
+    }
+
+    const currentChainHeights = await context.networkRepository.getCurrentChainHeights();
+    const startHeightOfChainsFrom9to19 = 852054;
+    const totalCount = chainIds.reduce((acum, chainId) => {
+      const endHeightToUse = endHeight ?? currentChainHeights[chainId];
+      let totalOfChain = endHeightToUse - startHeight + 1;
+      if (Number(chainId) > 9) {
+        if (startHeight >= startHeightOfChainsFrom9to19) {
+          totalOfChain = endHeightToUse - startHeight + 1;
+        } else if (endHeightToUse < startHeightOfChainsFrom9to19) {
+          totalOfChain = 0;
+        } else {
+          totalOfChain = endHeightToUse - startHeightOfChainsFrom9to19 + 1;
+        }
+      }
+      return totalOfChain + acum;
+    }, 0);
+
+    return totalCount;
+  };

indexer/src/kadena-server/resolvers/index.ts

@@ -101,6 +101,7 @@ import { transfersNonFungibleAccountResolver } from '@/kadena-server/resolvers/f
 import { totalCountNonFungibleAccountTransfersConnectionResolver } from '@/kadena-server/resolvers/fields/non-fungible-account/transfers-connection/total-count-non-fungible-account-transfers-connection-resolver';
 import { transfersNonFungibleChainAccountResolver } from '@/kadena-server/resolvers/fields/non-fungible-chain-account/transfers-non-fungible-chain-account-resolver';
 import { totalCountNonFungibleChainAccountTransfersConnectionResolver } from '@/kadena-server/resolvers/fields/non-fungible-chain-account/transfers-connection/total-count-non-fungible-chain-account-transfers-connection-resolver';
+import { totalCountQueryBlocksFromHeightConnectionResolver } from '@/kadena-server/resolvers/fields/query-blocks-from-height-connection/total-count-query-blocks-from-height-connection-resolver';
 /**
  * Complete resolver map for the GraphQL API
  *
@@ -238,6 +239,9 @@ export const resolvers: Resolvers<ResolverContext> = {
     signers: signersTransactionCommandResolver, // add dataloader
     meta: metaTransactionCommandResolver, // add dataloader
   },
+  QueryBlocksFromHeightConnection: {
+    totalCount: totalCountQueryBlocksFromHeightConnectionResolver,
+  },
   QueryBlocksFromDepthConnection: {
     totalCount: totalCountQueryBlocksFromDepthConnectionResolver,
   },

indexer/src/kadena-server/resolvers/query/blocks-from-height-query-resolver.ts

@@ -28,12 +28,24 @@ import { buildBlockOutput } from '../output/build-block-output';
  */
 export const blocksFromHeightQueryResolver: QueryResolvers<ResolverContext>['blocksFromHeight'] =
   async (_parent, args, context) => {
+    if (args.endHeight && args.startHeight > args.endHeight) {
+      throw new Error('Start height cannot be greater than end height');
+    }
+
     const output = await context.blockRepository.getBlocksBetweenHeights(args);
 
     const edges = output.edges.map(e => ({
       cursor: e.cursor,
       node: buildBlockOutput(e.node),
     }));
 
-    return { edges, pageInfo: output.pageInfo };
+    return {
+      edges,
+      pageInfo: output.pageInfo,
+
+      startHeight: args.startHeight,
+      endHeight: args.endHeight,
+      chainIds: args.chainIds,
+      totalCount: -1,
+    };
   };