IShardingRuntimeContext支持dbcontexttype区分

Author: xuejmnet

samples/Sample.MySql/Startup.cs

@@ -2,6 +2,7 @@
 using Microsoft.EntityFrameworkCore;
 using Microsoft.EntityFrameworkCore.Migrations;
 using Sample.MySql.DbContexts;
+using Sample.MySql.Domain.Entities;
 using Sample.MySql.Shardings;
 using ShardingCore;
 using ShardingCore.Bootstrappers;
@@ -153,6 +154,7 @@ public void Configure(IApplicationBuilder app, IWebHostEnvironment env)
             app.ApplicationServices.UseAutoShardingCreate();
             var shardingRuntimeContext = app.ApplicationServices.GetRequiredService<IShardingRuntimeContext>();
             var entityMetadataManager = shardingRuntimeContext.GetEntityMetadataManager();
+            var entityMetadata = entityMetadataManager.TryGet<SysUserMod>();
             using (var scope = app.ApplicationServices.CreateScope())
             {
                 var defaultShardingDbContext = scope.ServiceProvider.GetService<DefaultShardingDbContext>();

src/ShardingCore/Core/RuntimeContexts/IShardingRuntimeContext.cs

@@ -27,6 +27,7 @@ namespace ShardingCore.Core.RuntimeContexts
 
     public interface IShardingRuntimeContext
     {
+        Type DbContextType { get; }
         IShardingProvider GetShardingProvider();
         ShardingConfigOptions GetShardingConfigOptions();
         IShardingRouteConfigOptions GetShardingRouteConfigOptions();

src/ShardingCore/Core/RuntimeContexts/ShardingRuntimeContext.cs

@@ -39,7 +39,12 @@ public sealed class ShardingRuntimeContext : IShardingRuntimeContext
         private IServiceCollection _serviceMap = new ServiceCollection();
 
         private IServiceProvider _serviceProvider;
+        public Type DbContextType { get; }
         // private ILoggerFactory _applicationLoggerFactory;
+        public ShardingRuntimeContext(Type dbContextType)
+        {
+            DbContextType = dbContextType;
+        }
 
         public void AddServiceConfig(Action<IServiceCollection> configure)
         {
@@ -69,6 +74,7 @@ public void AutoShardingCreate()
         }
 
         private IShardingProvider _shardingProvider;
+
         public IShardingProvider GetShardingProvider()
         {
            return _shardingProvider??=GetRequiredService<IShardingProvider>();

src/ShardingCore/ShardingRuntimeBuilder.cs

@@ -73,7 +73,7 @@ public IShardingRuntimeContext Build(IServiceProvider appServiceProvider)
 
         public IShardingRuntimeContext Build(IServiceProvider appServiceProvider, ILoggerFactory loggerFactory)
         {
-            var shardingRuntimeContext = new ShardingRuntimeContext();
+            var shardingRuntimeContext = new ShardingRuntimeContext(typeof(TShardingDbContext));
             shardingRuntimeContext.AddServiceConfig(services =>
             {
                 // services.AddSingleton<IDbContextTypeCollector>(sp => new DbContextTypeCollector<TShardingDbContext>());