add TransactionMiddleware

add UseTransactionAttribute

Author: Ahoo-Wang

build/version.props

@@ -2,7 +2,7 @@
 	<PropertyGroup>
 		<VersionMajor>4</VersionMajor>
 		<VersionMinor>0</VersionMinor>
-		<VersionPatch>28</VersionPatch>
+		<VersionPatch>29</VersionPatch>
 		<VersionPrefix>$(VersionMajor).$(VersionMinor).$(VersionPatch)</VersionPrefix>
 	</PropertyGroup>
 </Project>

src/SmartSql.DyRepository/Annotations/UseTransactionAttribute.cs

@@ -0,0 +1,13 @@
+using System;
+using System.Collections.Generic;
+using System.Data;
+using System.Text;
+
+namespace SmartSql.DyRepository.Annotations
+{
+    [AttributeUsage(AttributeTargets.Method, AllowMultiple = false)]
+    public class UseTransactionAttribute : Attribute
+    {
+        public IsolationLevel Level { get; set; } = IsolationLevel.Unspecified;
+    }
+}

src/SmartSql.DyRepository/EmitRepositoryBuilder.cs

@@ -120,7 +120,7 @@ private void BuildMethod(Type interfaceType, TypeBuilder typeBuilder, MethodInfo
             EmitNewRequestContext(ilGen, isOnlyOneClassParam, firstParamType);
             EmitSetCommandType(ilGen, statementAttr);
             EmitSetDataSourceChoice(ilGen, statementAttr);
-
+            EmitSetTransaction(ilGen, methodInfo);
             if (String.IsNullOrEmpty(statementAttr.Sql))
             {
                 EmitSetScope(ilGen, statementAttr.Scope);
@@ -425,6 +425,16 @@ private static void EmitSetDataSourceChoice(ILGenerator ilGen, StatementAttribut
                 ilGen.Callvirt(RequestContextType.Method.SetDataSourceChoice);
             }
         }
+        private static void EmitSetTransaction(ILGenerator ilGen, MethodInfo methodInfo)
+        {
+            var useTransactionAttribute = methodInfo.GetCustomAttribute<UseTransactionAttribute>();
+            if (useTransactionAttribute != null)
+            {
+                ilGen.LoadLocalVar(0);
+                ilGen.LoadInt32(useTransactionAttribute.Level.GetHashCode());
+                ilGen.Callvirt(RequestContextType.Method.SetTransaction);
+            }
+        }
         private void EmitSetRealSql(ILGenerator ilGen, StatementAttribute statementAttr)
         {
             ilGen.LoadLocalVar(0);

src/SmartSql.Test.Unit/DbSessions/DbSessionTest.cs

@@ -2,6 +2,7 @@
 using SmartSql.Reflection;
 using SmartSql.Test.Entities;
 using System;
+using System.Data;
 using System.Threading.Tasks;
 using SmartSql.Data;
 using Xunit;
@@ -98,6 +99,35 @@ public void Insert()
             });
         }
         [Fact]
+        public void InsertByRequestTransaction()
+        {
+            var id = DbSession.ExecuteScalar<long>(new RequestContext
+            {
+                Scope = nameof(AllPrimitive),
+                SqlId = "Insert",
+                Transaction = IsolationLevel.Unspecified,
+                Request = new AllPrimitive
+                {
+                    DateTime = DateTime.Now,
+                    String = "SmartSql",
+                }
+            });
+        }
+        [Fact]
+        public void InsertByStatementTransaction()
+        {
+            var id = DbSession.ExecuteScalar<long>(new RequestContext
+            {
+                Scope = nameof(AllPrimitive),
+                SqlId = "InsertByStatementTransaction",
+                Request = new AllPrimitive
+                {
+                    DateTime = DateTime.Now,
+                    String = "SmartSql",
+                }
+            });
+        }
+        [Fact]
         public void InsertByIdGen()
         {
             var id = DbSession.ExecuteScalar<long>(new RequestContext

src/SmartSql.Test.Unit/DyRepository/RepositoryBuilder_Test.cs

@@ -33,8 +33,19 @@ public void CreateInstance()
                 String = "",
                 DateTime = DateTime.Now
             });
-            
         }
