Add disposable methods

Author: Morten Turn Pedersen

NetCoreEntityFramework/NetCoreEntityFramework.csproj

@@ -3,7 +3,7 @@
   <PropertyGroup>
     <TargetFramework>netstandard2.0</TargetFramework>
     <PackageId>Nodes.NetCore.EntityFramework.Helpers</PackageId>
-    <Version>0.1.0</Version>
+    <Version>0.1.1</Version>
     <Authors>Nodes</Authors>
     <Company>Nodes</Company>
     <Product>.NET Core Entity Framework Helpers</Product>

NetCoreEntityFramework/Repositories/EntityRepository.cs

@@ -5,18 +5,25 @@
 
 namespace Nodes.NetCore.EntityFramework.Repositories
 {
-    public abstract class EntityRepository<T> where T : EntityBase
+    public abstract class EntityRepository<T, TContext> : IDisposable where T : EntityBase where TContext : DbContext
     {
         protected DbSet<T> Table { get; private set; }
+        private TContext Context { get; set; }
 
-        protected EntityRepository(DbSet<T> table)
+        protected EntityRepository(TContext context, DbSet<T> table)
         {
+            Context = context;
             Table = table;
         }
 
         public async Task<T> Get(Guid id)
         {
             return await Table.FirstOrDefaultAsync(entity => !entity.Deleted && entity.Id == id);
         }
+
+        public void Dispose()
+        {
+            Context.SaveChanges();
+        }
     }
 }