From 11cc5cbcab362048644b873a10d0ea789a317d71 Mon Sep 17 00:00:00 2001
From: Jay Malhotra <5047192+SapiensAnatis@users.noreply.github.com>
Date: Wed, 13 Mar 2024 21:45:08 +0000
Subject: [PATCH] Optimize UnitRepository duplicate checking (#719)

- Only load up relevant rows
---
 .../Repositories/UnitRepository.cs            | 28 +++++++++++++++++--
 1 file changed, 25 insertions(+), 3 deletions(-)

diff --git a/DragaliaAPI/DragaliaAPI.Database/Repositories/UnitRepository.cs b/DragaliaAPI/DragaliaAPI.Database/Repositories/UnitRepository.cs
index 86ed2db68..e62914482 100644
--- a/DragaliaAPI/DragaliaAPI.Database/Repositories/UnitRepository.cs
+++ b/DragaliaAPI/DragaliaAPI.Database/Repositories/UnitRepository.cs
@@ -103,12 +103,19 @@ ILogger<UnitRepository> logger
     public async Task<IEnumerable<(Charas id, bool isNew)>> AddCharas(IEnumerable<Charas> idList)
     {
         List<Charas> enumeratedIdList = idList.ToList();
+        IEnumerable<int> storyIdList = GetFirstCharaStories(enumeratedIdList);
 
         // Generate result. The first occurrence of a character in the list should be new (if not in the DB)
         // but subsequent results should then not be labelled as new.
-        List<Charas> ownedCharas = await Charas.Select(x => x.CharaId).ToListAsync();
+        List<Charas> ownedCharas = await Charas
+            .Select(x => x.CharaId)
+            .Where(x => enumeratedIdList.Contains(x))
+            .ToListAsync();
+
         List<int> ownedCharaStories = await this
-            .apiContext.PlayerStoryState.Where(x => x.StoryType == StoryTypes.Chara)
+            .apiContext.PlayerStoryState.Where(x =>
+                x.StoryType == StoryTypes.Chara && storyIdList.Contains(x.StoryId)
+            )
             .Select(x => x.StoryId)
             .ToListAsync();
 
@@ -149,6 +156,17 @@ ILogger<UnitRepository> logger
         }
 
         return newMapping.Select(x => (x.Id, x.IsNew));
+
+        static IEnumerable<int> GetFirstCharaStories(IEnumerable<Charas> charaIdList)
+        {
+            foreach (Charas c in charaIdList)
+            {
+                if (MasterAsset.CharaStories.TryGetValue((int)c, out StoryData? storyData))
+                {
+                    yield return storyData.storyIds[0];
+                }
+            }
+        }
     }
 
     public async Task<bool> AddCharas(Charas id)
@@ -160,9 +178,13 @@ public async Task<bool> AddCharas(Charas id)
     {
         List<Dragons> enumeratedIdList = idList.ToList();
 
-        List<Dragons> ownedDragons = await Dragons.Select(x => x.DragonId).ToListAsync();
+        List<Dragons> ownedDragons = await Dragons
+            .Select(x => x.DragonId)
+            .Where(x => enumeratedIdList.Contains(x))
+            .ToListAsync();
         List<Dragons> ownedReliabilities = await DragonReliabilities
             .Select(x => x.DragonId)
+            .Where(x => enumeratedIdList.Contains(x))
             .ToListAsync();
 
         List<DragonNewCheckResult> newMapping = MarkNewDragons(