添加dbcontext类型发现接口用来判断如果存在分表路由那么必须是分表数据库上下文[#182]，将多个接口合并到IShardingDbContext接口上,并且默认的ShellDbContext接口支持track的直接调用

Author: xuejmnet

samples/Sample.MySql/Startup.cs

@@ -131,7 +131,7 @@ public void ConfigureServices(IServiceCollection services)
                     {
                         b.ReplaceService<IMigrationsSqlGenerator, ShardingMySqlMigrationsSqlGenerator>();
                     });
-                }).ReplaceService<ITableEnsureManager, MySqlTableEnsureManager>()
+                })
                 .AddShardingCore();
             // services.AddDbContext<DefaultShardingDbContext>(ShardingCoreExtension
             //     .UseMutliDefaultSharding<DefaultShardingDbContext>);

samples/Samples.AbpSharding/AbstractShardingAbpDbContext.cs

@@ -22,7 +22,7 @@
 namespace Samples.AbpSharding
 {
 
-    public abstract class AbstractShardingAbpDbContext<TDbContext> : AbpDbContext<TDbContext>, IShardingDbContext, ISupportShardingReadWrite
+    public abstract class AbstractShardingAbpDbContext<TDbContext> : AbpDbContext<TDbContext>, IShardingDbContext
                                 where TDbContext : DbContext
     {
         private readonly IShardingDbContextExecutor _shardingDbContextExecutor;
@@ -571,5 +571,9 @@ public IVirtualDataSource GetVirtualDataSource()
             return _shardingDbContextExecutor.GetVirtualDataSource();
         }
 
+        public IDictionary<string, IDataSourceDbContext> GetCurrentDbContexts()
+        {
+            return _shardingDbContextExecutor.GetCurrentDbContexts();
+        }
     }
 }

samples/Samples.AbpSharding/AbstractShardingAbpZeroDbContext.cs

@@ -28,9 +28,7 @@ namespace Samples.AbpSharding
 {
     public abstract class AbstractShardingAbpZeroDbContext<TTenant, TRole, TUser, TSelf>
         : AbpZeroDbContext<TTenant, TRole, TUser, TSelf>,
-        IShardingDbContext,
-        ISupportShardingReadWrite,
-        IShardingTableDbContext
+        IShardingDbContext
         where TTenant : AbpTenant<TUser>
         where TRole : AbpRole<TUser>
         where TUser : AbpUser<TUser>
@@ -650,5 +648,9 @@ protected virtual void FillDbContextInject(DbContext dbContext)
 
         #endregion
 
+        public IDictionary<string, IDataSourceDbContext> GetCurrentDbContexts()
+        {
+            return _shardingDbContextExecutor.GetCurrentDbContexts();
+        }
     }
 }

src/ShardingCore/Bootstrappers/ShardingInitializer.cs

@@ -2,14 +2,15 @@
 using System.Threading.Tasks;
 using Microsoft.Extensions.Logging;
 using ShardingCore.Core.DbContextCreator;
+using ShardingCore.Core.DbContextTypeAwares;
 using ShardingCore.Core.EntityMetadatas;
 using ShardingCore.Core.ServiceProviders;
 using ShardingCore.Core.ShardingConfigurations.Abstractions;
 using ShardingCore.Exceptions;
 using ShardingCore.Extensions;
 using ShardingCore.Jobs;
 using ShardingCore.Jobs.Abstaractions;
-
+using ShardingCore.Sharding.Abstractions;
 using ShardingCore.Sharding.MergeEngines.ParallelControl;
 using ShardingCore.Sharding.ParallelTables;
 
@@ -20,18 +21,25 @@ internal class ShardingInitializer:IShardingInitializer
     {
         private readonly ILogger<ShardingBootstrapper> _logger;
         private readonly IShardingProvider _shardingProvider;
+        private readonly IDbContextTypeAware _dbContextTypeAware;
         private readonly IShardingRouteConfigOptions _routeConfigOptions;
         private readonly IEntityMetadataManager _entityMetadataManager;
         private readonly IParallelTableManager _parallelTableManager;
         private readonly DoOnlyOnce _doOnlyOnce = new DoOnlyOnce();
 
-        public ShardingInitializer(IShardingProvider shardingProvider,ILogger<ShardingBootstrapper> logger)
+        public ShardingInitializer(IShardingProvider shardingProvider,
+            IDbContextTypeAware dbContextTypeAware,
+            IShardingRouteConfigOptions shardingRouteConfigOptions,
+            IEntityMetadataManager entityMetadataManager,
+            IParallelTableManager parallelTableManager,
+            ILogger<ShardingBootstrapper> logger)
         {
             _logger = logger;
             _shardingProvider = shardingProvider;
-            _routeConfigOptions = shardingProvider.GetRequiredService<IShardingRouteConfigOptions>();
-            _entityMetadataManager = shardingProvider.GetRequiredService<IEntityMetadataManager>();
-            _parallelTableManager = shardingProvider.GetRequiredService<IParallelTableManager>();
+            _dbContextTypeAware = dbContextTypeAware;
+            _routeConfigOptions = shardingRouteConfigOptions;
+            _entityMetadataManager = entityMetadataManager;
+            _parallelTableManager = parallelTableManager;
         }
         /// <summary>
         /// 初始化
@@ -63,6 +71,18 @@ private void InitializeEntityMetadata()
                 var entityMetadataInitializer =(IEntityMetadataInitializer)_shardingProvider.CreateInstance(entityMetadataInitializerType);
                 entityMetadataInitializer.Initialize();
             }
