add support DyRepository.UseTransactionAttribute

dotnetcore · Apr 18, 2019 · d83467b · d83467b
1 parent d840e36
commit d83467b
Show file tree

Hide file tree

Showing 5 changed files with 18 additions and 7 deletions.
diff --git a/src/SmartSql.DyRepository/EmitRepositoryBuilder.cs b/src/SmartSql.DyRepository/EmitRepositoryBuilder.cs
@@ -432,6 +432,7 @@ private static void EmitSetTransaction(ILGenerator ilGen, MethodInfo methodInfo)
             {
                 ilGen.LoadLocalVar(0);
                 ilGen.LoadInt32(useTransactionAttribute.Level.GetHashCode());
+                ilGen.New(NullableType<IsolationLevel>.Ctor);
                 ilGen.Callvirt(RequestContextType.Method.SetTransaction);
             }
         }

diff --git a/src/SmartSql/Reflection/TypeConstants/DataType.cs b/src/SmartSql/Reflection/TypeConstants/DataType.cs
@@ -2,7 +2,6 @@
 using System;
 using System.Collections;
 using System.Data;
-using System.Reflection;
 
 namespace SmartSql.Reflection.TypeConstants
 {

diff --git a/src/SmartSql/Reflection/TypeConstants/NullableType.cs b/src/SmartSql/Reflection/TypeConstants/NullableType.cs
@@ -0,0 +1,16 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Reflection;
+using System.Text;
+
+namespace SmartSql.Reflection.TypeConstants
+{
+    public static class NullableType<TTarget> where TTarget : struct
+    {
+        public static Type TargetType = typeof(TTarget);
+        public static readonly Type Type = typeof(TTarget?);
+
+        public static readonly ConstructorInfo Ctor = Type.GetConstructor(new Type[] { TargetType });
+    }
+}
diff --git a/src/SmartSql/Reflection/TypeConstants/RequestContextType.cs b/src/SmartSql/Reflection/TypeConstants/RequestContextType.cs
@@ -25,7 +25,7 @@ public static class Method
         {
             public static readonly MethodInfo SetDataSourceChoice = AbstractType.GetMethod("set_DataSourceChoice");
             public static readonly MethodInfo SetCommandType = AbstractType.GetMethod("set_CommandType");
-            public static readonly MethodInfo SetTransaction = AbstractType.GetMethod("SetTransaction");
+            public static readonly MethodInfo SetTransaction = AbstractType.GetMethod("set_Transaction");
             public static readonly MethodInfo SetScope = AbstractType.GetMethod("set_Scope");
             public static readonly MethodInfo SetSqlId = AbstractType.GetMethod("set_SqlId");
             public static readonly MethodInfo SetRequest = AbstractType.GetMethod("SetRequest");

diff --git a/src/SmartSql/RequestContext.cs b/src/SmartSql/RequestContext.cs
@@ -54,11 +54,6 @@ public ResultMap GetCurrentResultMap()
                 MultipleResultMap.Results[ExecutionContext.DataReaderWrapper.ResultIndex]?.Map : ResultMap;
         }
 
-        public void SetTransaction(IsolationLevel isolationLevel)
-        {
-            Transaction = isolationLevel;
-        }
-
         public abstract void SetupParameters();
         public abstract void SetRequest(object requestObj);
     }