Override SaveChangesAsync

sj-distributor · Apr 7, 2023 · 11894c8 · 11894c8
1 parent 284e18b
commit 11894c8
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/src/Wax.Core/Data/ApplicationDbContext.cs b/src/Wax.Core/Data/ApplicationDbContext.cs
@@ -9,6 +9,8 @@ public ApplicationDbContext(DbContextOptions<ApplicationDbContext> options) : ba
     {
     }
 
+    public bool HasEntitiesChanged { get; private set; }
+
     protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
     {
         optionsBuilder.UseInMemoryDatabase("__wax_database");
@@ -18,9 +20,17 @@ protected override void OnModelCreating(ModelBuilder modelBuilder)
     {
         modelBuilder.ApplyConfigurationsFromAssembly(typeof(CustomerEntityTypeConfiguration).Assembly);
     }
-
-    public bool HasEntitiesChanged { get; private set; }
-
+
+
+    public override Task<int> SaveChangesAsync(CancellationToken cancellationToken = new CancellationToken())
+    {
+        var result = base.SaveChangesAsync(cancellationToken);
+
+        HasEntitiesChanged = false;
+
+        return result;
+    }
+
     public async Task ChangeEntitiesAsync(bool saveNow = false, CancellationToken cancellationToken = default)
     {
         HasEntitiesChanged = true;