add nfts in collection

gear-foundation · Mar 21, 2024 · f56346e · f56346e
1 parent aa5ac0f
commit f56346e
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/indexer/src/server-extension/model/nfts-in-collection.model.ts b/indexer/src/server-extension/model/nfts-in-collection.model.ts
@@ -9,6 +9,6 @@ export class NftsInCollection {
   @Field(() => String)
   collection!: string;
 
-  @Field(() => String)
-  count!: string;
+  @Field(() => Number)
+  count!: number;
 }
diff --git a/indexer/src/server-extension/resolvers/nfts-in-collection.resolver.ts b/indexer/src/server-extension/resolvers/nfts-in-collection.resolver.ts
@@ -23,6 +23,17 @@ export class NftsInCollectionResolver {
       .select('count(nft.id) as count, collection_id as collection')
       .where('collection_id in (:...collections)', { collections })
       .groupBy('collection_id');
-    return query.getRawMany();
+    const result = await query.getRawMany<{
+      count: number;
+      collection: string;
+    }>();
+    const collectionMap = result.reduce((acc, { collection, count }) => {
+      acc[collection] = count;
+      return acc;
+    }, {} as Record<string, number>);
+    return collections.map((collection) => ({
+      collection,
+      count: collectionMap[collection] || 0,
+    }));
   }
 }