CodeLiturgy.Dashboard/BlueWest.Api/StartupExtensions.cs

using System;
using BlueWest.WebApi.EF;
using Microsoft.AspNetCore.Hosting;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Hosting;
using Microsoft.Extensions.Logging;
using Microsoft.Extensions.Configuration;

namespace BlueWest.WebApi
{
    /// <summary>
    /// Startup Extensions
    /// </summary>
    public static class StartupExtensions
    {
        private static MySqlServerVersion GetMySqlServerVersion(int major, int minor, int build) => new (new Version(major, minor, build));
        
        /// <summary>
        /// Get MYSQL Connection String
        /// </summary>
        /// <param name="optionsBuilder"></param>
        /// <param name="configuration"></param>
        /// <param name="environment"></param>
        private static DbContextOptionsBuilder GetMySqlSettings(
            this DbContextOptionsBuilder optionsBuilder,
            IConfiguration configuration,
            IWebHostEnvironment environment)
        {
            var sqlVersion = GetMySqlServerVersion(8, 0, 11);

            optionsBuilder
                .UseMySql(
                    configuration.GetConnectionString("DockerMySQL"), 
                    sqlVersion)
                    .UseMySql(sqlVersion,
                        builder =>
                        {
                            builder.EnableRetryOnFailure(6, TimeSpan.FromSeconds(3), null);
                        });

            // The following three options help with debugging, but should
            // be changed or removed for production.
            if (environment.IsDevelopment())
            {
                optionsBuilder
                    .LogTo(Console.WriteLine, LogLevel.Information)
                    .EnableSensitiveDataLogging()
                    .EnableDetailedErrors();
            }

            return optionsBuilder;
        }

        /// <summary>
        /// Setup database Contexts
        /// </summary>
        /// <param name="serviceCollection"></param>
        /// <param name="configuration"></param>
        /// <param name="environment"></param>
        /// <returns></returns>
        public static IServiceCollection PrepareMySqlDatabasePool(this IServiceCollection serviceCollection,
            IConfiguration configuration, IWebHostEnvironment environment)
        {
            return serviceCollection
                .AddDbContextPool<UserDbContext>(options => options.GetMySqlSettings(configuration, environment))
                .AddDbContextPool<CountryDbContext>(options => options.GetMySqlSettings(configuration, environment))
                .AddDbContextPool<FinanceDbContext>(options => options.GetMySqlSettings(configuration, environment))
                .AddDbContextPool<CompanyDbContext>(options => options.GetMySqlSettings(configuration, environment));
        }
        
        /// <summary>
        /// Setup database Contexts
        /// </summary>
        /// <param name="serviceCollection"></param>
        /// <param name="configuration"></param>
        /// <param name="environment"></param>
        /// <returns></returns>
        public static IServiceCollection PrepareSqlLiteDatabasePool(this IServiceCollection serviceCollection,
            IConfiguration configuration, IWebHostEnvironment environment)
        {
            var sqliteConString = "Data Source=BlueWest.Api.db";
            
            return serviceCollection
                .AddDbContextPool<UserDbContext>(options => options.UseSqlite(sqliteConString))
                .AddDbContextPool<CountryDbContext>(options => options.UseSqlite(sqliteConString))
                .AddDbContextPool<FinanceDbContext>(options => options.UseSqlite(sqliteConString))
                .AddDbContextPool<CompanyDbContext>(options => options.UseSqlite(sqliteConString));
        }
        
    }
}