#343 CancellationToken Support for SQL Server BulkDelete

mikependon · Oct 2, 2020 · 821e017 · 821e017
1 parent eda8e5b
commit 821e017
Show file tree

Hide file tree

Showing 6 changed files with 300 additions and 162 deletions.
diff --git a/...ons/RepoDb.SqlServer.BulkOperations/Microsoft.Data.SqlClient/BaseRepository/BulkDelete.cs b/...ons/RepoDb.SqlServer.BulkOperations/Microsoft.Data.SqlClient/BaseRepository/BulkDelete.cs
@@ -2,6 +2,7 @@
 using System;
 using System.Collections.Generic;
 using System.Linq.Expressions;
+using System.Threading;
 using System.Threading.Tasks;
 
 namespace RepoDb
@@ -126,20 +127,23 @@ public static int BulkDelete<TEntity>(this BaseRepository<TEntity, SqlConnection
         /// <param name="batchSize">The size per batch to be used.</param>
         /// <param name="usePhysicalPseudoTempTable">The flags that signify whether to create a physical pseudo table.</param>
         /// <param name="transaction">The transaction to be used.</param>
+        /// <param name="cancellationToken">The <see cref="CancellationToken"/> object to be used during the asynchronous operation.</param>
         /// <returns>The number of rows affected by the execution.</returns>
         public static Task<int> BulkDeleteAsync<TEntity>(this BaseRepository<TEntity, SqlConnection> repository,
             IEnumerable<object> primaryKeys,
             string hints = null,
             int? batchSize = null,
             bool? usePhysicalPseudoTempTable = null,
-            SqlTransaction transaction = null)
+            SqlTransaction transaction = null,
+            CancellationToken cancellationToken = default)
             where TEntity : class
         {
             return repository.DbRepository.BulkDeleteAsync<TEntity>(primaryKeys: primaryKeys,
                 hints: hints,
                 batchSize: batchSize,
                 usePhysicalPseudoTempTable: usePhysicalPseudoTempTable,
-                transaction: transaction);
+                transaction: transaction,
+                cancellationToken: cancellationToken);
         }
 
         /// <summary>
