ConnectionContextExtensions

Author: phongnguyend

README.md

@@ -3,7 +3,7 @@ A very simple .net core library that can help to sync a large number of records
 
 ## Overview
 This library provides extension methods so that you can use with your EntityFrameworkCore **DbContext** instance **DbContextExtensions.cs**
-or you can use **SqlConnectionExtensions.cs** to work directly with a **SqlConnection** instance without using EntityFrameworkCore.
+or you can use **ConnectionContextExtensions.cs** to work directly with a **SqlConnection** instance without using EntityFrameworkCore.
 
 ## Nuget
 | Database | Package | GitHub |
@@ -99,7 +99,7 @@ await new BulkInsertBuilder<Row>(dbct.GetSqlConnection())
 	.ExecuteAsync(rows);
 ```
 
-## SqlConnectionExtensions
+## ConnectionContextExtensions
 ### Using Lambda Expression
 ```c#
 using EntityFrameworkCore.SqlServer.SimpleBulks.BulkDelete;

src/EntityFrameworkCore.SqlServer.SimpleBulks/Extensions/ConnectionContextExtensions.cs

@@ -1,4 +1,5 @@
 using Microsoft.Data.SqlClient;
+using System.Collections.Generic;
 using System.Data;
 using System.Threading;
 using System.Threading.Tasks;
@@ -53,4 +54,52 @@ public static SqlCommand CreateTextCommand(this ConnectionContext connection, st
         command.CommandTimeout = options.Timeout;
         return command;
     }
+
+    public static void SqlBulkCopy(this ConnectionContext connectionContext, DataTable dataTable, string tableName, IReadOnlyDictionary<string, string> columnNameMappings, BulkOptions options = null)
+    {
+        options ??= DefaultBulkOptions;
+
+        using var bulkCopy = new SqlBulkCopy(connectionContext.Connection, SqlBulkCopyOptions.Default, connectionContext.Transaction)
+        {
+            BatchSize = options.BatchSize,
+            BulkCopyTimeout = options.Timeout,
+            DestinationTableName = $"{tableName}"
+        };
+
+        foreach (DataColumn dtColum in dataTable.Columns)
+        {
+            bulkCopy.ColumnMappings.Add(dtColum.ColumnName, GetDbColumnName(dtColum.ColumnName, columnNameMappings));
+        }
+
+        bulkCopy.WriteToServer(dataTable);
+    }
+
+    public static async Task SqlBulkCopyAsync(this ConnectionContext connectionContext, DataTable dataTable, string tableName, IReadOnlyDictionary<string, string> columnNameMappings, BulkOptions options = null, CancellationToken cancellationToken = default)
+    {
+        options ??= DefaultBulkOptions;
+
+        using var bulkCopy = new SqlBulkCopy(connectionContext.Connection, SqlBulkCopyOptions.Default, connectionContext.Transaction)
+        {
+            BatchSize = options.BatchSize,
+            BulkCopyTimeout = options.Timeout,
+            DestinationTableName = $"{tableName}"
+        };
+
+        foreach (DataColumn dtColum in dataTable.Columns)
+        {
+            bulkCopy.ColumnMappings.Add(dtColum.ColumnName, GetDbColumnName(dtColum.ColumnName, columnNameMappings));
+        }
+
+        await bulkCopy.WriteToServerAsync(dataTable, cancellationToken);
+    }
+
+    private static string GetDbColumnName(string columnName, IReadOnlyDictionary<string, string> columnNameMappings)
+    {
+        if (columnNameMappings == null)
+        {
+            return columnName;
+        }
+
+        return columnNameMappings.TryGetValue(columnName, out string value) ? value : columnName;
+    }
 }

src/EntityFrameworkCore.SqlServer.SimpleBulks/Extensions/DataTableExtensions.cs

@@ -1,20 +1,11 @@
-using Microsoft.Data.SqlClient;
-using System.Collections.Generic;
+using System.Collections.Generic;
 using System.Data;
 using System.Text;
-using System.Threading;
-using System.Threading.Tasks;
 
 namespace EntityFrameworkCore.SqlServer.SimpleBulks.Extensions;
 
 public static class DataTableExtensions
 {
-    private static readonly BulkOptions DefaultBulkOptions = new BulkOptions()
-    {
-        BatchSize = 0,
-        Timeout = 30,
-    };
-
     public static string GenerateTableDefinition(this DataTable table, string tableName,
         IReadOnlyDictionary<string, string> columnNameMappings,
         IReadOnlyDictionary<string, string> columnTypeMappings)
@@ -36,44 +27,6 @@ public static string GenerateTableDefinition(this DataTable table, string tableN
         return sql.ToString();
     }
 
-    public static void SqlBulkCopy(this ConnectionContext connectionContext, DataTable dataTable, string tableName, IReadOnlyDictionary<string, string> columnNameMappings, BulkOptions options = null)
-    {
-        options ??= DefaultBulkOptions;
-
-        using var bulkCopy = new SqlBulkCopy(connectionContext.Connection, SqlBulkCopyOptions.Default, connectionContext.Transaction)
-        {
-            BatchSize = options.BatchSize,
-            BulkCopyTimeout = options.Timeout,
-            DestinationTableName = $"{tableName}"
-        };
-
-        foreach (DataColumn dtColum in dataTable.Columns)
-        {
-            bulkCopy.ColumnMappings.Add(dtColum.ColumnName, GetDbColumnName(dtColum.ColumnName, columnNameMappings));
-        }
-
-        bulkCopy.WriteToServer(dataTable);
-    }
-
-    public static async Task SqlBulkCopyAsync(this ConnectionContext connectionContext, DataTable dataTable, string tableName, IReadOnlyDictionary<string, string> columnNameMappings, BulkOptions options = null, CancellationToken cancellationToken = default)
-    {
-        options ??= DefaultBulkOptions;
-
-        using var bulkCopy = new SqlBulkCopy(connectionContext.Connection, SqlBulkCopyOptions.Default, connectionContext.Transaction)
-        {
-            BatchSize = options.BatchSize,
-            BulkCopyTimeout = options.Timeout,
-            DestinationTableName = $"{tableName}"
-        };
-
-        foreach (DataColumn dtColum in dataTable.Columns)
-        {
-            bulkCopy.ColumnMappings.Add(dtColum.ColumnName, GetDbColumnName(dtColum.ColumnName, columnNameMappings));
-        }
-
-        await bulkCopy.WriteToServerAsync(dataTable, cancellationToken);
-    }
-
     private static string GetDbColumnName(string columnName, IReadOnlyDictionary<string, string> columnNameMappings)
     {
         if (columnNameMappings == null)