chore: optimize entity loading by reducing one entity

Author: kemuru

subgraph/core-university/src/SortitionModule.ts

@@ -1,25 +1,24 @@
 import { SortitionModule, StakeLocked, StakeSet as StakeSetEvent } from "../generated/SortitionModule/SortitionModule";
 import { StakeSet as StakeSetEntity } from "../generated/schema";
 
-import { ensureJurorTokensPerCourt, updateJurorDelayedStake, updateJurorStake } from "./entities/JurorTokensPerCourt";
+import { updateJurorDelayedStake, updateJurorStake } from "./entities/JurorTokensPerCourt";
 import { ensureUser } from "./entities/User";
 import { ZERO } from "./utils";
 
 export function handleStakeSet(event: StakeSetEvent): void {
   const jurorAddress = event.params._address.toHexString();
-  const juror = ensureUser(jurorAddress);
   const courtID = event.params._courtID.toString();
 
   updateJurorStake(jurorAddress, courtID.toString(), SortitionModule.bind(event.address), event.block.timestamp);
   //stake is updated instantly so no delayed amount, set delay amount to zero
   updateJurorDelayedStake(jurorAddress, courtID, ZERO);
 
   const stakeSet = new StakeSetEntity(event.transaction.hash.toHex() + "-" + event.logIndex.toString());
-  const jurorTokensPerCourt = ensureJurorTokensPerCourt(jurorAddress, "1");
+  const juror = ensureUser(jurorAddress);
   stakeSet.juror = juror.id;
   stakeSet.courtID = event.params._courtID;
   stakeSet.stake = event.params._amount;
-  stakeSet.newTotalStake = jurorTokensPerCourt.effectiveStake;
+  stakeSet.newTotalStake = juror.totalStake;
   stakeSet.blocknumber = event.block.number;
   stakeSet.timestamp = event.block.timestamp;
   stakeSet.logIndex = event.logIndex;

subgraph/core/src/SortitionModule.ts

@@ -8,7 +8,7 @@ import {
 } from "../generated/SortitionModule/SortitionModule";
 import { StakeSet as StakeSetEntity } from "../generated/schema";
 
-import { ensureJurorTokensPerCourt, updateJurorDelayedStake, updateJurorStake } from "./entities/JurorTokensPerCourt";
+import { updateJurorDelayedStake, updateJurorStake } from "./entities/JurorTokensPerCourt";
 import { ensureUser } from "./entities/User";
 import { ZERO } from "./utils";
 
@@ -26,19 +26,18 @@ export function handleStakeDelayedNotTransferred(event: StakeDelayedNotTransferr
 
 export function handleStakeSet(event: StakeSetEvent): void {
   const jurorAddress = event.params._address.toHexString();
-  const juror = ensureUser(jurorAddress);
   const courtID = event.params._courtID.toString();
 
   updateJurorStake(jurorAddress, courtID.toString(), SortitionModule.bind(event.address), event.block.timestamp);
   //stake is updated instantly so no delayed amount, set delay amount to zero
   updateJurorDelayedStake(jurorAddress, courtID, ZERO);
 
   const stakeSet = new StakeSetEntity(event.transaction.hash.toHex() + "-" + event.logIndex.toString());
-  const jurorTokensPerCourt = ensureJurorTokensPerCourt(jurorAddress, "1");
+  const juror = ensureUser(jurorAddress);
   stakeSet.juror = juror.id;
   stakeSet.courtID = event.params._courtID;
   stakeSet.stake = event.params._amount;
-  stakeSet.newTotalStake = jurorTokensPerCourt.effectiveStake;
+  stakeSet.newTotalStake = juror.totalStake;
   stakeSet.blocknumber = event.block.number;
   stakeSet.timestamp = event.block.timestamp;
   stakeSet.logIndex = event.logIndex;