首页 > 代码库 > UnitOfWork

UnitOfWork

接上文 项目架构开发:数据访问层之IQuery

技术分享

 

本章我们继续IUnitOfWork的开发,从之前的IRepository接口中就可以看出,我们并没有处理单元事务,

数据CUD每次都是立即执行的,这样有一些不好的地方,比如数据访问次数会增多,一笔数据的完整性无法保证

比如:批量新增2条记录,只有一条成功,这样的情况就应该Rollback;可能有人已经想到用数据库自带的事务保证完整性

虽然是可以,但是这样就不可避免地在业务层耦合SqlTransaction,我不想这种情况出现;

所以我们用windows自带的分布式事务TransactionScope来实现,TransactionScope可以实现多个数据库的事务锁,这点就比SqlTransaction强一些

当然我了解到分布式事务是不一般的复杂的,那些更好的处理方式我还没有掌握,所以先这么写吧,有这方面经验的朋友欢迎提建议

 

IUnitOfWork.cs

1     public interface IUnitOfWork<T> where T : class
2     {
3         void RegisterAdd(T entity, Action callback);
4         void RegisterUpdate(T entity, Action callback);
5         void RegisterDelete(T entity, Action callback);
6         void Commit();
7     }

注册新增操作:RegisterAdd

注册更新操作:RegisterUpdate

注册修改操作:RegisterDelete

提交数据:Commit

 

工作单元的实现

 1 public class UnitOfWork<T> : IUnitOfWork<T> where T : class
 2     {
 3         private Dictionary<T, Action> addEntities;
 4         private Dictionary<T, Action> updateEntities;
 5         private Dictionary<T, Action> deleteEntities;
 6 
 7         public UnitOfWork()
 8         {
 9             addEntities = new Dictionary<T, Action>();
10             updateEntities = new Dictionary<T, Action>();
11             deleteEntities = new Dictionary<T, Action>();
12         }
13 
14         public void RegisterAdd(T entity, Action callback)
15         {
16             this.addEntities.Add(entity, callback);
17         }
18 
19         public void RegisterUpdate(T entity, Action callback)
20         {
21             this.updateEntities.Add(entity, callback);
22         }
23 
24         public void RegisterDelete(T entity, Action callback)
25         {
26             this.deleteEntities.Add(entity, callback);
27         }
28 
29         public void Commit()
30         {
31             using (TransactionScope scope = new TransactionScope())
32             {
33                 foreach (var entity in deleteEntities.Keys)
34                 {
35                     this.deleteEntities[entity]();
36                 }
37 
38                 foreach (var entity in updateEntities.Keys)
39                 {
40                     this.updateEntities[entity]();
41                 }
42 
43                 foreach (var entity in addEntities.Keys)
44                 {
45                     this.addEntities[entity]();
46                 }
47 
48                 scope.Complete();
49             }
50         }
51     }

这里我们用了Action,这个是无返回值委托方法,如果大家需要返回值可以用Func

实现我们还是用DapperRepository,修改一下。在新增方法中注册数据持久化方法

技术分享

 

好了,现在我们来看看测试结果

 

测试工作单元

技术分享

 

还是用原来那个DapperRepositoryTest

我们修改一下实现过程:

单个新增

 技术分享

 

批量新增

技术分享

 

运行结果

技术分享

 

 我们把表结构改一下,名称改成不能为null

技术分享

 

AddBatch也要修改,正常的执行结果一个是抛出异常

 1         [TestMethod]
 2         public void AddBatch()
 3         {
 4             try
 5             {
 6                 var loginUser1 = new LoginUser()
 7                 {
 8                     Id = Guid.NewGuid(),
 9                     LoginName = "lanxiaoke-" + Guid.NewGuid().ToString(),
10                     Password = "mima1987",
11                     IsEnabled = 1,
12                     CreateTime = DateTime.Now
13                 };
14 
15                 var loginUser2 = new LoginUser()
16                 {
17                     Id = Guid.NewGuid(),
18                     //LoginName = "lanxiaoke-" + Guid.NewGuid().ToString(),
19                     Password = "mima1987",
20                     IsEnabled = 1,
21                     CreateTime = DateTime.Now
22                 };
23 
24                 var list = new List<LoginUser>();
25                 list.Add(loginUser1);
26                 list.Add(loginUser2);
27 
28                 repository.AddBatch(list);
29                 unitOfWork.Commit();
30             }
31             catch (Exception ex)
32             {
33                 var err = ex.Message;
34             }
35 
36             //long count = repository.Count(t => t.LoginName.In(new string[] { loginUser1.LoginName, loginUser2.LoginName }));
37 
38             //Assert.AreEqual(true, count >0);
39         }

运行看看

技术分享

 

如期运行,到这里,工作单元就讲完了

 

完整项目架构

我们来看看完整的数据访问层架构

技术分享

整个数据访问层就讲完了,虽然有些简陋,但是基本功能都有了

 

项目架构开发系列

  • 项目架构开发:数据访问层之Cache
  • 项目架构开发:数据访问层之Logger
  • 项目架构开发:数据访问层之Query
  • 项目架构开发:数据访问层之Repository
  • 项目架构开发:数据访问层之UnitOfWork

 

 
分类: 架构设计

UnitOfWork