From 5707801d6e7eefca19c12acf09da7075384c1acc Mon Sep 17 00:00:00 2001
From: Boldizsar Mezei <boldizsar@unchained.work>
Date: Fri, 10 May 2024 22:48:26 +0200
Subject: [PATCH] Fixes

---
 .../functions/scripts/dbUpgrades/soon.snapshot.ts  | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/packages/functions/scripts/dbUpgrades/soon.snapshot.ts b/packages/functions/scripts/dbUpgrades/soon.snapshot.ts
index bb389d5b5..dc2d15786 100644
--- a/packages/functions/scripts/dbUpgrades/soon.snapshot.ts
+++ b/packages/functions/scripts/dbUpgrades/soon.snapshot.ts
@@ -1,5 +1,5 @@
 import { MilestoneTransactions, database } from '@buildcore/database';
-import { COL, Network, SUB_COL, TokenDropStatus } from '@buildcore/interfaces';
+import { COL, Member, Network, SUB_COL, TokenDropStatus } from '@buildcore/interfaces';
 import {
   Address,
   AddressType,
@@ -38,7 +38,8 @@ export const soonSnapshot = async (tokenId?: string, network?: Network.SMR | Net
   MILESTONE_COL = NETWORK === Network.SMR ? COL.MILESTONE_SMR : COL.MILESTONE_RMS;
   tokensPerAddress = {};
 
-  await database().getCon()(COL.SOON_SNAP).delete();
+  const deleted = await database().getCon()(COL.SOON_SNAP).delete();
+  console.log('deleted', deleted);
 
   await createSoonSnapshot();
   const tokensPerMember = await addressesToMembers();
@@ -100,7 +101,7 @@ const createSoonSnapshot = async () => {
 const addressesToMembers = async () => {
   const tokensPerMember: { [key: string]: number } = {};
 
-  const batches = chunk(Object.entries(tokensPerAddress), 100);
+  const batches = chunk(Object.entries(tokensPerAddress), 10);
   let count = 0;
   console.log('addressesToMembers', 'total', Object.entries(tokensPerAddress).length);
 
@@ -119,17 +120,22 @@ const addressesToMembers = async () => {
   return tokensPerMember;
 };
 
+const members: { [key: string]: Member } = {};
+
 const addressToMember = async (address: string, count: number) => {
   const tokensPerMember: { [key: string]: number } = {};
+  console.log('addressToMember', 'getting airdrops', address);
   const airdropsSnap = await database()
     .collection(COL.AIRDROP)
     .where('sourceAddress', '==', address)
     .where('status', '==', TokenDropStatus.UNCLAIMED)
     .get();
+  console.log('addressToMember', 'airdrops', airdropsSnap.length);
 
   for (const airdrop of airdropsSnap) {
     if (airdrop.member.startsWith('0x')) {
-      const member = await database().doc(COL.MEMBER, airdrop.member).get();
+      const member =
+        members[airdrop.member] || (await database().doc(COL.MEMBER, airdrop.member).get());
       const memberAddress = member?.validatedAddress?.[NETWORK] || airdrop.member;
       tokensPerMember[memberAddress] = (tokensPerMember[memberAddress] || 0) + airdrop.count;
       continue;