mh_frame_sps/APT.Infrastructure.EF/Extensions/SqlServerTenantExtension.cs

using System;
using APT.Infrastructure.Api;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.DependencyInjection;

namespace APT.Infrastructure.EF.Extensions
{
    public static class SqlServerTenantExtension
    {
        public static IServiceCollection AddSqlServerPerConnection<TDbContext>(this IServiceCollection services,
            string key = "default",
            string connectionPrefix = "tenanted",
            Action<DbContextOptionsBuilder> optionAction = null,
            Action<IServiceProvider, string, DbContextOptionsBuilder> dbContextSetup = null)
            where TDbContext : DbContext, ITenantDbContext
        {
            return services.AddDbPerConnection<TDbContext>(DbIntegrationType.SqlServer, key, connectionPrefix, 
                optionAction, dbContextSetup ?? SetUpSqlServer<TDbContext>);
        }

        public static IServiceCollection AddSqlServerPerConnection<TDbContext>(this IServiceCollection services,
            Action<TenantSettings<TDbContext>> setupAction = null)
            where TDbContext : DbContext, ITenantDbContext
        {
            return services.AddDbPerConnection<TDbContext>(CombineSettings(setupAction));
        }

        public static IServiceCollection AddSqlServerPerTable<TDbContext>(this IServiceCollection services,
            string key = "default",
            string connectionName = "tenanted",
            Action<DbContextOptionsBuilder> optionAction = null,
            Action<IServiceProvider, string, DbContextOptionsBuilder> dbContextSetup = null)
            where TDbContext : DbContext, ITenantDbContext
        {
            return services.AddDbPerTable<TDbContext>(DbIntegrationType.SqlServer, key, connectionName,
                optionAction, dbContextSetup ?? SetUpSqlServer<TDbContext>);
        }

        public static IServiceCollection AddSqlServerPerTable<TDbContext>(this IServiceCollection services,
            Action<TenantSettings<TDbContext>> setupAction = null)
            where TDbContext : DbContext, ITenantDbContext
        {
            return services.AddDbPerTable<TDbContext>(CombineSettings(setupAction));
        }

        public static IServiceCollection AddSqlServerPerSchema<TDbContext>(this IServiceCollection services,
            string key = "default",
            string connectionName = "tenanted",
            Action<DbContextOptionsBuilder> optionAction = null,
            Action<IServiceProvider, string, DbContextOptionsBuilder> dbContextSetup = null)
            where TDbContext : DbContext, ITenantDbContext
        {
            return services.AddDbPerSchema<TDbContext>(DbIntegrationType.SqlServer, key, connectionName,
                optionAction, dbContextSetup ?? SetUpSqlServer<TDbContext>);
        }

        public static IServiceCollection AddSqlServerPerSchema<TDbContext>(this IServiceCollection services,
            Action<TenantSettings<TDbContext>> setupAction = null)
            where TDbContext : DbContext, ITenantDbContext
        {
            return services.AddDbPerSchema<TDbContext>(CombineSettings(setupAction));
        }

        static Action<TenantSettings<TDbContext>> CombineSettings<TDbContext>(
            Action<TenantSettings<TDbContext>> setupAction = null)
            where TDbContext : DbContext, ITenantDbContext
        {
            return (settings) =>
            {
                settings.DbContextSetup = SetUpSqlServer<TDbContext>;
                setupAction?.Invoke(settings);
            };
        }

        internal static void SetUpSqlServer<TDbContext>(IServiceProvider serviceProvider, string connectionString,
            DbContextOptionsBuilder optionsBuilder)
            where TDbContext : DbContext, ITenantDbContext
        {
            var settings = serviceProvider.GetService<TenantSettings<TDbContext>>();
            var tenant = serviceProvider.GetService<TenantInfo>();
            optionsBuilder.UseSqlServer(connectionString, builder =>
            {
                builder.TenantBuilderSetup(serviceProvider, settings, tenant);
            });
        }

    }
}