Merge pull request #70 from Giveth/hotfix/token-disto-balance-user-re…

…lation Hotfix in TokenDistroBalance
Giveth · Mar 8, 2024 · 3e9895b · 3e9895b
2 parents c10889a + aa7fd80
commit 3e9895b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 10 deletions.
diff --git a/src/mappings/tokenDistro.ts b/src/mappings/tokenDistro.ts
@@ -49,19 +49,10 @@ export function handleChangeAddress(event: ChangeAddress): void {
     event.address.toHex(),
     event.params.oldAddress.toHex(),
   );
-  if (!oldBalance) {
-    log.debug('Change Address oldAddress {} balance is null!', [
-      event.params.oldAddress.toHex(),
-    ]);
-    return;
-  }
-  let newBalance = getTokenDistroBalance(
+  const newBalance = getTokenDistroBalance(
     event.address.toHex(),
     event.params.newAddress.toHex(),
   );
-  if (!newBalance) {
-    newBalance = new TokenDistroBalance(event.params.newAddress.toHex());
-  }
 
   // New Address allocatedTokens amount should be zero
   newBalance.allocatedTokens = newBalance.allocatedTokens.plus(

diff --git a/src/utils/tokenDistroHelper.ts b/src/utils/tokenDistroHelper.ts
@@ -6,6 +6,7 @@ import {
 } from '../types/schema';
 import { Address, BigInt, log } from '@graphprotocol/graph-ts';
 import { TokenDistro as TokenDistroContract } from '../types/TokenDistro/TokenDistro';
+import { getUserEntity } from './misc';
 export function saveTokenAllocation(
   recipient: string,
   txHash: string,
@@ -39,6 +40,8 @@ export function getTokenDistroBalance(
   const id = tokenDistro + '-' + userAddress;
   let tokenDistroBalance = TokenDistroBalance.load(id);
 
+  getUserEntity(Address.fromString(userAddress));
+
   if (!tokenDistroBalance) {
     tokenDistroBalance = new TokenDistroBalance(id);
     tokenDistroBalance.user = userAddress;