概述

在上一篇 《》 的文章中，我们讲述了有关仓储的概念和使用规范。仓储为聚合提供了持久化到本地的功能，但是在持久化的过程中，有时一个聚合根中的各个领域对象会分散到不同的数据库表里面；又或者是一个用例操作需要操作多个仓储；而这些操作都应该要么同时成功，要么同时失败，因此就需要为这一系列操作提供事务的支持，而事务管理就是由工作单元来提供的。在上一篇中，可能已经提到了工作单元，但是仅仅是一笔带过，现在我们就来详细的探究该如何更好的来实现工作单元。（文章的代码片段都使用的是C#,案例项目也是基于 DotNet Core 平台）。

直接看东西

在上一篇文章中，已经为大家提供了一个Github的Demo。如果已经下载过该Demo的同学，您现在直接进行Pull就可以获得最新的版本了；如果还没有下载该Demo的同学也可以戳下方的跳转链接获取。

GitHub 地址，点击直达哟

在这里我们可以先来看一下，该项目的应用代码是什么样子：

复制代码

[HttpPost]public ActionResult
    
      Add(){    //使用仓储来处理聚合    _itineraryRepository.Add(        new Itinerary(            "奥特曼",            "赛文奥特曼",            "杰克奥特曼",            "佐菲奥特曼",            "泰罗奥特曼"));    _itineraryRepository.Add(        new Itinerary(            "盖亚奥特曼",            "戴拿奥特曼",            "阿古茹奥特曼",            "迪迦奥特曼", ""));    return "success";}[HttpGet]public ActionResult
     
       Get(){    var count = _itineraryRepository.GetCount();    return count;}
     
    

这是在Aspnet Core的Controller中的代码，也就是对外提供的Api。可以看到我们仅仅只是通过仓储的调用就完成了所有的操作。（ps:原谅我该演示api没有遵循restful风格(￣▽￣)"，还有就是那些奥特曼。。。）。

您可能会说，这里没有做操作，那肯定是在 ItineraryRepository 里面做了手脚。好吧，下面我们来看看该仓储的实现。

复制代码

public class ItineraryRepository        : EFRepository
    
     {    public void Add(Itinerary itinerary) => DbContext.Set
     
      ().Add(itinerary);}
     
    

是的，它也只有这么一点点代码。而作为后期的业务扩展和维护，我们只需要完善我们的Itinerary聚合（为它扩展行为和增加实体或值对象）以及ItineraryRepository仓储（为它添加对外检索意图的方法）就可以了。

这种做法的好处可能您很快就能发现：在我们代码中处处都是关于领域对象的操作，尽可能的避免其它基础构建或功能支持组件来干扰程序。除了代码量的减少之外，它也让可读性有着明显的提高，如果在此基础上能够构建出明确而干净的聚合根，那么您的程序将具备更高的可扩展性。

好吧，回到我们今天的主题：工作单元。其实上面的代码就是对仓储中工作单元的巧妙运用，它其实在后面默默的支持着程序的正常运转，这是在调用层面上我们完全感觉不到它的存在而已。下面就为您介绍它是怎么工作和实现的。

什么是工作单元

按照国际管理呢，这一章节都是解读有关原著《领域驱动设计:软件核心复杂性应对之道》 中的解释。但是！！！有关工作单元的概念在书里并没有被明确的提及到。所以为了证明我们确确实实是在前人的基础理念上来实践，而不是胡编乱造自己随便弄了一个概念出来。我特地去找了另外一本较为权威的领域驱动设计教材：《领域驱动设计模式、原理与实践》 。在该书中对工作单元的解释如下：

事务管理主要与应用程序服务层有关。存储库只与使用聚合根的单一集合的管理有关，而业务用例可能会造成对多个类型聚合的更新。事务管理是由工作单元处理的。工作单元模式的作用是保持追踪业务任务期间聚合的所有变化。一旦所有的变化都已发生，则之后工作单元会协调事务中持久化存储的更新。如果在将变更提交到数据存储的中途出现了问题，那么要确保不损坏数据完整性的话，就要回滚所有的变更以确保数据保持有效的状态。

其实上文的话真的很好理解（相对于原著而言(￣y▽,￣)╭ ）。首先我们可以得到的第一个结论：事务管理其实是应用服务层干的事。第二个结论：事务的协调管理都是由工作单元来负责的

所以，我们千万不能因为工作单元和仓储有联系就将它放置在领域层里面：事务的提供往往是由数据库管理程序来提供的，而这一类组件我们一般将它们放置在基础构架层，而领域层可以依赖于基础构架层，所以千万要注意，保持您的领域层足够干净，不要让其它的东西干扰它，也更不要将事务处理这类东西放到了您的领域层来。（这一点，您会在后期MiCake<米蛋糕>的使用中看到详细的案例）。

如何实现工作单元

实现工作单元，就是要实现仓储中的事务操作。您可能已经看到过有些实现Repository的框架，它的写法是注入一个unitOfWork,然后从uow中提取一个仓储，然后再用仓储来完成聚合根的持久化操作。类似的代码就像这样：

复制代码

var yourRepository = uow.GetRepository
    
     ();yourRepository.Add(yourEntity);uow.Commit();
    

这样做没有一点点的问题，而且是对工作单元和仓储模式的完美实现。uow工作单元中维持了一个事务，从该工作单元中创建的每一个仓储都可以获得该事务，仓储完成了自己的操作之后，工作单元使用Commit方法告诉事务管理器，该事务完成。

复制代码

夏目去参加了妖怪的聚会，一回到家，猫咪老师就发现了它沾染了妖怪的味道

当仓储的操作沾染上了工作单元的事务，它也就受到了事务的管理

如果您喜欢这种实现模式，可以参考 threenine的Threenine.Data项目。

懒的模式

其实在刚开始，为 MiCake(米蛋糕) 选取工作单元实现方案的时候，我也打算采用这种方式。但是在思考了一天之后，我还是放弃了。因为我发现这种模式在完成每一次仓储操作的时候，必须要从工作单元中去获取。在Aspnet Core中，不得不在Controller中注入工作单元对象，然后再从该对象里面去获取仓储。这显然削弱了依赖注入所为我们提供的依赖阅读性（原本在构造函数中，我能看出我需要注入的是A仓储，但是现在我看到的只有工作单元）。

其实最重要的一点就是：我太懒啦 o_o ....。为什么每次都要去多写一个uow.GetXXXXX()。每使用一个仓储就要多写一次获取语句，我就不能好好的只使用仓储吗？所以在这个想法的强烈刺激下，我选取了另外的实现方法。

接下来，就让我们来实现最开始演示代码中的工作单元吧。哦，对了，忘记说了，无论是演示的Github Demo还是本次的博文，我们都选取了Entity Framework Core来作为数据持久组件。所以有些小伙伴会说，那我使用Dapper或者原生的ADO怎么办？其实思路都是一样的，您也可以在看了EFCore的版本后，自己写出对应的工作单元版本。如果有机会的话，欢迎在Github的Demo上直接添加，就可以提交供更多的同学参考啦。

实现思路

• 找出当前数据库持久组件中具有事务特征的对象（比如在EF中就是DbContext）

• 创建一个容器去容纳这些对象

• 工作单元就是该容器的实现，它掌管了这些事务对象，并对外公布了提交事务的方法

• 工作单元管理器负责了对工作单元的创建工作

脑袋里有了这些还比较模糊的交互对象之后，我们可以来想一下一个仓储完成添加聚合根的操作是怎么样的：

• 在访问该API之前：使用工作单元管理器创建一个工作单元

• 访问API中的仓储时候：构造一个事务特征对象，并开启一个事务

• 事务开启完成之后：将该事务特征对象尝试放入到当前工作单元

• 仓储事务操作完成后：调用工作单元的提交方法，完成事务的提交，保证仓储的数据一致。

• 事务完成后：释放上面的各个对象

虽然步骤好像有5步，但总结下来，就是将具有事务的对象放置到工作单元中，让它去负责提交。对！就是这么简单，该方法与上面那种从工作单元中获取仓储的方法想法，它是往工作单元中提交。所以，我们此时可以构造出一个伪代码出来，大致理解它的实现：

复制代码

//1、使用工作单元管理器创建一个工作单元    using (var uow = unitOfWorkManager.Create())    {        //2、构造事务特征对象，开启事务并注册到工作单元        RegisteTransactonFeature(DbContext);        //3、执行仓储中的内容        DbContext.Set
    
     ().Add(itinerary)        //4、工作单元保存提交        uow.SaveChanges();        //5、dispose    }
    

至少到目前，我们可以抽象出上面的各个对象了。

您也可以先自己尝试着想一想，每个对象接口应该实现什么功能（方法）。

复制代码

//首先是事务特征对象，它提供了事务的基本Commit和Rollback方法public interface ITransactionFeature{    public bool IsCommit { get; }    public bool IsRollback { get; }    void Commit();    Task CommitAsync(CancellationToken cancellationToken = default);    void Rollback();    Task RollbackAsync(CancellationToken cancellationToken = default);}//然后是事务特征容器，它具有增加删除事务特征对象的方法public interface ITransactionFeatureContainer{    void RegisteTranasctionFeature(string key, ITransactionFeature TransactionFeature);    ITransactionFeature GetOrAddTransactionFeature(string key, ITransactionFeature TransactionFeature);    ITransactionFeature GetTransactionFeature(string key);    void RemoveTransaction(string key);}//接下来是工作单元，它实现了事务特征容器，并且对外提供提交的方法public interface IUnitOfWork : ITransactionFeatureContainer{    Guid ID { get; }    bool IsDisposed { get; }    void SaveChanges();    Task SaveChangesAsync(CancellationToken cancellationToken = default);    void Rollback();    Task RollbackAsync(CancellationToken cancellationToken = default);}//最后是工作单元管理器，它提供了创建工作单元的方法public interface IUnitOfWorkManager : IUnitOfWokrProvider, IDisposable{    IUnitOfWork Create();}

落地代码

在构建出接口之后，我们就可以写出具体的实现类了。首先是实现工作单元（UnitOfWork）对象。（由于具体代码实现较多，讲解部分只选取了核心部分，完整代码可以参考Github的项目）

复制代码

public class UnitOfWork : IUnitOfWork{    private readonly Dictionary
    
      _transactionFeatures;    public UnitOfWork()    {        _transactionFeatures = new Dictionary
     
      ();    }    //往容器中添加事物特征对象    public virtual ITransactionFeature GetOrAddTransactionFeature(        [NotNull]string key,        [NotNull] ITransactionFeature transcationFeature)    {        if (_transactionFeatures.ContainsKey(key))            return _transactionFeatures.GetValueOrDefault(key);        _transactionFeatures.Add(key, transcationFeature);        return transcationFeature;    }    //对外提供的保存方法，执行该方法时调用容器内所有事物特征对象的Commit方法    public virtual void SaveChanges()    {        foreach (var transactionFeature in _transactionFeatures.Values)        {            transactionFeature.Commit();        }    }}
     
    

接下来就是与ORM框架关联最深的事务特征对象的实现了，由于我们选取了EF，所以此处应该实现EF版本的事务特征对象：

复制代码

public class EFTransactionFeature : ITransactionFeature{    private IDbContextTransaction _dbContextTransaction;    private DbContext _dbContext;    public EFTransactionFeature(DbContext dbContext)    {        _dbContext = dbContext;    }    //设置事务    public void SetTransaction(IDbContextTransaction dbContextTransaction)    {        _isOpenTransaction = true;        _dbContextTransaction = dbContextTransaction;    }    public void Commit()    {        if (IsCommit)            return;        IsCommit = true;        //EF 事务的提交        _dbContext.SaveChanges();        _dbContextTransaction?.Commit();    }}

建立好了这两个对象之后，其实我们只需要一个流转过程就可以实现工作单元了。这个流程就是将事务特征对象添加到工作单元中，但是我们应该在什么时候将它添加进去呢？看过第一版Github代码的小伙伴可能知道，在仓储调用的时候就可以完成该操作。当时在第一版中，我们的实现代码是这样的：

复制代码

public class EFRepository{    protected IUnitOfWorkManager UnitOfWorkManager { get; private set; }    protected DbContext DbContext { get; private set; }    public EFRepository(IUnitOfWorkManager unitOfWorkManager, DbContext dbContext)    {        UnitOfWorkManager = unitOfWorkManager;        DbContext = dbContext;    }    public void Add(TAggregateRoot aggregateRoot)    {        RegistUnitOfWork(DbContext);        DbContext.Set
    
     ().Add(aggregateRoot);    }    private void RegistUnitOfWork(DbContext dbContext)    {        string key = $"EFTransactionFeature - {dbContext.ContextId.InstanceId.ToString()}";        unitOfWork.ResigtedTransactionFeature(key, new EFTransactionFeature(DbContext));    }}
    

在每一次进行仓储操作的时候，都调用了一个RegistUnitOfWork的方法，来完成事务特征对象和工作单元的流转工作。但是很快您就能发现问题：EFRepository是我们实现的一个基类，以后所有的仓储操作都继承该类来完成操作，那不是每扩展一个方法，我都要在该方法中写一句注册代码？如果我忘记写了怎么办。还有一点，该注册过程并没有开启一个事务，那么事务是怎么来的呢？

那么怎么才能避免用户每一次都要去显示调用注册呢，而是让用户在不知不觉中就完成了该操作。所以我们得思考在每一个方法中，用户都一定会写的代码是什么，然后在该代码上下手。可能您已经想到了，DbContext！！！是的，每一个方法里，用户都会去写DbContext，所以我们可以在他获取DbContext的时候就完成注册操作。所以，优化后的代码就是这样的：

复制代码

public class EFRepository{    public virtual TDbContext DbContext    {        get => _dbContextFactory.CreateDbContext();    }    public void Add(TAggregateRoot aggregateRoot)    {        DbContext.Set
    
     ().Add(aggregateRoot);    }}
    

而该_dbContextFactory的实现就更简单了，他要完成的任务就是注册到工作单元并且开启事务。

复制代码

internal class UowDbContextFactory
    
     {    private readonly IUnitOfWorkManager _uowManager;    public UowDbContextFactory(IUnitOfWorkManager uowManager)    {        _uowManager = uowManager;    }    public TDbContext CreateDbContext()    {        AddDbTransactionFeatureToUow(currentUow, DbContext);        return wantedDbContext;    }    private void AddDbTransactionFeatureToUow(IUnitOfWork uow, TDbContext dbContext)    {        string key = $"EFCore - {dbContext.ContextId.InstanceId.ToString()}";        var efFeature = uow.GetOrAddTransactionFeature(key, new EFTransactionFeature(dbContext));        if (IsFeatureNeedOpenTransaction(uow, efFeature))        {            var dbcontextTransaction = dbContext.Database.BeginTransaction();            efFeature.SetTransaction(dbcontextTransaction);        }    }    private bool IsFeatureNeedOpenTransaction(IUnitOfWork uow, EFTransactionFeature efFeature)    {        return !efFeature.IsOpenTransaction;    }}
    

dbContext.Database.BeginTransaction是EF为我们提供的手动开启事务的方法。如果您尝试实现另外ORM版本的工作单元，想一下在该ORM中是怎么开启的事务。

此时，我们就已经实现了工作单元的流转了，那么还有一个问题就是：我们怎么默认去实现一个工作单元，而不是每一次都需要手动去开启并提交。

AspNet Core为我们提供了很好的拦截方法。第一种方法： 我们可以在中间件中完成，因为所有的请求都要穿过中间件，我们可以在方法到API之前就开启事务，等API访问结束后就提交事务。第二种方法： 通过IActionFilter等周期接口来完成。本案例选取了第一种实现方法，您也可以根据您自己的爱好选取自己的实现方式。

缺陷

到这里我们已经实现了像上面Demo版本的工作单元，但是该工作单元其实还有许多特性没有实现：

• 一个业务操作（一个API）中没有创建多个工作单元的能力

• 目前事务的操作来源于EF Core的支持，如果项目存在多种数据访问方式（比如一个EF，一个ADO），它们之间如何依靠工作单元来完成事务

• 没有识别什么时候需要开启工作单元，如果一个操作仅仅需要获取数据，其实我们是不需要开启工作单元的

不过如果您的项目仅仅使用了一种ORM框架并且只需要开启一个工作单元，那么可以尝试使用该实现。

在实现MiCake真正的工作单元中，我尝试了很多方法来解决上面的问题。在后面的文章中，您也会看到MiCake真正的工作单元。

附上一个当时写工作单元的手记（￣︶￣）↗

总结

本来这篇文章不打算写在《如何运用领域驱动设计》这个系列的，但是后来纠结了一下，还是纳入了该系列。由于该篇文章是实现工作单元的，所以代码量就比较大，希望不会给您造成阅读上的困难。下一篇的文章，是一个谈了很久的问题————持久化值对象，现在终于是时候该解决它了。在本次Demo中您看到的聚合根Itinerary所有的属性都是string，很显然这是不符合常理的，所以在下一次就要让它成为真正的领域对象。（ps:改成真正的领域对象后，感觉都可以单体DDD应用落地了呢。（￣︶￣）↗醒醒！少年。）为了您不错过下一篇文章的内容，您也可也点击博客园右上角的关注，这样就能及时收到更新了哟。