ASP.NET MVC分層架構(7)-Unit of Work

public interface IUnitOfWork : IDisposable
{
	DbContext Context { get; }

	int SaveChange();
}

public class UnitOfWork : IUnitOfWork
{
	private IDbContextFactory _dbContextFactory { get; set; }

	private DbContext context;

	private bool disposed = false;

	public DbContext Context
	{
		get
		{
			if (this.context != null)
			{
				return this.context;
			}

			this.context = _dbContextFactory.GetDbContext();
			return this.context;
		}
	}

	public UnitOfWork(IDbContextFactory dbContextFactory)
	{
		this._dbContextFactory = dbContextFactory;
	}

	public int SaveChange()
	{
		int result = this.Context.SaveChanges();
		return result;
	}

	protected virtual void Dispose(bool disposing)
	{
		if (!this.disposed)
		{
			if (disposing)
			{
				this.Context.Dispose();
				this.context = null;
			}
		}
		this.disposed = true;
	}

	public void Dispose()
	{
		Dispose(true);
		GC.SuppressFinalize(this);
	}
}

public class GenericRepository<TEntity> : IRepository<TEntity>
        where TEntity : class 
{
	public IUnitOfWork _UnitOfWork { get; set; }
	
	private DbSet<TEntity> _dbSet { get; set; }
	
	public GenericRepository(IUnitOfWork unitOfWork)
	{
		this._UnitOfWork = unitOfWork;
		this._context = unitOfWork.Context;
		this._dbSet = unitOfWork.Context.Set<TEntity>();
	}
}

public class ProjectService : IProjectService
{
	private readonly IUnitOfWork _unitOfWork;
	
	private readonly IRepository<Project> _repository;

	public ProjectService(IUnitOfWork unitOfWork, IRepository<Project> repository)
	{
		this._unitOfWork = unitOfWork;
		this._repository = repository;
	}
	
	public void Create(Project entity)
	{
		if (entity == null)
		{
			throw new ArgumentNullException("entity");
		}
		else
		{
			_repository.Create(entity);
			_unitOfWork.SaveChange();
		}
	}

	public void Update(Project entity)
	{
		if (entity == null)
		{
			throw new ArgumentNullException("entity");
		}
		else
		{
			_repository.Update(entity);
			_unitOfWork.SaveChange();
		}
	}

	public void Delete(Guid id)
	{
		if (!IsExists(id))
		{

		}
		else
		{
			var entity = GetById(id);
			_repository.Delete(entity);
			_unitOfWork.SaveChange();
		}
	}
}

public static void RegisterTypes(IUnityContainer container)
{
	#region UnitOfWork
	container.RegisterType<IUnitOfWork, UnitOfWork>();
	#endregion
}