sharding/src/ShardingCore/DbContexts/ShardingDbContextFactory.cs

using System;
using System.Data.Common;
using System.Linq;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.DependencyInjection;
using ShardingCore.Core.VirtualTables;
using ShardingCore.DbContexts.Abstractions;
using ShardingCore.DbContexts.ShardingDbContexts;
using ShardingCore.DbContexts.VirtualDbContexts;
using ShardingCore.Extensions;
using ShardingCore.Sharding.Abstractions;

namespace ShardingCore.DbContexts
{
/*
* @Author: xjm
* @Description:
* @Date: Thursday, 24 December 2020 08:22:48
* @Email: 326308290@qq.com
*/
    public class ShardingDbContextFactory:IShardingDbContextFactory
    {
        private readonly IShardingCoreOptions _shardingCoreOptions;
        private readonly IShardingTableScopeFactory _shardingTableScopeFactory;
        private readonly IDbContextCreateFilterManager _dbContextCreateFilterManager;
        private readonly IDbContextOptionsProvider _dbContextOptionsProvider;

        public ShardingDbContextFactory(IShardingCoreOptions shardingCoreOptions,IShardingTableScopeFactory shardingTableScopeFactory, IDbContextCreateFilterManager dbContextCreateFilterManager,IDbContextOptionsProvider dbContextOptionsProvider)
        {
            _shardingCoreOptions = shardingCoreOptions;
            _shardingTableScopeFactory = shardingTableScopeFactory;
            _dbContextCreateFilterManager = dbContextCreateFilterManager;
            _dbContextOptionsProvider = dbContextOptionsProvider;
        }
        public DbContext Create(ShardingDbContextOptions shardingDbContextOptions)
        {
            var tail=shardingDbContextOptions.Tail;
            var shardingConfigEntry = _shardingCoreOptions.GetShardingConfig();
            
            var dbContext = shardingConfigEntry.Creator(shardingDbContextOptions);
            if (!string.IsNullOrWhiteSpace(tail) && dbContext is IShardingTableDbContext shardingTableDbContext)
            {
                shardingTableDbContext.SetShardingTableDbContextTail(tail);
            }

            var filters = _dbContextCreateFilterManager.GetFilters();
            if (filters.Any())
            {
                foreach (var dbContextCreateFilter in filters)
                {
                    dbContextCreateFilter.CreateAfter(dbContext);
                }
            }
            var dbContextModel = dbContext.Model;
            return dbContext;
        }

        public DbContext Create(DbConnection dbConnection,string tail)
        {
            var shardingDbContextOptions =
                new ShardingDbContextOptions(_dbContextOptionsProvider.GetDbContextOptions(dbConnection), tail);
           return Create(shardingDbContextOptions);
        }
    }
}