首页 > 代码库 > [Architect] Abp 框架原理解析(5) UnitOfWork

[Architect] Abp 框架原理解析(5) UnitOfWork

本节目录

  • 介绍
  • 分析Abp源码
  • 实现UOW

 

介绍

UOW(全称UnitOfWork)是指工作单元.

在Abp中,工作单元对于仓储和应用服务方法默认开启。并在一次请求中,共享同一个工作单元.

同时在Abp中,不仅支持同一个数据库连接,还支持事务处理.

 

分析Abp源码

1.UnitOfWorkRegistrar

技术分享

 

2.ComponentRegistered

技术分享

 

 

3.IsConventionalUowClass

技术分享

 

 

4.Intercept

技术分享

 

 

5.PerformSyncUow

 技术分享

 

 

实现UOW

定义IUnitOfWork

1
2
3
4
5
6
7
8
9
10
11
12
public interface IUnitOfWork
{
    //1.开启事务
    //2.设置Filter(本例中不做演示)
    void Begin(UnitOfWorkOptions options);
    void Complete();
}
public class UnitOfWorkOptions
{
    public bool? IsTransactional { getset; }
}

 

实现uow,在uow中会提供db的创建,这样才能管理到每个db.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
public class EfUnitOfWork : UnitOfWorkBase
{
    public static DbContext DbContext { getset; }
    public static DbContext GetDbContext()
    {
        if (DbContext == null)
        {
            DbContext = new DemoDb();
        }
        return DbContext;
    }
    public override void Begin(UnitOfWorkOptions options)
    {
        if (options.IsTransactional == true)
        {
            CurrentTransaction = new TransactionScope();
        }
    }
    public TransactionScope CurrentTransaction { getset; }
    public override void Complete()
    {
        SaveChanges();
        if (CurrentTransaction != null)
        {
            CurrentTransaction.Complete();
        }
    }
    private void SaveChanges()
    {
        DbContext.SaveChanges();
    }
}
public abstract class UnitOfWorkBase : IUnitOfWork
{
    public virtual void Begin(UnitOfWorkOptions options)
    {
    }
    public virtual void Complete()
    {
    }
}

 

定义与实现仓储层,这里不再做DbProvider.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
public interface IRepository
{
}
public interface ITaskRepository : IRepository
{
    void Add(Task task);
}
public class TaskRepository : ITaskRepository
{
    public void Add(Task task)
    {
        var db = (DemoDb)EfUnitOfWork.GetDbContext();
        db.Tasks.Add(task);
    }
}

 

定义与实现应用层

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
public interface IApplicationService
{
}
public interface ITaskAppService : IApplicationService
{
    void CreateTask(CreateTaskInput input);
}
public class TaskAppService : ITaskAppService
{
    private ITaskRepository _repository;
    public TaskAppService(ITaskRepository repository)
    {
        _repository = repository;
    }
    public void CreateTask(CreateTaskInput input)
    {
        _repository.Add(new Task(input.Name));
    }
}
public class CreateTaskInput
{
    public string Name { getset; }
}

 

定义与实现uow拦截器

1
2
3
4
5
6
7
8
9
10
11
12
13
14
internal class UnitOfWorkInterceptor : IInterceptor
{
    private IUnitOfWork _unitOfWork;
    public UnitOfWorkInterceptor(IUnitOfWork unitOfWork)
    {
        _unitOfWork = unitOfWork;
    }
    public void Intercept(IInvocation invocation)
    {
        _unitOfWork.Begin(new UnitOfWorkOptions());
        invocation.Proceed();
        _unitOfWork.Complete();
    }
}

 

定义在IApplicationService与IRepository接口下拦截

1
2
3
4
5
6
7
8
static void Kernel_ComponentRegistered(string key, Castle.MicroKernel.IHandler handler)
{
    var type = handler.ComponentModel.Implementation;
    if (typeof(IApplicationService).IsAssignableFrom(type) || typeof(IRepository).IsAssignableFrom(type))
    {
        handler.ComponentModel.Interceptors.Add(new InterceptorReference(typeof(UnitOfWorkInterceptor)));
    }
}

 

执行

1
2
3
4
5
6
7
8
9
10
11
12
13
14
static void Main(string[] args)
{
    using (var container = new WindsorContainer())
    {
        container.Register(Component.For<IInterceptor, UnitOfWorkInterceptor>());//先注入拦截器
        container.Register(Component.For<IUnitOfWork, EfUnitOfWork>());
        container.Kernel.ComponentRegistered += Kernel_ComponentRegistered;
        container.Register(Component.For<ITaskAppService, TaskAppService>());
        container.Register(Component.For<ITaskRepository, TaskRepository>());
        var person = container.Resolve<ITaskAppService>();
        person.CreateTask(new CreateTaskInput() { Name = "3333" });
    }
    Console.ReadKey();
}

 

会自动在application method的结尾调用Complete.

另外也可以在uow上定义option为启用事务.在本例中稍作扩展即可实现.

[Architect] Abp 框架原理解析(5) UnitOfWork