Delete schema's stakeType field from StakeData

A stake can have different stake types at the same time (T, NuInTStake and KeepInTStake), so the StakeType's field can be misleading.
threshold-network · Jun 8, 2022 · eeeac19 · eeeac19
1 parent 61b986a
commit eeeac19
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 9 deletions.
diff --git a/schema.graphql b/schema.graphql
@@ -1,9 +1,3 @@
-enum StakeType {
-  NU
-  KEEP
-  T
-}
-
 type Account @entity {
   # Id is the ethereum address of the account.
   id: ID!
@@ -16,7 +10,6 @@ type StakeData @entity {
   # ID is the staking provider address
   id: ID!
   owner: Account!
-  stakeType: StakeType!
   beneficiary: Bytes!
   authorizer: Bytes!
   tStake: BigInt!

diff --git a/src/mapping.ts b/src/mapping.ts
@@ -51,13 +51,12 @@ export function handleStaked(event: Staked): void {
     account.save()
   }
 
-  stakeData.stakeType = type
   stakeData.owner = account.id
   stakeData.beneficiary = event.params.beneficiary
   stakeData.authorizer = event.params.authorizer
   stakeData.tStake = type === "T" ? stakeAmount : BigInt.zero()
-  stakeData.nuInTStake = type === "NU" ? stakeAmount : BigInt.zero()
   stakeData.keepInTStake = type === "KEEP" ? stakeAmount : BigInt.zero()
+  stakeData.nuInTStake = type === "NU" ? stakeAmount : BigInt.zero()
   stakeData.totalStaked = stakeData.tStake
     .plus(stakeData.keepInTStake)
     .plus(stakeData.nuInTStake)