ABP使用及框架解析系列 - [Unit of Work part.2-框架实现] - An.Cup.Of.Coffee(4)_H5之家

private void PreventMultipleBegin() { if (_isBeginCalledBefore) {AbpException(); } _isBeginCalledBefore = true; }

private void PreventMultipleComplete() { if (_isCompleteCalledBefore) {AbpException(); } _isCompleteCalledBefore = true; }

UnitOfWorkBase的其他实现与上面的类似（非Unit of Work相关内容略去），便不全贴出。但是上面的代码，大家有发现一个问题吗？就是那些PreventMultiple…方法的实现，用来防止多次Begin、Complete等。这些方法的实现，在单线程下是没有问题的，但是里面并没有加锁，所以不适用于多线程。这是作者的BUG吗？然而，这却并不是BUG，而是设计如此，为何呢？因为在设计上，一个线程共用一个工作单元，也就不存在多线程了。关于这部分内容，后续将会介绍。

IUnitOfWorkManager目的是提供一个简洁的IUnitOfWork管理对象，而IUnitOfWork则提供了整个工作单元需要的所有控制（Begin、SaveChanges、Complete、Dispose）。而具体应该如何保证一个线程共用一个工作单元，如何获取当前的工作单元，则由ICurrentUnitOfWorkProvider进行管控，正如在解析UnitOfWorkManager时，说明了它的Current实际上就是调用ICurrentUnitOfWorkProvider实现对象的Current属性。

ICurrentUnitOfWorkProvider

public interface ICurrentUnitOfWorkProvider { IUnitOfWork Current { get; set; } }

ICurrentUnitOfWorkProvider仅仅只声明了一个Current属性，那么重点让我们来看看Current在实现类（CallContextCurrentUnitOfWorkProvider）中是如何写的吧：

[DoNotWire] public IUnitOfWork Current { get { return GetCurrentUow(Logger); } set { SetCurrentUow(value, Logger); } }

上面标注的DoNotWire是为了不让IoC进行属性注入，Current内部分别调用了GetCurrentUow和SetCurrentUow，要取值，先要设值，让我来先看看set吧：

SetCurrentUow(IUnitOfWork value, ILogger logger) { if (value == null) {//如果在set的时候设置为null，便表示要退出当前工作单元 (logger); return; } unitOfWorkKey = CallContext.LogicalGetData(ContextKey) as string; if (unitOfWorkKey != null) { IUnitOfWork outer; if (UnitOfWorkDictionary.TryGetValue(unitOfWorkKey, out outer)) { if (outer == value) { logger.Warn(); return; } //到这里也就表示当前存在工作单元，那么再次设置工作单元，不是替换掉当前的工作单元而是将当前工作单元作为本次设置的工作单元的外层工作单元 value.Outer = outer; } } unitOfWorkKey = value.Id; if (!UnitOfWorkDictionary.TryAdd(unitOfWorkKey, value)) {AbpException(); } //设置当前线程的工作单元key CallContext.LogicalSetData(ContextKey, unitOfWorkKey); }

ExitFromCurrentUowScope(ILogger logger) { //ContextKey为一个常量字符串 //CallContext可以理解为每个线程的独有key,value集合类，每个线程都会有自己的存储区， // 也就是在线程A中设置一个key为xx的value，在线程B中通过xx取不到，并可以存入相同键的value，但是不会相互覆盖、影响 unitOfWorkKey = CallContext.LogicalGetData(ContextKey) as string; if (unitOfWorkKey == null) {//没有取到值，表示当前无工作单元 logger.Warn(); return; } IUnitOfWork unitOfWork; (!UnitOfWorkDictionary.TryGetValue(unitOfWorkKey, out unitOfWork)) {//根据key没有取到value，从线程集合（CallContext）中释放该key CallContext.FreeNamedDataSlot(ContextKey); return; } //从工作单元集合中移除当前工作单元 UnitOfWorkDictionary.TryRemove(unitOfWorkKey, out unitOfWork); if (unitOfWork.Outer == null) {//如果当前工作单元没有外层工作单元，则从线程集合（CallContext）中释放该key CallContext.FreeNamedDataSlot(ContextKey); return; } (!UnitOfWorkDictionary.TryGetValue(outerUnitOfWorkKey, out unitOfWork)) {//如果当前工作单元有外层工作单元，但是从工作单元集合中没有取到了外层工作单元，那么同样从线程集合（CallContext）中释放该key CallContext.FreeNamedDataSlot(ContextKey); return; } //能到这里，就表示当前工作单元有外层工作单元，并且从工作单元集合中获取到了外层工作单元，那么就设外层工作单元为当前工作单元 CallContext.LogicalSetData(ContextKey, outerUnitOfWorkKey); }

看完set，再让我们来看看get吧：