Added public IHasDbTransaction interface similar to IHasDbConnection,…

… which allows "unwrapping" the real transaction in client code without Reflection hacks.
angelcolmenares · May 26, 2016 · bbdf480 · bbdf480
1 parent 1bfc3ff
commit bbdf480
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/src/ServiceStack.Common/Data/IHasDbConnection.cs b/src/ServiceStack.Common/Data/IHasDbConnection.cs
@@ -15,5 +15,10 @@ public interface IHasDbCommand
     {
         IDbCommand DbCommand { get; }
     }
+
+    public interface IHasDbTransaction
+    {
+        IDbTransaction DbTransaction { get; }
+    }
 }
 #endif
diff --git a/src/ServiceStack.Common/MiniProfiler/Data/ProfiledCommand.cs b/src/ServiceStack.Common/MiniProfiler/Data/ProfiledCommand.cs
@@ -76,7 +76,7 @@ protected override DbTransaction DbTransaction
             {
                 this._tran = value;
                 var awesomeTran = value as ProfiledDbTransaction;
-                _cmd.Transaction = awesomeTran == null ? value : awesomeTran.WrappedTransaction;
+                _cmd.Transaction = awesomeTran?.DbTransaction as DbTransaction ?? value;
             }
         }
 

diff --git a/src/ServiceStack.Common/MiniProfiler/Data/ProfiledDbTransaction.cs b/src/ServiceStack.Common/MiniProfiler/Data/ProfiledDbTransaction.cs
@@ -1,12 +1,13 @@
 using System;
 using System.Data.Common;
 using System.Data;
+using ServiceStack.Data;
 
 #pragma warning disable 1591 // xml doc comments warnings
 
 namespace ServiceStack.MiniProfiler.Data
 {
-    public class ProfiledDbTransaction : DbTransaction
+    public class ProfiledDbTransaction : DbTransaction, IHasDbTransaction
     {
         private ProfiledConnection _conn;
         private DbTransaction _trans;
@@ -24,7 +25,7 @@ protected override DbConnection DbConnection
             get { return _conn; }
         }
 
-        internal DbTransaction WrappedTransaction
+        public IDbTransaction DbTransaction
         {
             get { return _trans; }
         }