Apply more review suggestions

3 files changed, 8 insertions, 10 deletions

diff --git a/Jellyfin.Server.Implementations/Activity/ActivityManager.cs b/Jellyfin.Server.Implementations/Activity/ActivityManager.cs
index 0b398b60c..65ceee32b 100644
--- a/Jellyfin.Server.Implementations/Activity/ActivityManager.cs
+++ b/Jellyfin.Server.Implementations/Activity/ActivityManager.cs
@@ -1,5 +1,4 @@
 using System;
-using System.Collections.Generic;
 using System.Linq;
 using System.Threading.Tasks;
 using Jellyfin.Data.Entities;
@@ -56,7 +55,7 @@ namespace Jellyfin.Server.Implementations.Activity
         {
             using var dbContext = _provider.CreateContext();
 
-            var query = func(dbContext.ActivityLogs).OrderByDescending(entry => entry.DateCreated).AsQueryable();
+            var query = func(dbContext.ActivityLogs.OrderByDescending(entry => entry.DateCreated));
 
             if (startIndex.HasValue)
             {
@@ -69,12 +68,12 @@ namespace Jellyfin.Server.Implementations.Activity
             }
 
             // This converts the objects from the new database model to the old for compatibility with the existing API.
-            var list = query.AsEnumerable().Select(ConvertToOldModel).ToList();
+            var list = query.Select(ConvertToOldModel).ToList();
 
             return new QueryResult<ActivityLogEntry>
             {
                 Items = list,
-                TotalRecordCount = dbContext.ActivityLogs.Count()
+                TotalRecordCount = func(dbContext.ActivityLogs).Count()
             };
         }
 
diff --git a/Jellyfin.Server.Implementations/JellyfinDb.cs b/Jellyfin.Server.Implementations/JellyfinDb.cs
index 23714b24a..ec09a619f 100644
--- a/Jellyfin.Server.Implementations/JellyfinDb.cs
+++ b/Jellyfin.Server.Implementations/JellyfinDb.cs
@@ -107,10 +107,11 @@ namespace Jellyfin.Server.Implementations
 
         public override int SaveChanges()
         {
-            foreach (var entity in ChangeTracker.Entries().Where(e => e.State == EntityState.Modified))
+            foreach (var saveEntity in ChangeTracker.Entries()
+                .Where(e => e.State == EntityState.Modified)
+                .OfType<ISavingChanges>())
             {
-                var saveEntity = entity.Entity as ISavingChanges;
-                saveEntity?.OnSavingChanges();
+                saveEntity.OnSavingChanges();
             }
 
             return base.SaveChanges();
diff --git a/Jellyfin.Server.Implementations/Migrations/DesignTimeJellyfinDbFactory.cs b/Jellyfin.Server.Implementations/Migrations/DesignTimeJellyfinDbFactory.cs
index a1b58eb5a..72a4a8c3b 100644
--- a/Jellyfin.Server.Implementations/Migrations/DesignTimeJellyfinDbFactory.cs
+++ b/Jellyfin.Server.Implementations/Migrations/DesignTimeJellyfinDbFactory.cs
@@ -12,9 +12,7 @@ namespace Jellyfin.Server.Implementations.Migrations
         public JellyfinDb CreateDbContext(string[] args)
         {
             var optionsBuilder = new DbContextOptionsBuilder<JellyfinDb>();
-            optionsBuilder.UseSqlite(
-                "Data Source=jellyfin.db",
-                opt => opt.MigrationsAssembly("Jellyfin.Migrations"));
+            optionsBuilder.UseSqlite("Data Source=jellyfin.db");
 
             return new JellyfinDb(optionsBuilder.Options);
         }