DDD 領域驅動設計－談談 Repository、IUnitOfWork 和 IDbContext 的實踐（3）

轉載自： http://www.cnblogs.com/xishuai/ 

原做者： 田園裏的蟋蟀 

上一篇：《DDD 領域驅動設計－談談 Repository、IUnitOfWork 和 IDbContext 的實踐（2）》

這篇文章主要是對 DDD.Sample 框架增長 Transaction 事務操做，以及增長了一些必要項目。

雖然如今的 IUnitOfWork 實現中有 Commit 的實現，但也就是使用的 EF SaveChanges，知足一些簡單操做能夠，但一些稍微複雜點的實體操做就不行了，而且 Rollback 也沒有實現。

如今的 UnitOfWork 實現代碼：

public class UnitOfWork : IUnitOfWork { private IDbContext _dbContext; public UnitOfWork(IDbContext dbContext) { _dbContext = dbContext; } public void RegisterNew<TEntity>(TEntity entity) where TEntity : class { _dbContext.Set<TEntity>().Add(entity); } public void RegisterDirty<TEntity>(TEntity entity) where TEntity : class { _dbContext.Entry<TEntity>(entity).State = EntityState.Modified; } public void RegisterClean<TEntity>(TEntity entity) where TEntity : class { _dbContext.Entry<TEntity>(entity).State = EntityState.Unchanged; } public void RegisterDeleted<TEntity>(TEntity entity) where TEntity : class { _dbContext.Set<TEntity>().Remove(entity); } public async Task<bool> CommitAsync() { return await _dbContext.SaveChangesAsync() > 0; } public void Rollback() { throw new NotImplementedException(); } }

基於上面的實現，好比要處理這樣的一個操做：先添加一個 Teacher，而後再添加一個 Student，Student 實體中有一個 TeacherId，通常實現方式以下：

