一个业务功能往往不只由一次数据库请求(或者服务调用)实现。为了功能的完整性,我们希望如果该功能执行一半时出错,则撤销前面已执行的改动。在数据库层面上,事务管理实现了这种完整性需求。在ABP中,一个完整的业务功能称为一个工作单元(Unit of Work,简称UoW)。工作单元代表一种完整的、原子性的操作。即一个工作单元包含的步骤要么全部被执行,要么都不被执行。如果执行一半时出现异常,则必须讲已执行的步骤还原。通常我们将事务管理实现在工作单元中。下面我们从ABP源码入手研究如何使用工作单元。
ABP工作单元(UoW)的工作原理ABP默认将工作单元应用在Repositories、 Application Services、MVC控制器和Web API控制器等组件。也就是说,这些组件的每个方法都是一个工作单元。ABP文档对工作单元的原理讲得不是很详细,所以我们只能通过源码进行研究。这里我们以MVC控制器为例来了解一下ABP工作单元大致的工作原理。源码分析比较枯燥,最好配套ABP源码阅读,或者跳到后面看粗体字结论。
ABP在Web模块初始化时注册了过滤器AbpMvcUowFilter。AbpMvcUowFilter在请求处理前(OnActionExecuting方法)调用UnitOfWorkManager.Begin方法来开始一个工作单元。UnitOfWorkManager.Begin创建一个IUnitOfWork的实例并赋值给ICurrentUnitOfWorkProvider.Current,然后调用IUnitOfWork.Begin方法开始一个工作单元。在请求处理结束后(OnActionExecuted方法)如果处理过程没有异常就调用IUnitOfWork.Complete方法完成工作单元,并且无论请求处理是否成功,都调用IUnitOfWork.Dispose来结束工作单元。
ABP提供了一个实现IUnitOfWork的抽象基类UnitOfWorkBase,另外还有个继承了UnitOfWorkBase的类NullUnitOfWork。NullUnitOfWork定义上面有一段注释如此写到:
/// <summary> /// Null implementation of unit of work. /// It's used if no component registered for <see cref="IUnitOfWork"/>. /// This ensures working ABP without a database. /// </summary> public sealed class NullUnitOfWork : UnitOfWorkBaseNullUnitOfWork是一个“空”的工作单元,它不会做任何操作。如果我们没有在IoC容器中注册其它IUnitOfWork的实现类,则ABP默认使用不做任何事的NullUnitOfWork作为工作单元。所以如果我们要做一些保证功能完整性的工作(比如开启数据库事务),就要实现IUnitOfWork并注册到IoC容器。
阅读UnitOfWorkBase可以看到,UnitOfWorkBase分别在Begin方法、Complete方法和Dispose方法中调用了BeginUow方法、CompleteUow方法和DisposeUow方法。我们需要重写的主要是BeginUow、CompleteUow和DisposeUow这三个方法。
通过源码简单了解了原理后,我们后面写代码要注意的有下面几点:
写一个继承UnitOfWorkBase的类UnitOfWork,并实现接口ITransientDependency保证UnitOfWork被注册到IoC容器;
重写方法UnitOfWorkBase.BeginUow,实现工作单元开始时的启动操作;
重写方法UnitOfWorkBase.CompleteUow,实现工作单元正常结束时的保存操作;
重写方法UnitOfWorkBase.DisposeUow,实现工作单元结束时的清理操作;
通过ICurrentUnitOfWorkProvider.Current来获取当前的工作单元。
重写SessionProvider,并实现工作单元在之前文章(手工搭建基于ABP的框架(2) - 访问数据库)实现的LocalDbSessionProvider中,为了追求代码简单,我们粗暴地用一个实质上是全局的变量来保存数据库Session,在每次访问数据库时,flush上一个Session并创建新Session。另一方面,数据库连接的配置、Session的创建保存、以及Session的提供都胡乱地放在了这个类里。这其实是非常不合理而且会引发很多问题的实现方法。
下面我们重新设计这一块逻辑。我们将LocalDbSessionProvider所负责的功能拆分,分别实现在LocalDbSessionConfiguration、UnitOfWork和UnitOfWorkLocalDbSessionProvider三个类中:
LocalDbSessionConfiguration,单例。实现数据库连接配置,提供数据库Session工厂。
public class LocalDbSessionConfiguration : ILocalDbSessionConfiguration, IDisposable { protected FluentConfiguration FluentConfiguration { get; private set; } public ISessionFactory SessionFactory { get; } public LocalDbSessionConfiguration() { FluentConfiguration = Fluently.Configure(); // 数据库连接串 var connString = "data source=|DataDirectory|MySQLite.db;"; FluentConfiguration // 配置连接串 .Database(SQLiteConfiguration.Standard.ConnectionString(connString)) // 配置ORM .Mappings(m => m.FluentMappings.AddFromAssembly(Assembly.GetExecutingAssembly())); // 生成session factory SessionFactory = FluentConfiguration.BuildSessionFactory(); } public void Dispose() { SessionFactory.Dispose(); } }