From 7c10f3a74d0c953d2224e41f5c69ab2d7d13636b Mon Sep 17 00:00:00 2001
From: ta264 <ta264@users.noreply.github.com>
Date: Thu, 2 Sep 2021 21:12:13 +0100
Subject: [PATCH] New: Speed up metadata refresh

---
 .../Metadata/MetadataProfileService.cs        | 24 ++++++++++++++++++-
 1 file changed, 23 insertions(+), 1 deletion(-)

diff --git a/src/NzbDrone.Core/Profiles/Metadata/MetadataProfileService.cs b/src/NzbDrone.Core/Profiles/Metadata/MetadataProfileService.cs
index 840db3b67..53afee3d1 100644
--- a/src/NzbDrone.Core/Profiles/Metadata/MetadataProfileService.cs
+++ b/src/NzbDrone.Core/Profiles/Metadata/MetadataProfileService.cs
@@ -35,6 +35,7 @@ public class MetadataProfileService : IMetadataProfileService, IHandle<Applicati
         private readonly IMetadataProfileRepository _profileRepository;
         private readonly IAuthorService _authorService;
         private readonly IBookService _bookService;
+        private readonly IEditionService _editionService;
         private readonly IMediaFileService _mediaFileService;
         private readonly IImportListFactory _importListFactory;
         private readonly IRootFolderService _rootFolderService;
@@ -44,6 +45,7 @@ public class MetadataProfileService : IMetadataProfileService, IHandle<Applicati
         public MetadataProfileService(IMetadataProfileRepository profileRepository,
                                       IAuthorService authorService,
                                       IBookService bookService,
+                                      IEditionService editionService,
                                       IMediaFileService mediaFileService,
                                       IImportListFactory importListFactory,
                                       IRootFolderService rootFolderService,
@@ -53,6 +55,7 @@ public MetadataProfileService(IMetadataProfileRepository profileRepository,
             _profileRepository = profileRepository;
             _authorService = authorService;
             _bookService = bookService;
+            _editionService = editionService;
             _mediaFileService = mediaFileService;
             _importListFactory = importListFactory;
             _rootFolderService = rootFolderService;
@@ -112,7 +115,26 @@ public List<Book> FilterBooks(Author input, int profileId)
                 .ToDictionary(x => x.Key, y => y.ToList());
 
             var dbAuthor = _authorService.FindById(input.ForeignAuthorId);
-            var localBooks = dbAuthor?.Books.Value ?? new List<Book>();
+
+            var localBooks = new List<Book>();
+            if (dbAuthor != null)
+            {
+                localBooks = _bookService.GetBooksByAuthorMetadataId(dbAuthor.AuthorMetadataId);
+                var editions = _editionService.GetEditionsByAuthor(dbAuthor.Id).GroupBy(x => x.BookId).ToDictionary(x => x.Key, y => y.ToList());
+
+                foreach (var book in localBooks)
+                {
+                    if (editions.TryGetValue(book.Id, out var bookEditions))
+                    {
+                        book.Editions = bookEditions;
+                    }
+                    else
+                    {
+                        book.Editions = new List<Edition>();
+                    }
+                }
+            }
+
             var localFiles = _mediaFileService.GetFilesByAuthor(dbAuthor?.Id ?? 0);
 
             return FilterBooks(input.Books.Value, localBooks, localFiles, seriesLinks, profileId);