如何运用领域驱动设计 - 工作单元

新年伊始，祝大家喜乐如意，爱和幸福“鼠”不尽！♫. ♪~♬.♩♫~

概述

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

直接看东西

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

GitHub 地址，点击直达哟

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

[HttpPost]
public ActionResult<string> Add()
{
    //使用仓储来处理聚合
    _itineraryRepository.Add(
        new Itinerary(
            "奥特曼",
            "赛文奥特曼",
            "杰克奥特曼",
            "佐菲奥特曼",
            "泰罗奥特曼"));

    _itineraryRepository.Add(
        new Itinerary(
            "盖亚奥特曼",
            "戴拿奥特曼",
            "阿古茹奥特曼",
            "迪迦奥特曼", ""));

    return "success";
}

[HttpGet]
public ActionResult<long> Get()
{
    var count = _itineraryRepository.GetCount();
    return count;
}

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

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

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

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

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

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

什么是工作单元

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

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

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

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

如何实现工作单元

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

var yourRepository = uow.GetRepository<yourRepository>();
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<Itinerary>().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<string, ITransactionFeature> _transactionFeatures;

    public UnitOfWork()
    {
        _transactionFeatures = new Dictionary<string, ITransactionFeature>();
    }

    //往容器中添加事物特征对象
    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<TAggregateRoot>().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<TAggregateRoot>().Add(aggregateRoot);
    }
}

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

internal class UowDbContextFactory<TDbContext>
{
    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应用落地了呢。（￣︶￣）↗醒醒！少年。）为了您不错过下一篇文章的内容，您也可也点击博客园右上角的关注，这样就能及时收到更新了哟。