Added extension points for creating ProfiledDbCommand and ProfiledDbD…

…ataReader
MiniProfiler · Sep 14, 2018 · 0fb7080 · 0fb7080
1 parent cc91adf
commit 0fb7080
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/src/MiniProfiler.Shared/Data/ProfiledDbCommand.cs b/src/MiniProfiler.Shared/Data/ProfiledDbCommand.cs
@@ -196,6 +196,9 @@ public override UpdateRowSource UpdatedRowSource
             set => _command.UpdatedRowSource = value;
         }
 
+        protected virtual DbDataReader CreateDataReader(DbDataReader original, IDbProfiler profiler)
+            => new ProfiledDbDataReader(original, profiler);
+
         /// <summary>
         /// Executes a database data reader.
         /// </summary>
@@ -213,7 +216,7 @@ protected override DbDataReader ExecuteDbDataReader(CommandBehavior behavior)
             try
             {
                 result = _command.ExecuteReader(behavior);
-                result = new ProfiledDbDataReader(result, _profiler);
+                result = CreateDataReader(result, _profiler);
             }
             catch (Exception e)
             {
@@ -246,7 +249,7 @@ protected override async Task<DbDataReader> ExecuteDbDataReaderAsync(CommandBeha
             try
             {
                 result = await _command.ExecuteReaderAsync(behavior, cancellationToken).ConfigureAwait(false);
-                result = new ProfiledDbDataReader(result, _profiler);
+                result = CreateDataReader(result, _profiler);
             }
             catch (Exception e)
             {

diff --git a/src/MiniProfiler.Shared/Data/ProfiledDbConnection.cs b/src/MiniProfiler.Shared/Data/ProfiledDbConnection.cs
@@ -137,12 +137,15 @@ protected override DbTransaction BeginDbTransaction(IsolationLevel isolationLeve
         {
             return new ProfiledDbTransaction(_connection.BeginTransaction(isolationLevel), this);
         }
+
+        protected virtual DbCommand CreateDbCommand(DbCommand original, IDbProfiler profiler)
+                => new ProfiledDbCommand(original, this, profiler);
 
         /// <summary>
         /// Creates and returns a <see cref="DbCommand"/> object associated with the current connection.
         /// </summary>
         /// <returns>A <see cref="ProfiledDbCommand"/> wrapping the created <see cref="DbCommand"/>.</returns>
-        protected override DbCommand CreateDbCommand() => new ProfiledDbCommand(_connection.CreateCommand(), this, _profiler);
+        protected override DbCommand CreateDbCommand() => CreateDbCommand(_connection.CreateCommand(), _profiler);
 
         /// <summary>
         /// Dispose the underlying connection.