+        [Fact]
+        public void InsertByAnnotationTransaction()
+        {
+            var sqlMapper = SmartSqlBuilder.GetSqlMapper();
+            var repository = _repositoryFactory.CreateInstance(typeof(IAllPrimitiveRepository), sqlMapper) as IAllPrimitiveRepository;
+            var id = repository.InsertByAnnotationTransaction(new Entities.AllPrimitive
+            {
+                String = "",
+                DateTime = DateTime.Now
+            });
+        }
+        
 
         [Fact]
         public void NoMapperRepository_GetGuidFromDb()

src/SmartSql.Test.Unit/Maps/AllPrimitive.xml

@@ -199,6 +199,69 @@
       )
       ;Select Scope_Identity();
     </Statement>
+    <Statement Id="InsertByStatementTransaction" Transaction="Unspecified">
+      INSERT INTO T_AllPrimitive
+      (
+      Boolean,
+      Char,
+      Byte,
+      Int16,
+      Int32,
+      Int64,
+      Single,
+      Decimal,
+      DateTime,
+      String,
+      Guid,
+      TimeSpan,
+      NumericalEnum,
+      NullableBoolean,
+      NullableChar,
+      NullableByte,
+      NullableInt16,
+      NullableInt32,
+      NullableInt64,
+      NullableSingle,
+      NullableDouble,
+      NullableDecimal,
+      NullableDateTime,
+      NullableGuid,
+      NullableTimeSpan,
+      NullableNumericalEnum,
+      NullableString
+      )
+      VALUES
+      (
+      @Boolean,
+      @Char,
+      @Byte,
+      @Int16,
+      @Int32,
+      @Int64,
+      @Single,
+      @Decimal,
+      @DateTime,
+      @String,
+      @Guid,
+      @TimeSpan,
+      @NumericalEnum,
+      @NullableBoolean,
+      @NullableChar,
+      @NullableByte,
+      @NullableInt16,
+      @NullableInt32,
+      @NullableInt64,
+      @NullableSingle,
+      @NullableDouble,
+      @NullableDecimal,
+      @NullableDateTime,
+      @NullableGuid,
+      @NullableTimeSpan,
+      @NullableNumericalEnum,
+      @NullableString
+      )
+      ;Select Scope_Identity();
+    </Statement>
     <Statement Id="InsertByIdGen">
       <IdGenerator Id="Int64"/>
       INSERT INTO T_AllPrimitive

src/SmartSql.Test.Unit/Maps/User.xml

@@ -41,7 +41,7 @@
     </Statement>
 
     <!--删除-->
-    <Statement Id="Delete">
+    <Statement Id="Delete" Transaction="Unspecified">
       Delete From T_User
       Where Id=@Id
     </Statement>

src/SmartSql.Test/Repositories/IAllPrimitiveRepository.cs

@@ -1,5 +1,6 @@
 using System;
 using System.Collections.Generic;
+using System.Data;
 using System.Text;
 using SmartSql.DyRepository.Annotations;
 using SmartSql.Test.Entities;
@@ -13,5 +14,8 @@ public interface IAllPrimitiveRepository
         long Insert(AllPrimitive entity);
 
         (IList<AllPrimitive>, int) GetByPage_ValueTuple(int PageSize = 10, int PageIndex = 1);
+        [UseTransaction]
+        [Statement(Id = "Insert")]
+        long InsertByAnnotationTransaction(AllPrimitive entity);
     }
 }

src/SmartSql/Configuration/Statement.cs

@@ -13,7 +13,6 @@ public class Statement
         public StatementType StatementType { get; set; } = StatementType.Unknown;
         public CommandType? CommandType { get; set; }
         public DataSourceChoice? SourceChoice { get; set; }
-        [Obsolete("弃用")]
         public IsolationLevel? Transaction { get; set; }
         public String ReadDb { get; set; }
         public String FullSqlId => $"{SqlMap.Scope}.{Id}";

src/SmartSql/Middlewares/InitializerMiddleware.cs

@@ -68,6 +68,7 @@ private void InitByStatement(AbstractRequestContext requestContext, SqlMap sqlMa
             {
                 requestContext.CommandType = requestContext.Statement.CommandType.Value;
             }
+            requestContext.Transaction = requestContext.Transaction ?? requestContext.Statement.Transaction;
             requestContext.ReadDb = requestContext.Statement.ReadDb;
             requestContext.CacheId = requestContext.Statement.CacheId;
             requestContext.Cache = requestContext.Statement.Cache;