关于使用Transaction对于非数据库事务的操作

时间:2022-07-01 09:10:50

在操作数据库的过程中,为了数据的一致性,我们可以使用Transaction,要么成功的时候全部提交,要么有任何一个操作失败立即全部回滚。不仅仅是在数据库方面,有时候操作其他的内容,比如说对于系统文件的操作,也需要把一些操作组合看做是一个事务。

现在我们看这样一个例子。现在我们需要在计算机的硬盘上创建3个目录A,B,C,要求要么3个全部创建成功,要么一个也不要创建。我们可以把这个看成是一个事务。如果我们自己写代码来操作,可以这样写。

  1. bool createA = false;
  2. bool createB = false;
  3. bool createC = false;
  4. try
  5. {
  6. //这里的操作是创建3个目录
  7. Directory.CreateDirectory("\\A");
  8. createA = true;
  9. Directory.CreateDirectory("\\B");
  10. createB = true;
  11. Directory.CreateDirectory("\\C");
  12. createC = true;
  13. }
  14. catch (System.Exception ex)
  15. {
  16. //这里在捕捉到异常时,根据运行的结果进行回滚
  17. if (createB)
  18. {
  19. Directory.Delete("\\B");
  20. Directory.Delete("\\A");
  21. }
  22. if (!createB && createA)
  23. {
  24. Directory.Delete("\\A");
  25. }
  26. }

但是这里我们是把这3个操作当成一个整体来回滚,及时是简单的创建删除文件夹,我们可以看到catch中的回滚逻辑已经很复杂了。可以想象,如果A,B,C这3步不仅仅是创建目录,还有一些其他的操作,那么回滚的逻辑就非常复杂,此时,我们可以考虑把这一个分成几个小事务,分开回滚。代码可以这样写,这里我们使用了Framework提供的Transactin以及TransactioScope类。

  1. class Program
  2. {
  3. static void Main(string[] args)
  4. {
  5. //这里就不需要了try catch,因为scope就已经完成了这个功能,
  6. //即当有异常发生向外抛的时候,会尝试跳出这个using代码块,
  7. //CLR会在向代码块外边跑异常之前,分别取调用每个transaction的
  8. //roolback方法,只要rollback中你定义的逻辑没有问题,那么所有
  9. //的已发生的操作就会安全的回滚。
  10. using (var scope = new TransactionScope())
  11. {
  12. var A = new OperationA();
  13. Transaction.Current.EnlistVolatile(A, EnlistmentOptions.None);
  14. A.DoWork();
  15. OperationB B = new OperationB();
  16. Transaction.Current.EnlistVolatile(B, EnlistmentOptions.None);
  17. B.DoWork();
  18. scope.Complete();
  19. }
  20. }
  21. }
  22. }
  23. class OperationA : IEnlistmentNotification
  24. {
  25. private bool _isCommitSucceed = false;
  26. public void Commit(Enlistment enlistment)
  27. {
  28. enlistment.Done();
  29. }
  30. public void DoWork()//这是自定义的方法,不是继承IEnlistmentNotification
  31. {
  32. Directory.CreateDirectory("\\A");
  33. //这里还有一些其他的关于A的复杂操作
  34. _isCommitSucceed = true;
  35. }
  36. public void InDoubt(Enlistment enlistment)
  37. {
  38. enlistment.Done();
  39. }
  40. public void Prepare(PreparingEnlistment preparingEnlistment)
  41. {
  42. preparingEnlistment.Prepared();
  43. }
  44. public void Rollback(Enlistment enlistment)
  45. {
  46. //这里回滚A的一些操作,当然这里的操作逻辑需要你自己来写。
  47. //比如说查看是否创建成功,或者有一些其他的信号标记,通过这些
  48. //标记你来决定是删除目录还是其他的什么回滚操作。
  49. if (_isCommitSucceed)
  50. Directory.Delete("\\A");
  51. enlistment.Done();
  52. }
  53. }
  54. class OperationB : IEnlistmentNotification
  55. {
  56. private bool _isCommitSucceed = false;
  57. public void Commit(Enlistment enlistment)
  58. {
  59. enlistment.Done();
  60. }
  61. public void DoWork()
  62. {
  63. //这里是关于B的一些复杂操作
  64. throw new Exception("test");
  65. //这里依然有一些操作代码,但是我们模拟的是B操作途中抛出异常,所以这里的代码不会执行
  66. }
  67. public void InDoubt(Enlistment enlistment)
  68. {
  69. enlistment.Done();
  70. }
  71. public void Prepare(PreparingEnlistment preparingEnlistment)
  72. {
  73. preparingEnlistment.Prepared();
  74. }
  75. public void Rollback(Enlistment enlistment)
  76. {
  77. if (_isCommitSucceed)
  78. {
  79. //这里回滚B的一些已经存在的操作。
  80. }
  81. enlistment.Done();
  82. }

这里的机制是这样的,可能是scope.Complete()方法中有某种特别的操作,去告诉CLR这次所有的操作都顺利完成了,在跳出这个scope的时候就不用调用那些transaction的rollback方法了。如果没有执行到scope.Complete()方法,那么就会在跳出这个scope代码块的时候调用rollback方法。一般造成这个的原因是在某个transaction的逻辑操作中出现异常,造成从此次直接抛出异常跳出这个代码块,没有机会执行下边的代码。可以看出这里的重点也是写rollback的逻辑,但是相对于原来的catch中的逻辑,这里分开为多个小的逻辑,相对来说容易了很多。

这只是个人写的一种使用回滚的逻辑。在使用IEnlistmentNotification的时候,也有人把业务逻辑写入到Commit中。如果想真正理解transaction回滚的机制,建议深入理解一下TransactionScope与Transactiond的实现机制。