Add support for fetching user balances for multiple chains

beethovenxfi · Oct 16, 2023 · 1fb4c7b · 1fb4c7b
1 parent 725d93f
commit 1fb4c7b
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 5 deletions.
diff --git a/modules/user/lib/user-balance.service.ts b/modules/user/lib/user-balance.service.ts
@@ -14,10 +14,10 @@ export class UserBalanceService {
             where: { address: address.toLowerCase() },
             include: {
                 walletBalances: {
-                    where: { chain: networkContext.chain, poolId: { not: null }, balanceNum: { gt: 0 } },
+                    where: { poolId: { not: null }, balanceNum: { gt: 0 } },
                 },
                 stakedBalances: {
-                    where: { chain: networkContext.chain, poolId: { not: null }, balanceNum: { gt: 0 } },
+                    where: { poolId: { not: null }, balanceNum: { gt: 0 } },
                 },
             },
         });
@@ -43,6 +43,8 @@ export class UserBalanceService {
                 totalBalance: formatFixed(stakedNum.add(walletNum), 18),
                 stakedBalance: stakedBalance?.balance || '0',
                 walletBalance: walletBalance?.balance || '0',
+                // the prisma query above ensures that one of these balances exists
+                chain: (stakedBalance?.chain || walletBalance?.chain)!,
             };
         });
     }
@@ -68,6 +70,7 @@ export class UserBalanceService {
             totalBalance: formatFixed(stakedNum.add(walletNum), 18),
             stakedBalance: stakedBalance?.balance || '0',
             walletBalance: walletBalance?.balance || '0',
+            chain: networkContext.chain,
         };
     }
 

diff --git a/modules/user/user-types.ts b/modules/user/user-types.ts
@@ -1,5 +1,5 @@
 import { AmountHumanReadable } from '../common/global-types';
-import { PrismaPoolStaking, PrismaPoolStakingType } from '@prisma/client';
+import { Chain, PrismaPoolStaking, PrismaPoolStakingType } from '@prisma/client';
 import { Relic } from '../subgraphs/reliquary-subgraph/generated/reliquary-subgraph-types';
 
 export interface UserStakedBalanceService {
@@ -14,6 +14,7 @@ export interface UserPoolBalance {
     totalBalance: AmountHumanReadable;
     walletBalance: AmountHumanReadable;
     stakedBalance: AmountHumanReadable;
+    chain: Chain;
 }
 
 export interface UserSyncUserBalanceInput {

diff --git a/modules/user/user.gql b/modules/user/user.gql
@@ -1,5 +1,5 @@
 extend type Query {
-    userGetPoolBalances: [GqlUserPoolBalance!]!
+    userGetPoolBalances(chains: [GqlChain!]!): [GqlUserPoolBalance!]!
     userGetStaking: [GqlPoolStaking!]!
     userGetPoolJoinExits(first: Int = 10, skip: Int = 0, poolId: String!): [GqlPoolJoinExit!]!
     userGetSwaps(first: Int = 10, skip: Int = 0, poolId: String!): [GqlPoolSwap!]!
@@ -24,4 +24,5 @@ type GqlUserPoolBalance {
     totalBalance: AmountHumanReadable!
     walletBalance: AmountHumanReadable!
     stakedBalance: AmountHumanReadable!
+    chain: GqlChain!
 }
diff --git a/modules/user/user.resolvers.ts b/modules/user/user.resolvers.ts
@@ -5,7 +5,7 @@ import { tokenService } from '../token/token.service';
 
 const resolvers: Resolvers = {
     Query: {
-        userGetPoolBalances: async (parent, {}, context) => {
+        userGetPoolBalances: async (parent, { chains }, context) => {
             const accountAddress = getRequiredAccountAddress(context);
             const tokenPrices = await tokenService.getTokenPrices();
             const balances = await userService.getUserPoolBalances(accountAddress);