@@ -155,6 +159,7 @@ public static Task<int> BulkDeleteAsync<TEntity>(this BaseRepository<TEntity, Sq
         /// <param name="batchSize">The size per batch to be used.</param>
         /// <param name="usePhysicalPseudoTempTable">The flags that signify whether to create a physical pseudo table.</param>
         /// <param name="transaction">The transaction to be used.</param>
+        /// <param name="cancellationToken">The <see cref="CancellationToken"/> object to be used during the asynchronous operation.</param>
         /// <returns>The number of rows affected by the execution.</returns>
         public static Task<int> BulkDeleteAsync<TEntity>(this BaseRepository<TEntity, SqlConnection> repository,
             IEnumerable<TEntity> entities,
@@ -164,7 +169,8 @@ public static Task<int> BulkDeleteAsync<TEntity>(this BaseRepository<TEntity, Sq
             string hints = null,
             int? batchSize = null,
             bool? usePhysicalPseudoTempTable = null,
-            SqlTransaction transaction = null)
+            SqlTransaction transaction = null,
+            CancellationToken cancellationToken = default)
             where TEntity : class
         {
             return repository.DbRepository.BulkDeleteAsync<TEntity>(entities: entities,
@@ -174,7 +180,8 @@ public static Task<int> BulkDeleteAsync<TEntity>(this BaseRepository<TEntity, Sq
                 hints: hints,
                 batchSize: batchSize,
                 usePhysicalPseudoTempTable: usePhysicalPseudoTempTable,
-                transaction: transaction);
+                transaction: transaction,
+                cancellationToken: cancellationToken);
         }
 
         /// <summary>
@@ -191,6 +198,7 @@ public static Task<int> BulkDeleteAsync<TEntity>(this BaseRepository<TEntity, Sq
         /// <param name="batchSize">The size per batch to be used.</param>
         /// <param name="usePhysicalPseudoTempTable">The flags that signify whether to create a physical pseudo table.</param>
         /// <param name="transaction">The transaction to be used.</param>
+        /// <param name="cancellationToken">The <see cref="CancellationToken"/> object to be used during the asynchronous operation.</param>
         /// <returns>The number of rows affected by the execution.</returns>
         public static Task<int> BulkDeleteAsync<TEntity>(this BaseRepository<TEntity, SqlConnection> repository,
             string tableName,
@@ -201,7 +209,8 @@ public static Task<int> BulkDeleteAsync<TEntity>(this BaseRepository<TEntity, Sq
             string hints = null,
             int? batchSize = null,
             bool? usePhysicalPseudoTempTable = null,
-            SqlTransaction transaction = null)
+            SqlTransaction transaction = null,
+            CancellationToken cancellationToken = default)
             where TEntity : class
         {
             return repository.DbRepository.BulkDeleteAsync<TEntity>(tableName: tableName,
@@ -212,7 +221,8 @@ public static Task<int> BulkDeleteAsync<TEntity>(this BaseRepository<TEntity, Sq
                 hints: hints,
                 batchSize: batchSize,
                 usePhysicalPseudoTempTable: usePhysicalPseudoTempTable,
-                transaction: transaction);
+                transaction: transaction,
+                cancellationToken: cancellationToken);
         }
 
         #endregion

diff --git a/...tions/RepoDb.SqlServer.BulkOperations/Microsoft.Data.SqlClient/DbRepository/BulkDelete.cs b/...tions/RepoDb.SqlServer.BulkOperations/Microsoft.Data.SqlClient/DbRepository/BulkDelete.cs
@@ -5,6 +5,7 @@
 using System.Data;
 using System.Data.Common;
 using System.Linq.Expressions;
+using System.Threading;
 using System.Threading.Tasks;
 
 namespace RepoDb
@@ -500,13 +501,15 @@ public static int BulkDelete(this DbRepository<SqlConnection> repository,
         /// <param name="batchSize">The size per batch to be used.</param>
         /// <param name="usePhysicalPseudoTempTable">The flags that signify whether to create a physical pseudo table.</param>
         /// <param name="transaction">The transaction to be used.</param>
+        /// <param name="cancellationToken">The <see cref="CancellationToken"/> object to be used during the asynchronous operation.</param>
         /// <returns>The number of rows affected by the execution.</returns>
         public static async Task<int> BulkDeleteAsync<TEntity>(this DbRepository<SqlConnection> repository,
             IEnumerable<object> primaryKeys,
             string hints = null,
             int? batchSize = null,
             bool? usePhysicalPseudoTempTable = null,
-            SqlTransaction transaction = null)
+            SqlTransaction transaction = null,
+            CancellationToken cancellationToken = default)
             where TEntity : class
         {
             // Create a connection
@@ -520,7 +523,8 @@ public static async Task<int> BulkDeleteAsync<TEntity>(this DbRepository<SqlConn
                     bulkCopyTimeout: repository.CommandTimeout,
                     batchSize: batchSize,
                     usePhysicalPseudoTempTable: usePhysicalPseudoTempTable,
-                    transaction: transaction);
+                    transaction: transaction,
+                    cancellationToken: cancellationToken);
             }
             catch
             {
@@ -553,6 +557,7 @@ public static async Task<int> BulkDeleteAsync<TEntity>(this DbRepository<SqlConn
         /// <param name="batchSize">The size per batch to be used.</param>
         /// <param name="usePhysicalPseudoTempTable">The flags that signify whether to create a physical pseudo table.</param>
         /// <param name="transaction">The transaction to be used.</param>
+        /// <param name="cancellationToken">The <see cref="CancellationToken"/> object to be used during the asynchronous operation.</param>
         /// <returns>The number of rows affected by the execution.</returns>
         public static async Task<int> BulkDeleteAsync<TEntity>(this DbRepository<SqlConnection> repository,
             IEnumerable<TEntity> entities,
@@ -562,7 +567,8 @@ public static async Task<int> BulkDeleteAsync<TEntity>(this DbRepository<SqlConn
             string hints = null,
             int? batchSize = null,
             bool? usePhysicalPseudoTempTable = null,
-            SqlTransaction transaction = null)
+            SqlTransaction transaction = null,
+            CancellationToken cancellationToken = default)
             where TEntity : class
         {
             // Create a connection
@@ -579,7 +585,8 @@ public static async Task<int> BulkDeleteAsync<TEntity>(this DbRepository<SqlConn
                     bulkCopyTimeout: repository.CommandTimeout,
                     batchSize: batchSize,
                     usePhysicalPseudoTempTable: usePhysicalPseudoTempTable,
-                    transaction: transaction);
+                    transaction: transaction,
+                    cancellationToken: cancellationToken);
             }
             catch
             {
@@ -613,6 +620,7 @@ public static async Task<int> BulkDeleteAsync<TEntity>(this DbRepository<SqlConn
         /// <param name="batchSize">The size per batch to be used.</param>
         /// <param name="usePhysicalPseudoTempTable">The flags that signify whether to create a physical pseudo table.</param>
         /// <param name="transaction">The transaction to be used.</param>
+        /// <param name="cancellationToken">The <see cref="CancellationToken"/> object to be used during the asynchronous operation.</param>
         /// <returns>The number of rows affected by the execution.</returns>
         public static async Task<int> BulkDeleteAsync<TEntity>(this DbRepository<SqlConnection> repository,
             string tableName,
@@ -623,7 +631,8 @@ public static async Task<int> BulkDeleteAsync<TEntity>(this DbRepository<SqlConn
             string hints = null,
             int? batchSize = null,
             bool? usePhysicalPseudoTempTable = null,
-            SqlTransaction transaction = null)
+            SqlTransaction transaction = null,
+            CancellationToken cancellationToken = default)
             where TEntity : class
         {
             // Create a connection
@@ -641,7 +650,8 @@ public static async Task<int> BulkDeleteAsync<TEntity>(this DbRepository<SqlConn
                     bulkCopyTimeout: repository.CommandTimeout,
                     batchSize: batchSize,
                     usePhysicalPseudoTempTable: usePhysicalPseudoTempTable,
-                    transaction: transaction);
+                    transaction: transaction,
+                    cancellationToken: cancellationToken);
             }
             catch
             {
@@ -674,6 +684,7 @@ public static async Task<int> BulkDeleteAsync<TEntity>(this DbRepository<SqlConn
         /// <param name="batchSize">The size per batch to be used.</param>
         /// <param name="usePhysicalPseudoTempTable">The flags that signify whether to create a physical pseudo table.</param>
         /// <param name="transaction">The transaction to be used.</param>
+        /// <param name="cancellationToken">The <see cref="CancellationToken"/> object to be used during the asynchronous operation.</param>
         /// <returns>The number of rows affected by the execution.</returns>
         public static async Task<int> BulkDeleteAsync<TEntity>(this DbRepository<SqlConnection> repository,
             DbDataReader reader,
@@ -683,7 +694,8 @@ public static async Task<int> BulkDeleteAsync<TEntity>(this DbRepository<SqlConn
             string hints = null,
             int? batchSize = null,
             bool? usePhysicalPseudoTempTable = null,
-            SqlTransaction transaction = null)
+            SqlTransaction transaction = null,
+            CancellationToken cancellationToken = default)
             where TEntity : class
         {
             // Create a connection
@@ -700,7 +712,8 @@ public static async Task<int> BulkDeleteAsync<TEntity>(this DbRepository<SqlConn
                     bulkCopyTimeout: repository.CommandTimeout,
                     batchSize: batchSize,
                     usePhysicalPseudoTempTable: usePhysicalPseudoTempTable,
-                    transaction: transaction);
+                    transaction: transaction,
+                    cancellationToken: cancellationToken);
             }
             catch
             {
@@ -734,6 +747,7 @@ public static async Task<int> BulkDeleteAsync<TEntity>(this DbRepository<SqlConn
         /// <param name="batchSize">The size per batch to be used.</param>
         /// <param name="usePhysicalPseudoTempTable">The flags that signify whether to create a physical pseudo table.</param>
         /// <param name="transaction">The transaction to be used.</param>
+        /// <param name="cancellationToken">The <see cref="CancellationToken"/> object to be used during the asynchronous operation.</param>
         /// <returns>The number of rows affected by the execution.</returns>
         public static async Task<int> BulkDeleteAsync<TEntity>(this DbRepository<SqlConnection> repository,
             DataTable dataTable,
@@ -744,7 +758,8 @@ public static async Task<int> BulkDeleteAsync<TEntity>(this DbRepository<SqlConn
             string hints = null,
             int? batchSize = null,
             bool? usePhysicalPseudoTempTable = null,
-            SqlTransaction transaction = null)
+            SqlTransaction transaction = null,
+            CancellationToken cancellationToken = default)
             where TEntity : class
         {
             // Create a connection
@@ -762,7 +777,8 @@ public static async Task<int> BulkDeleteAsync<TEntity>(this DbRepository<SqlConn
                     bulkCopyTimeout: repository.CommandTimeout,
                     batchSize: batchSize,
                     usePhysicalPseudoTempTable: usePhysicalPseudoTempTable,
-                    transaction: transaction);
+                    transaction: transaction,
+                    cancellationToken: cancellationToken);
             }
             catch
             {
@@ -796,14 +812,16 @@ public static async Task<int> BulkDeleteAsync<TEntity>(this DbRepository<SqlConn
         /// <param name="batchSize">The size per batch to be used.</param>
         /// <param name="usePhysicalPseudoTempTable">The flags that signify whether to create a physical pseudo table.</param>
         /// <param name="transaction">The transaction to be used.</param>
+        /// <param name="cancellationToken">The <see cref="CancellationToken"/> object to be used during the asynchronous operation.</param>
         /// <returns>The number of rows affected by the execution.</returns>
         public static async Task<int> BulkDeleteAsync(this DbRepository<SqlConnection> repository,
             string tableName,
             IEnumerable<object> primaryKeys,
             string hints = null,
             int? batchSize = null,
             bool? usePhysicalPseudoTempTable = null,
-            SqlTransaction transaction = null)
+            SqlTransaction transaction = null,
+            CancellationToken cancellationToken = default)
         {
             // Create a connection
             var connection = (transaction?.Connection ?? repository.CreateConnection());
@@ -817,7 +835,8 @@ public static async Task<int> BulkDeleteAsync(this DbRepository<SqlConnection> r
                     bulkCopyTimeout: repository.CommandTimeout,
                     batchSize: batchSize,
                     usePhysicalPseudoTempTable: usePhysicalPseudoTempTable,
-                    transaction: transaction);
+                    transaction: transaction,
+                    cancellationToken: cancellationToken);
             }
             catch
             {
@@ -850,6 +869,7 @@ public static async Task<int> BulkDeleteAsync(this DbRepository<SqlConnection> r
         /// <param name="batchSize">The size per batch to be used.</param>
         /// <param name="usePhysicalPseudoTempTable">The flags that signify whether to create a physical pseudo table.</param>
         /// <param name="transaction">The transaction to be used.</param>
+        /// <param name="cancellationToken">The <see cref="CancellationToken"/> object to be used during the asynchronous operation.</param>
         /// <returns>The number of rows affected by the execution.</returns>
         public static async Task<int> BulkDeleteAsync(this DbRepository<SqlConnection> repository,
             string tableName,
@@ -860,7 +880,8 @@ public static async Task<int> BulkDeleteAsync(this DbRepository<SqlConnection> r
             string hints = null,
             int? batchSize = null,
             bool? usePhysicalPseudoTempTable = null,
-            SqlTransaction transaction = null)
+            SqlTransaction transaction = null,
+            CancellationToken cancellationToken = default)
         {
             // Create a connection
             var connection = (transaction?.Connection ?? repository.CreateConnection());
@@ -877,7 +898,8 @@ public static async Task<int> BulkDeleteAsync(this DbRepository<SqlConnection> r
                     bulkCopyTimeout: repository.CommandTimeout,
                     batchSize: batchSize,
                     usePhysicalPseudoTempTable: usePhysicalPseudoTempTable,
-                    transaction: transaction);
+                    transaction: transaction,
+                    cancellationToken: cancellationToken);
             }
             catch
             {
@@ -911,6 +933,7 @@ public static async Task<int> BulkDeleteAsync(this DbRepository<SqlConnection> r
         /// <param name="batchSize">The size per batch to be used.</param>
         /// <param name="usePhysicalPseudoTempTable">The flags that signify whether to create a physical pseudo table.</param>
         /// <param name="transaction">The transaction to be used.</param>
+        /// <param name="cancellationToken">The <see cref="CancellationToken"/> object to be used during the asynchronous operation.</param>
         /// <returns>The number of rows affected by the execution.</returns>
         public static async Task<int> BulkDeleteAsync(this DbRepository<SqlConnection> repository,
             string tableName,
@@ -922,7 +945,8 @@ public static async Task<int> BulkDeleteAsync(this DbRepository<SqlConnection> r
             string hints = null,
             int? batchSize = null,
             bool? usePhysicalPseudoTempTable = null,
-            SqlTransaction transaction = null)
+            SqlTransaction transaction = null,
+            CancellationToken cancellationToken = default)
         {
             // Create a connection
             var connection = (transaction?.Connection ?? repository.CreateConnection());
@@ -940,7 +964,8 @@ public static async Task<int> BulkDeleteAsync(this DbRepository<SqlConnection> r
                     bulkCopyTimeout: repository.CommandTimeout,
                     batchSize: batchSize,
                     usePhysicalPseudoTempTable: usePhysicalPseudoTempTable,
-                    transaction: transaction);
+                    transaction: transaction,
+                    cancellationToken: cancellationToken);
             }
             catch
             {