all recipients in parallel

Author: benesjan

yarn-project/pxe/src/contract_function_simulator/pxe_oracle_interface.ts

@@ -321,32 +321,35 @@ export class PXEOracleInterface implements ExecutionDataProvider {
     // Determine recipients: use scopes if provided, otherwise get all accounts
     const recipients = scopes && scopes.length > 0 ? scopes : await this.keyStore.getAccounts();
 
-    // Process each recipient
-    for (const recipient of recipients) {
-      // Get all secrets for this recipient (one per sender)
-      const secrets = await this.#getSecretsForSenders(contractAddress, recipient);
-
-      // Load logs for all sender-recipient pairs in parallel
-      const logArrays = await Promise.all(
-        secrets.map(secret =>
-          loadPrivateLogsForSenderRecipientPair(
-            secret,
-            contractAddress,
-            this.aztecNode,
-            this.newRecipientTaggingDataProvider,
-            anchorBlockNumber,
+    // For each recipient, fetch secrets, load logs, and store them.
+    // We run these per-recipient tasks in parallel so that logs are loaded for all recipients concurrently.
+    await Promise.all(
+      recipients.map(async recipient => {
+        // Get all secrets for this recipient (one per sender)
+        const secrets = await this.#getSecretsForSenders(contractAddress, recipient);
+
+        // Load logs for all sender-recipient pairs in parallel
+        const logArrays = await Promise.all(
+          secrets.map(secret =>
+            loadPrivateLogsForSenderRecipientPair(
+              secret,
+              contractAddress,
+              this.aztecNode,
+              this.newRecipientTaggingDataProvider,
+              anchorBlockNumber,
+            ),
           ),
-        ),
-      );
+        );
 
-      // Flatten all logs from all secrets
-      const allLogs = logArrays.flat();
+        // Flatten all logs from all secrets
+        const allLogs = logArrays.flat();
 
-      // Store the logs for this recipient
-      if (allLogs.length > 0) {
-        await this.#storePendingTaggedLogs(contractAddress, pendingTaggedLogArrayBaseSlot, recipient, allLogs);
-      }
-    }
+        // Store the logs for this recipient
+        if (allLogs.length > 0) {
+          await this.#storePendingTaggedLogs(contractAddress, pendingTaggedLogArrayBaseSlot, recipient, allLogs);
+        }
+      }),
+    );
   }
 
   async #storePendingTaggedLogs(