1 /// <summary>
 2     /// This class is used to register interceptor for needed classes for Unit Of Work mechanism.
 3     /// </summary>
 4     internal static class UnitOfWorkRegistrar
 5     {
 6         /// <summary>
 7         /// Initializes the registerer.
 8         /// </summary>
 9         /// <param name="iocManager">IOC manager</param>
10         public static void Initialize(IIocManager iocManager)
11         {
12             iocManager.IocContainer.Kernel.ComponentRegistered += ComponentRegistered;
13         }
14
15         private static void ComponentRegistered(string key, IHandler handler)
16         {
17             if (UnitOfWorkHelper.IsConventionalUowClass(handler.ComponentModel.Implementation))
18             {
19                 //Intercept all methods of all repositories.
20                 handler.ComponentModel.Interceptors.Add(new InterceptorReference(typeof(UnitOfWorkInterceptor)));
21             }
22             else if (handler.ComponentModel.Implementation.GetMethods(BindingFlags.Instance | BindingFlags.Public | BindingFlags.NonPublic).Any(UnitOfWorkHelper.HasUnitOfWorkAttribute))
23             {
24                 //Intercept all methods of classes those have at least one method that has UnitOfWork attribute.
25                 //TODO: Intecept only UnitOfWork methods, not other methods!
26                 handler.ComponentModel.Interceptors.Add(new InterceptorReference(typeof(UnitOfWorkInterceptor)));
27             }
28         }
29     }

 1 private void PerformUow(IInvocation invocation, UnitOfWorkOptions options)
 2         {
 3             if (AsyncHelper.IsAsyncMethod(invocation.Method))
 4             {
 5                 PerformAsyncUow(invocation, options);
 6             }
 7             else
 8             {
 9                 PerformSyncUow(invocation, options);
10             }
11         }
12
13         private void PerformSyncUow(IInvocation invocation, UnitOfWorkOptions options)
14         {
15             using (var uow = _unitOfWorkManager.Begin(options))
16             {
17                 invocation.Proceed();
18                 uow.Complete();
19             }
20         }

 1 /// <summary>
 2     /// Unit of work manager.
 3     /// Used to begin and control a unit of work.
 4     /// </summary>
 5     public interface IUnitOfWorkManager
 6     {
 7         /// <summary>
 8         /// Gets currently active unit of work (or null if not exists).
 9         /// </summary>
10         IActiveUnitOfWork Current { get; }
11
12         /// <summary>
13         /// Begins a new unit of work.
14         /// </summary>
15         /// <returns>A handle to be able to complete the unit of work</returns>
16         IUnitOfWorkCompleteHandle Begin();
17
18         /// <summary>
19         /// Begins a new unit of work.
20         /// </summary>
21         /// <returns>A handle to be able to complete the unit of work</returns>
22         IUnitOfWorkCompleteHandle Begin(TransactionScopeOption scope);
23
24         /// <summary>
25         /// Begins a new unit of work.
26         /// </summary>
27         /// <returns>A handle to be able to complete the unit of work</returns>
28         IUnitOfWorkCompleteHandle Begin(UnitOfWorkOptions options);
29     }

 1     /// <summary>
 2     /// Defines a unit of work.
 3     /// This interface is internally used by ABP.
 4     /// Use <see cref="IUnitOfWorkManager.Begin()"/> to start a new unit of work.
 5     /// </summary>
 6     public interface IUnitOfWork : IActiveUnitOfWork, IUnitOfWorkCompleteHandle
 7     {
 8         /// <summary>
 9         /// Unique id of this UOW.
10         /// </summary>
11         string Id { get; }
12
13         /// <summary>
14         /// Reference to the outer UOW if exists.
15         /// </summary>
16         IUnitOfWork Outer { get; set; }
17
18         /// <summary>
19         /// Begins the unit of work with given options.
20         /// </summary>
21         /// <param name="options">Unit of work options</param>
22         void Begin(UnitOfWorkOptions options);
23     }