添加dbcontextbuildercreator的接口依赖支持替换,支持添加executordbcontext的applicationserivceprovider的注入

Author: xuejmnet

samples/Sample.AutoCreateIfPresent/DefaultDbContext.cs

@@ -1,5 +1,7 @@
 using System;
 using Microsoft.EntityFrameworkCore;
+using Microsoft.EntityFrameworkCore.Infrastructure;
+using ShardingCore.Core.RuntimeContexts;
 using ShardingCore.Core.VirtualRoutes.TableRoutes.RouteTails.Abstractions;
 using ShardingCore.Sharding;
 using ShardingCore.Sharding.Abstractions;
@@ -24,6 +26,7 @@ protected override void OnModelCreating(ModelBuilder modelBuilder)
             base.OnModelCreating(modelBuilder);
             modelBuilder.ApplyConfiguration(new OrderByHourMap());
             modelBuilder.ApplyConfiguration(new AreaDeviceMap());
+            Console.WriteLine(this.IsExecutor);
         }
 
         public IRouteTail RouteTail { get; set; }

src/ShardingCore/Core/DbContextOptionBuilderCreator/ActivatorDbContextOptionBuilderCreator.cs

@@ -0,0 +1,31 @@
+using System;
+using Microsoft.EntityFrameworkCore;
+using ShardingCore.Core.DbContextTypeAwares;
+using ShardingCore.Core.ServiceProviders;
+
+namespace ShardingCore.Core.DbContextOptionBuilderCreator
+{
+    public class ActivatorDbContextOptionBuilderCreator:IDbContextOptionBuilderCreator
+    {
+        private readonly IShardingProvider _shardingProvider;
+        private readonly IDbContextTypeAware _dbContextTypeAware;
+
+        public ActivatorDbContextOptionBuilderCreator(IShardingProvider shardingProvider,IDbContextTypeAware dbContextTypeAware)
+        {
+            _shardingProvider = shardingProvider;
+            _dbContextTypeAware = dbContextTypeAware;
+        }
+        public DbContextOptionsBuilder CreateDbContextOptionBuilder()
+        {
+            var dbContextType = _dbContextTypeAware.GetContextType();
+            Type type = typeof(DbContextOptionsBuilder<>);
+            type = type.MakeGenericType(dbContextType);
+            var dbContextOptionsBuilder = (DbContextOptionsBuilder)Activator.CreateInstance(type);
+            if (_shardingProvider.ApplicationServiceProvider != null)
+            {
+                dbContextOptionsBuilder.UseApplicationServiceProvider(_shardingProvider.ApplicationServiceProvider);
+            }
+            return dbContextOptionsBuilder;
+        }
+    }
+}

src/ShardingCore/Core/DbContextOptionBuilderCreator/IDbContextOptionBuilderCreator.cs

@@ -0,0 +1,9 @@
+using Microsoft.EntityFrameworkCore;
+
+namespace ShardingCore.Core.DbContextOptionBuilderCreator
+{
+    public interface IDbContextOptionBuilderCreator
+    {
+        DbContextOptionsBuilder CreateDbContextOptionBuilder();
+    }
+}

src/ShardingCore/Core/RuntimeContexts/IShardingRuntimeContext.cs

@@ -2,6 +2,7 @@
 using Microsoft.EntityFrameworkCore;
 using Microsoft.Extensions.Logging;
 using ShardingCore.Core.DbContextCreator;
+using ShardingCore.Core.DbContextOptionBuilderCreator;
 using ShardingCore.Core.DbContextTypeAwares;
 using ShardingCore.Core.EntityMetadatas;
 using ShardingCore.Core.QueryRouteManagers.Abstractions;
@@ -31,6 +32,7 @@ public interface IShardingRuntimeContext
         Type DbContextType { get; }
         IDbContextTypeAware GetDbContextTypeAware();
         IShardingProvider GetShardingProvider();
