Merge pull request #74 from duckietm/Dev

🆙  more optimization mem.

src/api/room/widgets/ChatBubbleUtilities.ts

@@ -2,6 +2,8 @@ import { AvatarFigurePartType, AvatarScaleType, AvatarSetType, GetAvatarRenderMa
 
 export class ChatBubbleUtilities
 {
+    private static MAX_CACHE_SIZE: number = 200;
+
     public static AVATAR_COLOR_CACHE: Map<string, number> = new Map();
     public static AVATAR_IMAGE_CACHE: Map<string, string> = new Map();
     public static PET_IMAGE_CACHE: Map<string, string> = new Map();
@@ -9,6 +11,19 @@ export class ChatBubbleUtilities
 
     private static placeHolderImageUrl: string = '';
 
+    private static pruneCache<T>(cache: Map<string, T>, maxSize: number = ChatBubbleUtilities.MAX_CACHE_SIZE): void
+    {
+        if(cache.size <= maxSize) return;
+
+        const deleteCount = cache.size - maxSize;
+        const iterator = cache.keys();
+
+        for(let i = 0; i < deleteCount; i++)
+        {
+            cache.delete(iterator.next().value as string);
+        }
+    }
+
     public static async setFigureImage(figure: string): Promise<string>
     {
         const avatarImage = GetAvatarRenderManager().createAvatarImage(figure, AvatarScaleType.LARGE, null, {
@@ -34,6 +49,9 @@ export class ChatBubbleUtilities
         this.AVATAR_COLOR_CACHE.set(figure, ((color && color.rgb) || 16777215));
         this.AVATAR_IMAGE_CACHE.set(figure, imageUrl);
 
+        this.pruneCache(this.AVATAR_COLOR_CACHE);
+        this.pruneCache(this.AVATAR_IMAGE_CACHE);
+
         avatarImage.dispose();
 
         return imageUrl;
@@ -100,7 +118,11 @@ export class ChatBubbleUtilities
 
             if(!resolvedImage) resolvedImage = await getImageUrl(imageResult);
 
-            if(resolvedImage) this.PET_IMAGE_CACHE.set(cacheKey, resolvedImage);
+            if(resolvedImage)
+            {
+                this.PET_IMAGE_CACHE.set(cacheKey, resolvedImage);
+                this.pruneCache(this.PET_IMAGE_CACHE);
+            }
 
             return resolvedImage;
         })();