Automatically set update timestamp.

leaderboardsgg · Sep 20, 2024 · 2c8dfed · 2c8dfed
1 parent 3aa32d0
commit 2c8dfed
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/LeaderboardBackend/Models/Entities/ApplicationContext.cs b/LeaderboardBackend/Models/Entities/ApplicationContext.cs
@@ -20,19 +20,28 @@ static ApplicationContext()
         NpgsqlConnection.GlobalTypeMapper.MapEnum<RunType>();
     }
 
-    private void AddCreationTimeStamp(object? sender, EntityEntryEventArgs e)
+    private void AddCreationTimestamp(object? sender, EntityEntryEventArgs e)
     {
         if (e.Entry.State is EntityState.Added && e.Entry.Entity is IHasCreationTimestamp entity)
         {
             entity.CreatedAt = _clock.GetCurrentInstant();
         }
     }
 
+    private void SetUpdateTimestamp(object? sender, EntityEntryEventArgs e)
+    {
+        if (e.Entry.State is EntityState.Modified && e.Entry.Entity is IHasUpdateTimestamp entity)
+        {
+            entity.UpdatedAt = _clock.GetCurrentInstant();
+        }
+    }
+
     public ApplicationContext(DbContextOptions<ApplicationContext> options, IClock clock)
         : base(options)
     {
         _clock = clock;
-        ChangeTracker.Tracked += AddCreationTimeStamp;
+        ChangeTracker.Tracked += AddCreationTimestamp;
+        ChangeTracker.Tracked += SetUpdateTimestamp;
     }
 
     public DbSet<AccountRecovery> AccountRecoveries { get; set; } = null!;