Add totalBalance

schema.graphql

@@ -288,6 +288,8 @@ type ClassicPosition @entity {
   vaultBalance: BigInt!
   "The amount of vault shares the investor holds in a boost"
   boostBalance: BigInt!
+  "Total amount of vault shares the investor holds. Should always equal vaultBalance + boostBalance. This is mostly used for filtering."
+  totalBalance: BigInt!
 
   "All investor position interactions"
   interactions: [ClassicPositionInteraction!]! @derivedFrom(field: "investorPosition")
@@ -330,6 +332,8 @@ type ClassicPositionInteraction @entity(immutable: true) {
   vaultBalance: BigInt!
   "The amount of vault shares the investor holds in a boost at the time of the interaction"
   boostBalance: BigInt!
+  "Total amount of vault shares the investor holds. Should always equal vaultBalance + boostBalance. This is mostly used for filtering."
+  totalBalance: BigInt!
 
   "Amount of vault shares change in the interaction"
   vaultBalanceDelta: BigInt!
@@ -642,6 +646,8 @@ type ClmPosition @entity {
   managerBalance: BigInt!
   "Amount of reward pool shares the investor holds"
   rewardPoolBalance: BigInt!
+  "Total amount of CLM shares the investor holds. Should always equal to managerBalance + rewardPoolBalance. This is mostly used for filtering."
+  totalBalance: BigInt!
 
   "All investor position interactions"
   interactions: [ClmPositionInteraction!]! @derivedFrom(field: "investorPosition")
@@ -686,6 +692,8 @@ type ClmPositionInteraction @entity(immutable: true) {
   managerBalance: BigInt!
   "The amount of reward pool shares the investor holds at the time of the interaction"
   rewardPoolBalance: BigInt!
+  "Total amount of CLM shares the investor holds. Should always equal to managerBalance + rewardPoolBalance. This is mostly used for filtering."
+  totalBalance: BigInt!
   "The amount of first underlying tokens the investor is entitled to at the time of the interaction"
   underlyingBalance0: BigInt!
   "The amount of second underlying tokens the investor is entitled to at the time of the interaction"

src/classic/entity/position.ts

@@ -19,6 +19,7 @@ export function getClassicPosition(classic: Classic, investor: Investor): Classi
     position.createdWith = ADDRESS_ZERO
     position.vaultBalance = ZERO_BI
     position.boostBalance = ZERO_BI
+    position.totalBalance = ZERO_BI
   }
   return position
 }

src/classic/interaction.ts

@@ -66,6 +66,7 @@ function updateUserPosition(
   }
   position.vaultBalance = position.vaultBalance.plus(vaultBalanceDelta)
   position.boostBalance = position.boostBalance.plus(boostBalanceDelta)
+  position.totalBalance = position.vaultBalance.plus(position.boostBalance)
   position.save()
 
   ///////
@@ -94,6 +95,7 @@ function updateUserPosition(
   }
   interaction.vaultBalance = position.vaultBalance
   interaction.boostBalance = position.boostBalance
+  interaction.totalBalance = position.totalBalance
   interaction.vaultBalanceDelta = vaultBalanceDelta
   interaction.boostBalanceDelta = boostBalanceDelta
 

src/clm/entity/position.ts

@@ -19,6 +19,7 @@ export function getClmPosition(clm: CLM, investor: Investor): ClmPosition {
     position.createdWith = ADDRESS_ZERO
     position.managerBalance = ZERO_BI
     position.rewardPoolBalance = ZERO_BI
+    position.totalBalance = ZERO_BI
   }
   return position
 }

src/clm/interaction.ts

@@ -115,6 +115,7 @@ function updateUserPosition(
   }
   position.managerBalance = position.managerBalance.plus(managerBalanceDelta)
   position.rewardPoolBalance = position.rewardPoolBalance.plus(rewardPoolBalanceDelta)
+  position.totalBalance = position.managerBalance.plus(position.rewardPoolBalance)
   position.save()
 
   ///////
@@ -143,6 +144,7 @@ function updateUserPosition(
   }
   interaction.managerBalance = position.managerBalance
   interaction.rewardPoolBalance = position.rewardPoolBalance
+  interaction.totalBalance = position.totalBalance
   interaction.managerBalanceDelta = managerBalanceDelta
   interaction.rewardPoolBalanceDelta = rewardPoolBalanceDelta