getCachedCollection cache

Expensify · Aug 1, 2024 · cb15a80 · cb15a80
1 parent b2e5384
commit cb15a80
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/lib/OnyxUtils.ts b/lib/OnyxUtils.ts
@@ -804,6 +804,9 @@ function keyChanged<TKey extends OnyxKey>(
             return;
         }
     }
+
+    const cachedCollections: Record<string, ReturnType<typeof getCachedCollection>> = {};
+
     for (let i = 0; i < stateMappingKeys.length; i++) {
         const subscriber = callbackToStateMapping[stateMappingKeys[i]];
         if (!subscriber || !isKeyMatch(subscriber.key, key) || !canUpdateSubscriber(subscriber)) {
@@ -820,7 +823,7 @@ function keyChanged<TKey extends OnyxKey>(
             }
 
             if (isCollectionKey(subscriber.key) && subscriber.waitForCollectionCallback) {
-                const cachedCollection = getCachedCollection(subscriber.key);
+                const cachedCollection = cachedCollections[subscriber.key] ?? getCachedCollection(subscriber.key);
 
                 cachedCollection[key] = value;
                 subscriber.callback(cachedCollection);