public async Task<bool> Add(string name) { var teacher = new Teacher { Name = "teacher one" }; _unitOfWork.RegisterNew(teacher); await _unitOfWork.CommitAsync(); //可能還有一些 web 請求操做，好比 httpClient.Post(tearch); 可能會出選異常。 var student = new Student { Name = name, TeacherId = teacher.Id }; _unitOfWork.RegisterNew(student); await _unitOfWork.CommitAsync(); return true; }

上面的實現可能會出現一些問題，好比添加 Teacher 出現了異常，web 請求出現了異常，添加 Student 出現了異常等，該如何進行處理？因此你可能會增長不少判斷，還有就是異常出現後的修復操做，當需求很複雜的時候，咱們基於上面的處理也就會更加複雜，根本緣由是並無真正的實現 Transaction 事務操做。

若是單獨在 EF 中實現 Transaction 操做，可使用 TransactionScope，參考文章：在 Entity Framework 中使用事務

TransactionScope 的實現比較簡單，如何在 DDD.Sample 框架中，結合 IUnitOfWork 和 IDbContext 進行使用呢？可能實現方式有不少中，如今個人實現是這樣：

首先，IDbContext 中增長 Database 屬性定義：

public interface IDbContext { Database Database { get; } //add DbSet<TEntity> Set<TEntity>() where TEntity : class; DbEntityEntry<TEntity> Entry<TEntity>(TEntity entity) where TEntity : class; Task<int> SaveChangesAsync(); }

增長 Database 屬性的目的是，便於咱們在 UnitOfWork 中訪問到 Transaction，其實還能夠定義這樣的接口：DbContextTransaction BeginTransaction();，但 EF 中的 DbContext 並無進行實現，而是須要經過 Database 屬性，因此還須要在 IDbContext 實現中額外實現，另外增長 Database 屬性的好處，還有就是能夠在 UnitOfWork 中訪問執行不少的操做，好比執行 SQL 語句等等。

這裏須要說下 IDbContext 定義，我原先的設計初衷是，讓它脫離 EF，做爲全部數據操做上下文的定義，但其實實現的時候，仍是脫離不了 EF，由於接口返回的類型都在 EF 下，最後 IDbContext 就變成了 EF DbContext 的部分接口定義，因此這部分是須要再進行設計的，但好在有了 IDbContext，可讓 EF 和 UnitOfWork 隔離開來。

SchoolDbContext 中的實現沒有任何變換，由於繼承的 EF DbContext 已經有了實現，UnitOfWork 改動比較大，代碼以下：

public class UnitOfWork : IUnitOfWork { private IDbContext _dbContext; private DbContextTransaction _dbTransaction; public UnitOfWork(IDbContext dbContext) { _dbContext = dbContext; _dbTransaction = _dbContext.Database.BeginTransaction(); } public async Task<bool> RegisterNew<TEntity>(TEntity entity) where TEntity : class { _dbContext.Set<TEntity>().Add(entity); return await _dbContext.SaveChangesAsync() > 0; } public async Task<bool> RegisterDirty<TEntity>(TEntity entity) where TEntity : class { _dbContext.Entry<TEntity>(entity).State = EntityState.Modified; return await _dbContext.SaveChangesAsync() > 0; } public async Task<bool> RegisterClean<TEntity>(TEntity entity) where TEntity : class { _dbContext.Entry<TEntity>(entity).State = EntityState.Unchanged; return await _dbContext.SaveChangesAsync() > 0; } public async Task<bool> RegisterDeleted<TEntity>(TEntity entity) where TEntity : class { _dbContext.Set<TEntity>().Remove(entity); return await _dbContext.SaveChangesAsync() > 0; } public void Commit() { _dbTransaction.Commit(); } public void Rollback() { _dbTransaction.Rollback(); } }

UnitOfWork 構造函數中，根據 DbContext 建立 DbContextTransaction 對象，而後在實體每一個操做中，都添加了 SaveChanges，由於咱們用了 Transaction，因此在執行 SaveChanges 的時候，並無應用到數據庫，但能夠獲取到新添加實體的 Id，好比上面示例 Student 中的 TeacherId，而且用 Sql Profiler 能夠檢測到執行的 SQL 代碼，當執行 Commit 的時候，數據對應進行更新。

測試代碼：

public async Task<bool> AddWithTransaction(string name) { var teacher = new Teacher { Name = "teacher one" }; await _unitOfWork.RegisterNew(teacher); //可能還有一些 web 請求操做，好比 httpClient.Post(tearch); 可能會出選異常。 var student = new Student { Name = name, TeacherId = teacher.Id }; await _unitOfWork.RegisterNew(student); _unitOfWork.Commit(); return true; }

在上面代碼中，首先在沒執行到 Commit 以前，是能夠獲取到新添加 Teacher 的 Id，而且若是出現了任何異常，都是能夠進行回滾的，固然也能夠手動進行 catch 異常，並執行_unitOfWork.Rollback()。

不過上面的實現有一個問題，就是每次實體操做，都用了 Transaction，性能我沒測試，但確定會有影響，好處就是 IUnitOfWork 基本沒有改動，仍是按照官方的定義，只不過部分接口改爲了異步接口。

除了上面的實現，還有一種解決方式，就是在 IUnitOfWork 中增長一個相似 BeginTransaction 的接口，大體實現代碼：

public class UnitOfWork : IUnitOfWork { private IDbContext _dbContext; private DbContextTransaction _dbTransaction; public UnitOfWork(IDbContext dbContext) { _dbContext = dbContext; } //add public void BeginTransaction() { _dbTransaction = _dbContext.Database.BeginTransaction(); } public async Task<bool> RegisterNew<TEntity>(TEntity entity) where TEntity : class { _dbContext.Set<TEntity>().Add(entity); if (_dbTransaction != null) return await _dbContext.SaveChangesAsync() > 0; return true; } public async Task<bool> RegisterDirty<TEntity>(TEntity entity) where TEntity : class { _dbContext.Entry<TEntity>(entity).State = EntityState.Modified; if (_dbTransaction != null) return await _dbContext.SaveChangesAsync() > 0; return true; } public async Task<bool> RegisterClean<TEntity>(TEntity entity) where TEntity : class { _dbContext.Entry<TEntity>(entity).State = EntityState.Unchanged; if (_dbTransaction != null) return await _dbContext.SaveChangesAsync() > 0; return true; } public async Task<bool> RegisterDeleted<TEntity>(TEntity entity) where TEntity : class { _dbContext.Set<TEntity>().Remove(entity); if (_dbTransaction != null) return await _dbContext.SaveChangesAsync() > 0; return true; } public async Task<bool> Commit() { if (_dbTransaction == null) return await _dbContext.SaveChangesAsync() > 0; else _dbTransaction.Commit(); return true; } public void Rollback() { if (_dbTransaction != null) _dbTransaction.Rollback(); } }

上面這種實現方式就解決了第一種方式的問題，須要使用 Transaction 的時候，直接在操做以前調用 BeginTransaction 就好了，但很差的地方就是改動了 IUnitOfWork 的接口定義。

除了上面兩種實現方式，你們若是有更好的解決方案，歡迎提出。

另外，DDD.Sample 增長和改動了一些東西：

• 增長 DDD.Sample.Domain.DomainEvents、DDD.Sample.Domain.DomainServices 和 DDD.Sample.Domain.ValueObjects。
• 從 DDD.Sample.Domain 分離出 DDD.Sample.Domain.Repository.Interfaces。
• 增長 DDD.Sample.BootStrapper，執行 Startup.Configure 用於系統啓動的配置。
• 去除 IEntity，在 IAggregateRoot 中添加 Id 屬性定義。