Added valid non-found in subgraph error (vs schema error) and removed…

… wasteful retries from wallet function getRegisteredStealthKeys
ScopeLift · May 2, 2024 · 6d831ac · 6d831ac
1 parent bc1f6a8
commit 6d831ac
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 14 deletions.
diff --git a/frontend/src/store/wallet.ts b/frontend/src/store/wallet.ts
@@ -631,19 +631,11 @@ const hasSetPublicKeysLegacy = async (name: string, provider: Provider) => {
 
 // Helper method to check if user has registered public keys in the StealthKeyRegistry
 async function getRegisteredStealthKeys(account: string, provider: Provider) {
-  let retryCounter = 0;
-  while (retryCounter < 3) {
-    try {
-      console.log(`getting stealth keys for ${account}, try ${retryCounter + 1} of 3`);
-      const stealthPubKeys = await utils.lookupRecipient(account, provider); // throws if no keys found
-      return stealthPubKeys;
-    } catch (err) {
-      window.logger.warn(err);
-      retryCounter++;
-      if (retryCounter < 3) {
-        await new Promise((resolve) => setTimeout(resolve, 2000)); // Wait for 2 seconds
-      }
-    }
+  try {
+    const stealthPubKeys = await utils.lookupRecipient(account, provider); // throws if no keys found
+    return stealthPubKeys;
+  } catch (err) {
+    window.logger.warn(err);
+    return null;
   }
-  return null;
 }
diff --git a/umbra-js/src/utils/utils.ts b/umbra-js/src/utils/utils.ts
@@ -340,6 +340,7 @@ export async function getMostRecentSubgraphStealthKeyChangedEventFromAddress(
   }
 
   if (!theEvent) {
+    console.log(`Searched the subgraph, but found no StealthKeyChangedEvents for address ${address}`);
     throw new Error('No stealthKeyChangedEvents found matching address in subgraph');
   }
   return theEvent;