Fix bug in stats listener

PSRewired · Aug 29, 2023 · 61373ad · 61373ad
1 parent 037d1db
commit 61373ad
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/src/Fragment.NetSlum.Persistence/Interceptors/EntityChangeInterceptor.cs b/src/Fragment.NetSlum.Persistence/Interceptors/EntityChangeInterceptor.cs
@@ -46,7 +46,9 @@ private void ExecuteListeners(DbContext context)
     {
         context.ChangeTracker.DetectChanges();
 
-        var changedEntities = context.ChangeTracker.Entries().ToArray();
+        var changedEntities = context.ChangeTracker.Entries()
+            .Where(e => e.State is EntityState.Added or EntityState.Modified or EntityState.Deleted)
+            .ToArray();
 
         foreach (var entry in changedEntities)
         {

diff --git a/src/Fragment.NetSlum.Persistence/Listeners/CharacterStatsChangeListener.cs b/src/Fragment.NetSlum.Persistence/Listeners/CharacterStatsChangeListener.cs
@@ -9,14 +9,13 @@ public class CharacterStatsChangeListener : AbstractEntityChangeListener<Fragmen
 {
     protected override Task OnEntityChanged(FragmentContext context, EntityEntry entry)
     {
-
         if (entry.Entity is not CharacterStats statsEntry)
         {
             return Task.CompletedTask;
         }
 
-        // Only update the history tables when stats are modified or added for the first time
-        if (entry.State != EntityState.Modified || entry.State != EntityState.Added)
+        // If the player stats are deleted (which should never happen), we can ignore stats updates
+        if (entry.State == EntityState.Deleted)
         {
             return Task.CompletedTask;
         }