本文共 6887 字，大约阅读时间需要 22 分钟。
现在的 UnitOfWork 实现代码：
public class UnitOfWork : IUnitOfWork{ private IDbContext _dbContext; public UnitOfWork(IDbContext dbContext) { _dbContext = dbContext; } public void RegisterNew (TEntity entity) where TEntity : class { _dbContext.Set ().Add(entity); } public void RegisterDirty (TEntity entity) where TEntity : class { _dbContext.Entry (entity).State = EntityState.Modified; } public void RegisterClean (TEntity entity) where TEntity : class { _dbContext.Entry (entity).State = EntityState.Unchanged; } public void RegisterDeleted (TEntity entity) where TEntity : class { _dbContext.Set ().Remove(entity); } public async Task CommitAsync() { return await _dbContext.SaveChangesAsync() > 0; } public void Rollback() { throw new NotImplementedException(); }} 基于上面的实现，比如要处理这样的一个操作：先添加一个 Teacher，然后再添加一个 Student，Student 实体中有一个 TeacherId，一般实现方式如下：
public async TaskAdd(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，参考文章：
TransactionScope 的实现比较简单，如何在 DDD.Sample 框架中，结合 IUnitOfWork 和 IDbContext 进行使用呢？可能实现方式有很多中，现在我的实现是这样：
首先，IDbContext 中增加 Database 属性定义：
public interface IDbContext{ Database Database { get; } //add DbSet Set () where TEntity : class; DbEntityEntry Entry (TEntity entity) where TEntity : class; Task 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 RegisterNew (TEntity entity) where TEntity : class { _dbContext.Set ().Add(entity); return await _dbContext.SaveChangesAsync() > 0; } public async Task RegisterDirty (TEntity entity) where TEntity : class { _dbContext.Entry (entity).State = EntityState.Modified; return await _dbContext.SaveChangesAsync() > 0; } public async Task RegisterClean (TEntity entity) where TEntity : class { _dbContext.Entry (entity).State = EntityState.Unchanged; return await _dbContext.SaveChangesAsync() > 0; } public async Task RegisterDeleted (TEntity entity) where TEntity : class { _dbContext.Set ().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 TaskAddWithTransaction(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 RegisterNew (TEntity entity) where TEntity : class { _dbContext.Set ().Add(entity); if (_dbTransaction != null) return await _dbContext.SaveChangesAsync() > 0; return true; } public async Task RegisterDirty (TEntity entity) where TEntity : class { _dbContext.Entry (entity).State = EntityState.Modified; if (_dbTransaction != null) return await _dbContext.SaveChangesAsync() > 0; return true; } public async Task RegisterClean (TEntity entity) where TEntity : class { _dbContext.Entry (entity).State = EntityState.Unchanged; if (_dbTransaction != null) return await _dbContext.SaveChangesAsync() > 0; return true; } public async Task RegisterDeleted (TEntity entity) where TEntity : class { _dbContext.Set ().Remove(entity); if (_dbTransaction != null) return await _dbContext.SaveChangesAsync() > 0; return true; } public async Task 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 增加和改动了一些东西：
本文转自田园里的蟋蟀博客园博客，原文链接：http://www.cnblogs.com/xishuai/p/ddd-repository-iunitofwork-and-idbcontext-part-3.html，如需转载请自行联系原作者