From 3657336b31d743c101a200330799e148cb760077 Mon Sep 17 00:00:00 2001
From: noufionline <noufionline@gmail.com>
Date: Wed, 28 Sep 2016 16:06:28 +0400
Subject: [PATCH 1/2] UnitOfWork with RepositoryFactory

Unit of work which accepts repository factory
---
 .../UnitOfWork                                | 99 +++++++++++++++++++
 1 file changed, 99 insertions(+)
 create mode 100644 Source/TrackableEntities.Patterns.EF.6/UnitOfWork

diff --git a/Source/TrackableEntities.Patterns.EF.6/UnitOfWork b/Source/TrackableEntities.Patterns.EF.6/UnitOfWork
new file mode 100644
index 00000000..af8c126d
--- /dev/null
+++ b/Source/TrackableEntities.Patterns.EF.6/UnitOfWork
@@ -0,0 +1,99 @@
+using System;
+using System.Threading;
+using System.Threading.Tasks;
+
+/// <summary>
+/// General unit of work for committing changes across one or more repositories. 
+/// Inherit from this class to supply a specific DbContext. 
+/// Add a property for each respository for which unit of work must be done.
+/// </summary>
+public abstract class UnitOfWork : IUnitOfWork, IUnitOfWorkAsync, IDisposable
+{
+    private readonly IRepositoryFactory _repositoryFactory;
+    private bool _disposed;
+
+    /// <summary>
+    /// Constructs a new general unit of work.
+    /// </summary>
+    protected UnitOfWork() { }
+
+    /// <summary>
+    /// Constructs a new general unit of work.
+    /// </summary>
+    /// <param name="context">Entity Framework DbContext-derived class.</param>
+    protected UnitOfWork(DbContext context, IRepositoryFactory repositoryFactory)
+    {
+        _repositoryFactory = repositoryFactory;
+        Context = context;
+    }
+
+    /// <summary>
+    /// Gets the DbContext for the unit of work.
+    /// </summary>
+    protected DbContext Context { get; set; }
+
+    /// <summary>
+    /// Disposes the DbContext.
+    /// </summary>
+    public void Dispose()
+    {
+        Dispose(true);
+        GC.SuppressFinalize(this);
+    }
+
+    public T GetRepository<T>()
+    {
+        return _repositoryFactory.GetRepository<T>();
+    }
+
+    /// <summary>
+    /// Disposes the DbContext.
+    /// </summary>
+    /// <param name="disposing">True to release both managed and unmanaged resources; false to release only unmanaged resources.</param>
+    protected virtual void Dispose(bool disposing)
+    {
+        if (_disposed) return;
+        if (disposing) Context.Dispose();
+        _disposed = true;
+    }
+
+    /// <summary>
+    /// Saves changes made to one or more repositories.
+    /// </summary>
+    /// <returns>The number of objects saved.</returns>
+    public virtual int SaveChanges()
+    {
+        if (_disposed)
+            throw new ObjectDisposedException("UnitOfWork");
+        return Context.SaveChanges();
+    }
+
+    /// <summary>
+    /// Saves changes made to one or more repositories.
+    /// </summary>
+    /// <returns>A task that represents the asynchronous save operation. The task result contains the number of objects saved.</returns>
+    public virtual async Task<int> SaveChangesAsync()
+    {
+        if (_disposed)
+            throw new ObjectDisposedException("UnitOfWork");
+        return await SaveChangesAsync(CancellationToken.None);
+    }
+
+    /// <summary>
+    /// Saves changes made to one or more repositories.
+    /// </summary>
+    /// <param name="cancellationToken">A CancellationToken to observe while waiting for the task to complete.</param>
+    /// <returns>A task that represents the asynchronous save operation. The task result contains the number of objects saved.</returns>
+    public virtual async Task<int> SaveChangesAsync(CancellationToken cancellationToken)
+    {
+        if (_disposed)
+            throw new ObjectDisposedException("UnitOfWork");
+        return await Context.SaveChangesAsync(cancellationToken);
+    }
+}
+
+
+public interface IRepositoryFactory
+{
+    T GetRepository<T>();
+}

From 484f3c99d9873c53d8362b3a1dc76e6ec5d72fe0 Mon Sep 17 00:00:00 2001
From: noufionline <noufionline@gmail.com>
Date: Wed, 28 Sep 2016 16:14:21 +0400
Subject: [PATCH 2/2] Update UnitOfWork

---
 Source/TrackableEntities.Patterns.EF.6/UnitOfWork | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Source/TrackableEntities.Patterns.EF.6/UnitOfWork b/Source/TrackableEntities.Patterns.EF.6/UnitOfWork
index af8c126d..8864d25b 100644
--- a/Source/TrackableEntities.Patterns.EF.6/UnitOfWork
+++ b/Source/TrackableEntities.Patterns.EF.6/UnitOfWork
@@ -41,7 +41,7 @@ public abstract class UnitOfWork : IUnitOfWork, IUnitOfWorkAsync, IDisposable
         GC.SuppressFinalize(this);
     }
 
-    public T GetRepository<T>()
+    protected T GetRepository<T>()
     {
         return _repositoryFactory.GetRepository<T>();
     }