+
+            var allShardingEntities = _entityMetadataManager.GetAllShardingEntities();
+            //判断如果有分表的对象那么dbcontext必须继承IShardingTableDbContext
+            if (allShardingEntities.Any(o => _entityMetadataManager.IsShardingTable(o)))
+            {
+                var contextType = _dbContextTypeAware.GetContextType();
+                if (!contextType.IsShardingTableDbContext())
+                {
+                    throw new ShardingCoreInvalidOperationException(
+                        $"db context {contextType},has sharding table route,should be implement {nameof(IShardingTableDbContext)}");
+                }
+            }
         }
 
         private void InitializeParallelTables()

src/ShardingCore/Core/DbContextTypeAwares/DbContextTypeAware.cs

@@ -0,0 +1,17 @@
+using System;
+
+namespace ShardingCore.Core.DbContextTypeAwares;
+
+public class DbContextTypeAware:IDbContextTypeAware
+{
+    private readonly Type _dbContextType;
+
+    public DbContextTypeAware(Type dbContextType)
+    {
+        _dbContextType = dbContextType;
+    }
+    public Type GetContextType()
+    {
+        return _dbContextType;
+    }
+}

src/ShardingCore/Core/DbContextTypeAwares/IDbContextTypeAware.cs

@@ -0,0 +1,8 @@
+using System;
+
+namespace ShardingCore.Core.DbContextTypeAwares;
+
+public interface IDbContextTypeAware
+{
+    Type GetContextType();
+}

src/ShardingCore/Core/RuntimeContexts/IShardingRuntimeContext.cs

@@ -2,6 +2,7 @@
 using Microsoft.EntityFrameworkCore;
 using Microsoft.Extensions.Logging;
 using ShardingCore.Core.DbContextCreator;
+using ShardingCore.Core.DbContextTypeAwares;
 using ShardingCore.Core.EntityMetadatas;
 using ShardingCore.Core.QueryRouteManagers.Abstractions;
 using ShardingCore.Core.QueryTrackers;
@@ -28,6 +29,7 @@ namespace ShardingCore.Core.RuntimeContexts
     public interface IShardingRuntimeContext
     {
         Type DbContextType { get; }
+        IDbContextTypeAware GetDbContextTypeAware();
         IShardingProvider GetShardingProvider();
         ShardingConfigOptions GetShardingConfigOptions();
         IShardingRouteConfigOptions GetShardingRouteConfigOptions();

src/ShardingCore/Core/RuntimeContexts/IShardingRuntimeContextManager.cs

@@ -4,6 +4,7 @@
 using Microsoft.Extensions.Logging;
 using ShardingCore.Bootstrappers;
 using ShardingCore.Core.DbContextCreator;
+using ShardingCore.Core.DbContextTypeAwares;
 using ShardingCore.Core.EntityMetadatas;
 using ShardingCore.Core.QueryRouteManagers.Abstractions;
 using ShardingCore.Core.QueryTrackers;
@@ -72,9 +73,15 @@ public void AutoShardingCreate()
         {
             GetRequiredService<IShardingBootstrapper>().AutoShardingCreate();
         }
+        private IDbContextTypeAware _dbContextTypeAware;
+        public IDbContextTypeAware GetDbContextTypeAware()
+        {
+            return _dbContextTypeAware??=GetRequiredService<IDbContextTypeAware>();
+        }
 
         private IShardingProvider _shardingProvider;
 
+
         public IShardingProvider GetShardingProvider()
         {
            return _shardingProvider??=GetRequiredService<IShardingProvider>();
@@ -314,6 +321,7 @@ public IShardingRouteManager GetShardingRouteManager()
 
         private void InitFieldValue()
         {
+            GetDbContextTypeAware();
             GetShardingProvider();
             GetShardingConfigOptions();
             GetShardingRouteConfigOptions();