Merge pull request #805 from medkg15/future-dbcommand-interception

Add extension points before and after a Future query is run.

src/shared/Z.EF.Plus.QueryFuture.Shared/QueryFutureBatch.cs

@@ -183,6 +183,7 @@ public void ExecuteQueries()
 
                     using (command)
                     {
+                        QueryFutureManager.OnBatchExecuting?.Invoke(command);
 #if EF5
                     using (var reader = command.ExecuteReader())
                     {
@@ -213,6 +214,7 @@ public void ExecuteQueries()
                             }
                         }
 #endif
+                        QueryFutureManager.OnBatchExecuted?.Invoke(command);
                     }
                 }
                 finally

src/shared/Z.EF.Plus.QueryFuture.Shared/QueryFutureManager.cs

@@ -7,6 +7,8 @@
 
 using System.Runtime.CompilerServices;
 using Z.EntityFramework.Extensions;
+using System;
+using System.Data.Common;
 
 #if NET45 || EFCORE
 
@@ -53,6 +55,20 @@ static QueryFutureManager()
         /// <value>True if allow query batch, false if not.</value>
         public static bool AllowQueryBatch { get; set; } = true;
 
+        /// <summary>Gets or sets a delegate to be invoked directly before executing the batch DbCommand.</summary>
+        /// <remarks>
+        /// This delegate is only invoked when queries are actually executed as a batch containing multiple queries.  
+        /// i.e. When AllowQueryBatch=false or only a single query is pending, this delegate is not invoked.
+        /// </remarks>
+        public static Action<DbCommand> OnBatchExecuting { get; set; } = null;
+
+        /// <summary>Gets or sets a delegate to be invoked directly after executing the batch DbCommand.</summary>
+        /// <remarks>
+        /// This delegate is only invoked when queries are actually executed as a batch containing multiple queries.  
+        /// i.e. When AllowQueryBatch=false or only a single query is pending, this delegate is not invoked.
+        /// </remarks>
+        public static Action<DbCommand> OnBatchExecuted { get; set; } = null;
+
         /// <summary>Gets or sets the weak table used to cache future batch associated to a context.</summary>
         /// <value>The weak table used to cache future batch associated to a context.</value>
 #if EF5 || EF6