SqlConnection config action

Signed-off-by: Nathan Hart <nhart12@gmail.com>

Author: nhart12

src/Serilog.Sinks.MSSqlServer/Sinks/MSSqlServer/Dependencies/SinkDependenciesFactory.cs

@@ -22,7 +22,7 @@ internal static SinkDependencies Create(
             // unless sink option EnlistInTransaction is set to true.
             var sqlConnectionStringBuilderWrapper = new SqlConnectionStringBuilderWrapper(
                 connectionString, sinkOptions.EnlistInTransaction);
-            var sqlConnectionFactory = new SqlConnectionFactory(sqlConnectionStringBuilderWrapper);
+            var sqlConnectionFactory = new SqlConnectionFactory(sqlConnectionStringBuilderWrapper, sinkOptions.ConnectionConfiguration);
             var sqlCommandFactory = new SqlCommandFactory();
             var dataTableCreator = new DataTableCreator(sinkOptions.TableName, columnOptions);
             var sqlCreateTableWriter = new SqlCreateTableWriter(sinkOptions.SchemaName,
@@ -34,7 +34,7 @@ internal static SinkDependencies Create(
                 InitialCatalog = ""
             };
             var sqlConnectionFactoryNoDb =
-                new SqlConnectionFactory(sqlConnectionStringBuilderWrapperNoDb);
+                new SqlConnectionFactory(sqlConnectionStringBuilderWrapperNoDb, sinkOptions.ConnectionConfiguration);
             var sqlCreateDatabaseWriter = new SqlCreateDatabaseWriter(sqlConnectionStringBuilderWrapper.InitialCatalog);
 
             var logEventDataGenerator =

src/Serilog.Sinks.MSSqlServer/Sinks/MSSqlServer/MSSqlServerSinkOptions.cs

@@ -1,4 +1,5 @@
 using System;
+using Microsoft.Data.SqlClient;
 using Serilog.Core;
 
 namespace Serilog.Sinks.MSSqlServer
@@ -83,5 +84,10 @@ internal MSSqlServerSinkOptions(
         /// Flag to use <see cref="Microsoft.Data.SqlClient.SqlBulkCopy"/> instead of individual INSERT statements (default: true)
         /// </summary>
         public bool UseSqlBulkCopy { get; set; }
+
+        /// <summary>
+        /// Supplies an extension point for customizing the underyling SqlConnection
+        /// </summary>
+        public Action<SqlConnection> ConnectionConfiguration { get; set; }
     }
 }

src/Serilog.Sinks.MSSqlServer/Sinks/MSSqlServer/Platform/SqlClient/SqlConnectionWrapper.cs

@@ -9,9 +9,13 @@ internal class SqlConnectionWrapper : ISqlConnectionWrapper
         private readonly SqlConnection _sqlConnection;
         private bool _disposedValue;
 
-        public SqlConnectionWrapper(string connectionString)
+        public SqlConnectionWrapper(string connectionString, Action<SqlConnection> connectionConfiguration = null)
         {
             _sqlConnection = new SqlConnection(connectionString);
+            if (connectionConfiguration != null)
+            {
+                connectionConfiguration(_sqlConnection);
+            }
         }
 
         public SqlConnection SqlConnection => _sqlConnection;

src/Serilog.Sinks.MSSqlServer/Sinks/MSSqlServer/Platform/SqlConnectionFactory.cs

@@ -1,4 +1,5 @@
 using System;
+using Microsoft.Data.SqlClient;
 using Serilog.Sinks.MSSqlServer.Platform.SqlClient;
 
 namespace Serilog.Sinks.MSSqlServer.Platform
@@ -7,18 +8,21 @@ internal class SqlConnectionFactory : ISqlConnectionFactory
     {
         private readonly string _connectionString;
         private readonly ISqlConnectionStringBuilderWrapper _sqlConnectionStringBuilderWrapper;
+        private readonly Action<SqlConnection> _connectionConfiguration;
 
-        public SqlConnectionFactory(ISqlConnectionStringBuilderWrapper sqlConnectionStringBuilderWrapper)
+        public SqlConnectionFactory(ISqlConnectionStringBuilderWrapper sqlConnectionStringBuilderWrapper,
+            Action<SqlConnection> connectionConfiguration = null)
         {
             _sqlConnectionStringBuilderWrapper = sqlConnectionStringBuilderWrapper
                 ?? throw new ArgumentNullException(nameof(sqlConnectionStringBuilderWrapper));
 
             _connectionString = _sqlConnectionStringBuilderWrapper.ConnectionString;
+            _connectionConfiguration = connectionConfiguration;
         }
 
         public ISqlConnectionWrapper Create()
         {
-            return new SqlConnectionWrapper(_connectionString);
+            return new SqlConnectionWrapper(_connectionString, _connectionConfiguration);
         }
     }
 }