+        IDbContextOptionBuilderCreator GetDbContextOptionBuilderCreator();
         ShardingConfigOptions GetShardingConfigOptions();
         IShardingRouteConfigOptions GetShardingRouteConfigOptions();
         IShardingMigrationManager GetShardingMigrationManager();

src/ShardingCore/Core/RuntimeContexts/ShardingRuntimeContext.cs

@@ -4,6 +4,7 @@
 using Microsoft.Extensions.Logging;
 using ShardingCore.Bootstrappers;
 using ShardingCore.Core.DbContextCreator;
+using ShardingCore.Core.DbContextOptionBuilderCreator;
 using ShardingCore.Core.DbContextTypeAwares;
 using ShardingCore.Core.EntityMetadatas;
 using ShardingCore.Core.QueryRouteManagers.Abstractions;
@@ -87,6 +88,12 @@ public IShardingProvider GetShardingProvider()
            return _shardingProvider??=GetRequiredService<IShardingProvider>();
         }
 
+        private IDbContextOptionBuilderCreator _dbContextOptionBuilderCreator;
+        public IDbContextOptionBuilderCreator GetDbContextOptionBuilderCreator()
+        {
+            return _dbContextOptionBuilderCreator??=GetRequiredService<IDbContextOptionBuilderCreator>();
+        }
+
         private ShardingConfigOptions _shardingConfigOptions;
         public ShardingConfigOptions GetShardingConfigOptions()
         {
@@ -323,6 +330,7 @@ private void InitFieldValue()
         {
             GetDbContextTypeAware();
             GetShardingProvider();
+            GetDbContextOptionBuilderCreator();
             GetShardingConfigOptions();
             GetShardingRouteConfigOptions();
             GetShardingMigrationManager();

src/ShardingCore/EFCores/AbstractScriptMigrationGenerator.cs

@@ -80,7 +80,6 @@ private string ExecuteMigrateUnits(IShardingRuntimeContext shardingRuntimeContex
                         shardingMigrationManager.Current.CurrentDataSourceName = migrateUnit.DataSourceName;
 
                         var dbContextOptions = DynamicShardingHelper.CreateShellDbContextOptions(shardingRuntimeContext,
-                            migrateUnit.ShellDbContext.GetType(),
                             migrateUnit.DataSourceName);
 
                         using (var dbContext = dbContextCreator.CreateDbContext(migrateUnit.ShellDbContext,

src/ShardingCore/Helpers/DynamicShardingHelper.cs

@@ -101,7 +101,7 @@ public static async Task DynamicMigrateWithDataSourcesAsync(IShardingRuntimeCont
                     {
                         shardingMigrationManager.Current.CurrentDataSourceName = migrateUnit.DataSourceName;
 
-                        var dbContextOptions = CreateShellDbContextOptions(shardingRuntimeContext,migrateUnit.ShellDbContext.GetType(),
+                        var dbContextOptions = CreateShellDbContextOptions(shardingRuntimeContext,
                             migrateUnit.DataSourceName);
 
                         using (var dbContext = dbContextCreator.CreateDbContext(migrateUnit.ShellDbContext,
@@ -123,11 +123,11 @@ public static async Task DynamicMigrateWithDataSourcesAsync(IShardingRuntimeCont
             await TaskHelper.WhenAllFastFail(migrateTasks).ConfigureAwait(false);
         }
 
-        public static DbContextOptions CreateShellDbContextOptions(IShardingRuntimeContext shardingRuntimeContext,Type dbContextType,string dataSourceName)
+        public static DbContextOptions CreateShellDbContextOptions(IShardingRuntimeContext shardingRuntimeContext,string dataSourceName)
         {
             var virtualDataSource = shardingRuntimeContext.GetVirtualDataSource();
             var shardingConfigOptions = shardingRuntimeContext.GetShardingConfigOptions();
-            var dbContextOptionBuilder = DataSourceDbContext.CreateDbContextOptionBuilder(dbContextType);
+            var dbContextOptionBuilder =shardingRuntimeContext.GetDbContextOptionBuilderCreator().CreateDbContextOptionBuilder();
             var connectionString = virtualDataSource.GetConnectionString(dataSourceName);
             virtualDataSource.UseDbContextOptionsBuilder(connectionString, dbContextOptionBuilder);
             shardingConfigOptions.ShardingMigrationConfigure?.Invoke(dbContextOptionBuilder);

src/ShardingCore/Sharding/ShardingDbContextExecutors/DataSourceDbContext.cs

@@ -154,8 +154,9 @@ private DbContextOptions CreateShareDbContextOptionsBuilder()
             try
             {
                 //先创建dbcontext option builder
-                var dbContextOptionsBuilder = CreateDbContextOptionBuilder(DbContextType).UseShardingOptions(_shardingRuntimeContext);
-
+                var dbContextOptionBuilderCreator = _shardingRuntimeContext.GetDbContextOptionBuilderCreator();
+                var dbContextOptionsBuilder = dbContextOptionBuilderCreator.CreateDbContextOptionBuilder().UseShardingOptions(_shardingRuntimeContext);
+            
                 if (IsDefault)
                 {
                     //如果是默认的需要使用shell的dbconnection为了保证可以使用事务
@@ -191,13 +192,6 @@ private DbContextOptions CreateShareDbContextOptionsBuilder()
             }
         }
 
-        public static DbContextOptionsBuilder CreateDbContextOptionBuilder(Type dbContextType)
-        {
-            Type type = typeof(DbContextOptionsBuilder<>);
-            type = type.MakeGenericType(dbContextType);
-            return (DbContextOptionsBuilder)Activator.CreateInstance(type);
-        }
-
         /// <summary>
         /// 不支持并发后期发现直接报错而不是用lock
         /// </summary>

src/ShardingCore/Sharding/ShardingDbContextExecutors/ShardingDbContextExecutor.cs

@@ -114,7 +114,7 @@ public DbContext CreateDbContext(CreateDbContextStrategyEnum strategy, string da
 
         private DbContextOptions CreateParallelDbContextOptions(string dataSourceName,CreateDbContextStrategyEnum strategy)
         {
-            var dbContextOptionBuilder = DataSourceDbContext.CreateDbContextOptionBuilder(_shardingDbContext.GetType());
+            var dbContextOptionBuilder =_shardingRuntimeContext.GetDbContextOptionBuilderCreator().CreateDbContextOptionBuilder();
             var connectionString = _actualConnectionStringManager.GetConnectionString(dataSourceName, CreateDbContextStrategyEnum.IndependentConnectionWrite==strategy);
             _virtualDataSource.UseDbContextOptionsBuilder(connectionString, dbContextOptionBuilder).UseShardingOptions(_shardingRuntimeContext);
             return dbContextOptionBuilder.Options;

src/ShardingCore/ShardingCoreExtension.cs

@@ -31,6 +31,7 @@
 using Microsoft.EntityFrameworkCore.Migrations;
 using ShardingCore.Bootstrappers;
 using ShardingCore.Core.DbContextCreator;
+using ShardingCore.Core.DbContextOptionBuilderCreator;
 using ShardingCore.Core.DbContextTypeAwares;
 using ShardingCore.Core.QueryTrackers;
 using ShardingCore.Core.RuntimeContexts;
@@ -144,6 +145,7 @@ internal static IServiceCollection AddInternalShardingCore<TShardingDbContext>(t
                 .TryAddSingleton<IVirtualDataSourceConfigurationParams, SimpleVirtualDataSourceConfigurationParams>();
             //分表dbcontext创建
             services.TryAddSingleton<IDbContextCreator, ActivatorDbContextCreator<TShardingDbContext>>();
+            services.TryAddSingleton<IDbContextOptionBuilderCreator, ActivatorDbContextOptionBuilderCreator>();
 
 
             // services.TryAddSingleton<IDataSourceInitializer<TShardingDbContext>, DataSourceInitializer<TShardingDbContext>>();