fix bug where you couldn't submit updates to an existing entity after…

… creating a new data model instance.
sillsdev · Sep 20, 2024 · 0063f05 · 0063f05
1 parent 59a1700
commit 0063f05
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 6 deletions.
diff --git a/src/SIL.Harmony/Db/CrdtRepository.cs b/src/SIL.Harmony/Db/CrdtRepository.cs
@@ -143,14 +143,17 @@ public async Task<Commit[]> GetCommitsAfter(Commit? commit)
             .ToArrayAsync();
     }
 
-    public async Task<ObjectSnapshot?> FindSnapshot(Guid id)
+    public async Task<ObjectSnapshot?> FindSnapshot(Guid id, bool tracking = false)
     {        
-        return await Snapshots.Include(s => s.Commit).SingleOrDefaultAsync(s => s.Id == id);
+        return await Snapshots
+            .AsTracking(tracking)
+            .Include(s => s.Commit)
+            .SingleOrDefaultAsync(s => s.Id == id);
     }
 
-    public async Task<ObjectSnapshot?> GetCurrentSnapshotByObjectId(Guid objectId)
+    public async Task<ObjectSnapshot?> GetCurrentSnapshotByObjectId(Guid objectId, bool tracking = false)
     {
-        return await Snapshots.Include(s => s.Commit)
+        return await Snapshots.AsTracking(tracking).Include(s => s.Commit)
             .DefaultOrder()
             .LastOrDefaultAsync(s => s.EntityId == objectId && (ignoreChangesAfter == null || s.Commit.DateTime <= ignoreChangesAfter));
     }

diff --git a/src/SIL.Harmony/Db/DbSetExtensions.cs b/src/SIL.Harmony/Db/DbSetExtensions.cs
@@ -1,3 +1,5 @@
+using Microsoft.EntityFrameworkCore;
+
 namespace SIL.Harmony.Db;
 
 //todo, I would like to move these extensions into QueryHelperTests but that's in Core and ObjectSnapshot is not part of core
@@ -26,4 +28,9 @@ public static IQueryable<ObjectSnapshot> WhereAfter(this IQueryable<ObjectSnapsh
                  || (after.HybridDateTime.DateTime == s.Commit.HybridDateTime.DateTime && after.HybridDateTime.Counter < s.Commit.HybridDateTime.Counter)
                  || (after.HybridDateTime.DateTime == s.Commit.HybridDateTime.DateTime && after.HybridDateTime.Counter == s.Commit.HybridDateTime.Counter && after.Id < s.Commit.Id));
     }
+
+    public static IQueryable<T> AsTracking<T>(this IQueryable<T> queryable, bool tracking = true) where T : class
+    {
+        return queryable.AsTracking(tracking ? QueryTrackingBehavior.TrackAll : QueryTrackingBehavior.NoTracking);
+    }
 }
diff --git a/src/SIL.Harmony/SnapshotWorker.cs b/src/SIL.Harmony/SnapshotWorker.cs
@@ -177,10 +177,10 @@ private async ValueTask MarkDeleted(Guid deletedEntityId, Commit commit)
         if (_snapshotLookup.TryGetValue(entityId, out var snapshotId))
         {
             if (snapshotId is null) return null;
-            return await _crdtRepository.FindSnapshot(snapshotId.Value);
+            return await _crdtRepository.FindSnapshot(snapshotId.Value, true);
         }
 
-        snapshot = await _crdtRepository.GetCurrentSnapshotByObjectId(entityId);
+        snapshot = await _crdtRepository.GetCurrentSnapshotByObjectId(entityId, true);
         _snapshotLookup[entityId] = snapshot?.Id;
 
         return snapshot;