laatzen/LaaProductionWeb/LaaProductionWeb.Data/DbContextExtensions.cs

namespace LaaProductionWeb.Data
{
    using LaaProductionWeb.Data.Interfaces;

    using Microsoft.Extensions.DependencyInjection;

    using System;
    using System.Data.SqlClient;
    using System.Diagnostics;

    public static class DbContextExtensions
    {
        public static IServiceCollection AddDbContext(this IServiceCollection services, string connectionString)
        {
            services.AddTransient<IDbContext>(_ => new DbContext(connectionString));

            return services;
        }

        internal static void OpenWithErrorHandling(this SqlConnection sqlConnection, SqlInfoMessageEventHandler messageHandler = null)
        {
            sqlConnection.FireInfoMessageEventOnUserErrors = true;
            sqlConnection.InfoMessage += SqlConnectionInfoMessage;

            if (messageHandler != null)
            {
                sqlConnection.InfoMessage += messageHandler;
            }

            sqlConnection.Open();
        }

        internal static void SqlConnectionInfoMessage(object sender, SqlInfoMessageEventArgs e)
        {
            Debug.WriteLine(e);

            // TODO: Handle errors
        }
    }
}