Cache avatars based on just type and id (#259)

Joystream · Nov 29, 2023 · c9f3739 · c9f3739
1 parent 94503ff
commit c9f3739
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/utils/notification/notificationAvatars.ts b/src/utils/notification/notificationAvatars.ts
@@ -6,10 +6,10 @@ import { getAssetUrls } from '../../server-extension/resolvers/AssetsResolver/ut
 import { ConfigVariable, config } from '../config'
 
 export const getNotificationAvatar = memoize(
-  async (em: EntityManager, type: 'channelId' | 'membershipId', param: string): Promise<string> => {
+  async (em: EntityManager, type: 'channelId' | 'membershipId', id: string): Promise<string> => {
     switch (type) {
       case 'channelId': {
-        const channel = await em.getRepository(Channel).findOneBy({ id: param })
+        const channel = await em.getRepository(Channel).findOneBy({ id })
         const objectId = channel?.avatarPhotoId
         if (!objectId) break
 
@@ -21,7 +21,7 @@ export const getNotificationAvatar = memoize(
       }
 
       case 'membershipId': {
-        const member = await em.getRepository(MemberMetadata).findOneBy({ id: param })
+        const member = await em.getRepository(MemberMetadata).findOneBy({ id })
         const avatar = member?.avatar
 
         // AvatarObject is not yet supported
@@ -34,5 +34,6 @@ export const getNotificationAvatar = memoize(
     // Fallback to a placeholder
     const notificationAssetRoot = await config.get(ConfigVariable.AppAssetStorage, em)
     return `${notificationAssetRoot}/placeholder/avatar.png`
-  }
+  },
+  (_, type, id) => `${type}:${id}`
 )