Implementing dispose

sj-distributor · Feb 2, 2023 · c521599 · c521599
1 parent 73ca9b8
commit c521599
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/Wax.Core/Repositories/IUnitOfWork.cs b/src/Wax.Core/Repositories/IUnitOfWork.cs
@@ -1,6 +1,6 @@
 namespace Wax.Core.Repositories;
 
-public interface IUnitOfWork
+public interface IUnitOfWork : IDisposable
 {
     public ICustomerRepository Customers { get; }
     Task SaveChangesAsync(CancellationToken cancellationToken = default);

diff --git a/src/Wax.Core/Repositories/UnitOfWork.cs b/src/Wax.Core/Repositories/UnitOfWork.cs
@@ -20,4 +20,9 @@ public Task SaveChangesAsync(CancellationToken cancellationToken = default)
     {
         return _context.SaveChangesAsync(cancellationToken);
     }
+
+    public void Dispose()
+    {
+        _context?.Dispose();
+    }
 }