public interface IUnitOfWork : IActiveUnitOfWork, IUnitOfWorkCompleteHandle{/// <summary>/// Begins the unit of work with given options./// 开始 unit of work的一些配置UnitOfWorkOptions，主要是事务的级别，超时时间，配置文件等/// </summary>/// <param name="options">Unit of work options</param>void Begin(UnitOfWorkOptions options);}public interface IUnitOfWorkCompleteHandle : IDisposable{/// <summary>/// Completes this unit of work./// It saves all changes and commit transaction if exists./// 统一事务提交/// </summary>void Complete();/// <summary>/// Completes this unit of work./// It saves all changes and commit transaction if exists./// 异步的Complete方法/// </summary>Task CompleteAsync();}

    public abstract class UnitOfWorkBase : IUnitOfWork{public UnitOfWorkOptions Options { get; private set; }/// <summary>/// 开始UnitOfWork的一些配置，和事务的初始化/// </summary>/// <param name="options"></param>public void Begin(UnitOfWorkOptions options){if (options == null){throw new ArgumentNullException("options");}PreventMultipleBegin();Options = options; //TODO: Do not set options like that!SetFilters(options.FilterOverrides);BeginUow();}/// <summary>/// 事务的提交，异常的捕获/// </summary>public void Complete(){PreventMultipleComplete();try{CompleteUow();_succeed = true;OnCompleted();}catch (Exception ex){_exception = ex;throw;}}/// <summary>/// 结束事务，失败就回滚/// </summary>public void Dispose(){if (IsDisposed){return;}IsDisposed = true;if (!_succeed){OnFailed(_exception);}DisposeUow();OnDisposed();}}

    public class EfUnitOfWork : UnitOfWorkBase, ITransientDependency{private readonly IDictionary<Type, DbContext> _activeDbContexts;private readonly IIocResolver _iocResolver;private TransactionScope _transaction;/// <summary>/// Creates a new <see cref="EfUnitOfWork"/>./// </summary>public EfUnitOfWork(IIocResolver iocResolver, IUnitOfWorkDefaultOptions defaultOptions): base(defaultOptions){_iocResolver = iocResolver;_activeDbContexts = new Dictionary<Type, DbContext>();}protected override void BeginUow(){if (Options.IsTransactional == true){var transactionOptions = new TransactionOptions{IsolationLevel = Options.IsolationLevel.GetValueOrDefault(IsolationLevel.ReadUncommitted),};if (Options.Timeout.HasValue){transactionOptions.Timeout = Options.Timeout.Value;}_transaction = new TransactionScope(TransactionScopeOption.Required,transactionOptions,Options.AsyncFlowOption.GetValueOrDefault(TransactionScopeAsyncFlowOption.Enabled));}}public override void SaveChanges(){_activeDbContexts.Values.ForEach(SaveChangesInDbContext);}
...

    // 账号转账领域服务类public class AccountService{private readonly IAccountRepository _productRepository;private readonly IUnitOfWork _unitOfWork;public AccountService(IAccountRepository productRepository, IUnitOfWork unitOfWork){_productRepository = productRepository;_unitOfWork = unitOfWork;            }public void Transfer(Account from, Account to, decimal amount){if (from.Balance >= amount){from.Balance -= amount;to.Balance += amount;_productRepository.Save(from);_productRepository.Save(to);_unitOfWork.Commit();}} }

    internal class UnitOfWorkInterceptor : IInterceptor{private readonly IUnitOfWorkManager _unitOfWorkManager;public UnitOfWorkInterceptor(IUnitOfWorkManager unitOfWorkManager){_unitOfWorkManager = unitOfWorkManager;}public void Intercept(IInvocation invocation){if (_unitOfWorkManager.Current != null){//Continue with current uowinvocation.Proceed();return;}var unitOfWorkAttr = UnitOfWorkAttribute.GetUnitOfWorkAttributeOrNull(invocation.MethodInvocationTarget);if (unitOfWorkAttr == null || unitOfWorkAttr.IsDisabled){//No need to a uowinvocation.Proceed();return;}//No current uow, run a new onePerformUow(invocation, unitOfWorkAttr.CreateOptions());}

    [AttributeUsage(AttributeTargets.Method)]public class UnitOfWorkAttribute : Attribute{/// <summary>/// Is this UOW transactional?/// Uses default value if not supplied./// </summary>public bool? IsTransactional { get; private set; }/// <summary>/// Timeout of UOW As milliseconds./// Uses default value if not supplied./// </summary>public TimeSpan? Timeout { get; private set; }

        [UnitOfWork]public virtual async Task<AbpLoginResult> LoginAsync(string userNameOrEmailAddress, string plainPassword, string tenancyName = null){if (userNameOrEmailAddress.IsNullOrEmpty()){throw new ArgumentNullException("userNameOrEmailAddress");}if (plainPassword.IsNullOrEmpty()){throw new ArgumentNullException("plainPassword");}using (_unitOfWorkManager.Current.DisableFilter(AbpDataFilters.MayHaveTenant)){TUser user;if (!_multiTenancyConfig.IsEnabled){using (_unitOfWorkManager.Current.EnableFilter(AbpDataFilters.MayHaveTenant)){//Log in with default denantuser = await FindByNameOrEmailAsync(userNameOrEmailAddress);if (user == null){return new AbpLoginResult(AbpLoginResultType.InvalidUserNameOrEmailAddress);}}}

    internal static class UnitOfWorkRegistrar{/// <summary>/// Initializes the registerer./// </summary>sssss/// <param name="iocManager">IOC manager</param>public static void Initialize(IIocManager iocManager){iocManager.IocContainer.Kernel.ComponentRegistered += ComponentRegistered;}/// <summary>/// 拦截注册事件 /// </summary>/// <param name="key"></param>/// <param name="handler"></param>private static void ComponentRegistered(string key, IHandler handler){if (UnitOfWorkHelper.IsConventionalUowClass(handler.ComponentModel.Implementation)){//判断如果是IRepository和IApplicationService，就注册动态代理 Intercept all methods of all repositories.handler.ComponentModel.Interceptors.Add(new InterceptorReference(typeof(UnitOfWorkInterceptor)));}else if (handler.ComponentModel.Implementation.GetMethods(BindingFlags.Instance | BindingFlags.Public | BindingFlags.NonPublic).Any(UnitOfWorkHelper.HasUnitOfWorkAttribute)){//判断如果是被标识了UnitOfWork attribute的就注册动态代理 Intercept all methods of classes those have at least one method that has UnitOfWork attribute.//TODO: Intecept only UnitOfWork methods, not other methods!handler.ComponentModel.Interceptors.Add(new InterceptorReference(typeof(UnitOfWorkInterceptor)));}}}

        public static bool HasUnitOfWorkAttribute(MemberInfo methodInfo){return methodInfo.IsDefined(typeof(UnitOfWorkAttribute), true);}