利用AOP实现SqlSugar自动事务

2019-12-30 18:26:11丽君

_cache保存SqlSugar实例,_threadLocal确保同一线程下取出的是同一个SqlSugar实例。

不知道SqlSugar判断当前实例是否已经开启事务,所以又将SqlSugar包了一层。


 public class SqlClient
   {
     public SqlSugarClient SqlSugarClient;
     public bool IsBeginTran = false;
     public int TranCount = 0;

     public SqlClient(SqlSugarClient sqlSugarClient)
     {
       this.SqlSugarClient = sqlSugarClient;
     }
   }

IsBeginTran标识当前SqlSugar实例是否已经开启事务,TranCount是一个避免事务嵌套的计数器。

一开始的例子


 [TransactionCallHandler]
      public IList<Student> GetStudentList(Hashtable paramsHash)
      {
        var list = mStudentDa.GetStudents(paramsHash);
        var value = mValueService.FindAll();
        return list;
      }

TransactionCallHandler表明该方法要开启事务,但是如果mValueService.FindAll也标识了TransactionCallHandler,又要开启一次事务?所以用TranCount做一个计数。

使用Castle.DynamicProxy

要实现标识了TransactionCallHandler的方法实现自动事务,使用Castle.DynamicProxy实现BL类的代理

Castle.DynamicProxy一般操作

 


 public class MyClass : IMyClass
  {
    public void MyMethod()
    {
      Console.WriteLine("My Mehod");
    }
 }
 public class TestIntercept : IInterceptor
   {
     public void Intercept(IInvocation invocation)
     {
       Console.WriteLine("before");
       invocation.Proceed();
       Console.WriteLine("after");
     }
   }

  var proxyGenerate = new ProxyGenerator();
  TestIntercept t=new TestIntercept();
  var pg = proxyGenerate.CreateClassProxy<MyClass>(t);
  pg.MyMethod();
  //输出是
  //before
  //My Mehod
  //after

before就是要开启事务的地方,after就是提交事务的地方

最后实现


 public class TransactionInterceptor : IInterceptor
   {
     private readonly ILogger logger;
     public TransactionInterceptor()
     {
       logger = LogManager.GetCurrentClassLogger();
     }
     public void Intercept(IInvocation invocation)
     {
       MethodInfo methodInfo = invocation.MethodInvocationTarget;
       if (methodInfo == null)
       {
         methodInfo = invocation.Method;
       }

       TransactionCallHandlerAttribute transaction =
         methodInfo.GetCustomAttributes<TransactionCallHandlerAttribute>(true).FirstOrDefault();
       if (transaction != null)
       {
         SugarManager.BeginTran();
         try
         {
           SugarManager.TranCountAddOne();
           invocation.Proceed();
           SugarManager.TranCountMunisOne();
           SugarManager.CommitTran();
         }
         catch (Exception e)
         {
           SugarManager.RollbackTran();
           logger.Error(e);
           throw e;
         }

       }
       else
       {
         invocation.Proceed();
       }
     }
   }
   [AttributeUsage(AttributeTargets.Method, Inherited = true)]
   public class TransactionCallHandlerAttribute : Attribute
   {
     public TransactionCallHandlerAttribute()
     {

     }
   }