在Saas系统下多租户零脚本分表分库读写分离解决方案

打印 上一主题 下一主题

主题 899|帖子 899|积分 2697

在Saas系统下多租户零脚本分表分库读写分离解决方案

## 介绍本文ShardinfCore版本x.6.0.20+本期主角:- [`ShardingCore`](https://github.com/dotnetcore/sharding-core) 一款ef-core下高性能、轻量级针对分表分库读写分离的解决方案,具有零依赖、零学习成本、零业务代码入侵适配单dbcontext多数据库自动迁移

之前发过一篇文章 EFCore高级Saas系统下单DbContext如何支持不同数据库的迁移 这篇文章让efcore可以支持在单dbcontext下支持多个数据库的迁移来实现多租户下的不同数据库实现
前言

您是否有以下场景:

  • 多租户系统,数据库级别隔离
  • 大数据量,需要分表分库(动态添加),分库分表全自动维护处理
  • 租户之间可能需要使用不同的数据库模式,譬如有些租户要求用oracle,或者mssql,或者mysql或者pgsql
  • 多租户系统在不同的数据库环境下需要维护的表结构复杂繁琐,需要维护许多脚本
  • 业务代码需要进行大范围的妥协来适应上述支持
  • 系统需要支持读写分离(动态添加)
  • 无需停机状态实时添加租户(租户线上签约)
当然我是一开始想先写这篇文章,但是写着写着发现有些时候这个问题就来了,譬如多数据库下efcore默认不支持迁移,经过不断地努力,大脑的思维宫殿我下意识就发现了解决方案,最终用一天时间解决了就是前面的一篇文章 EFCore高级Saas系统下单DbContext如何支持不同数据库的迁移 那么我们话不多说马上开始
接下来我们将实现A,B,C三个租户,其中A租户我们使用MSSQL的订单表使用按月分表,B租户我们使用MYSQL的订单表我们采用Id取模分表,C租户我们使用MSSQL也是使用订单按月分表但是起始时间和A不一样
管理租户数据

首先我们新建一个DbContext用来管理我们的租户信息
租户用户表

首先我们新建一张租户登录的用户表,每个用户就是我们对外的租户
  1.     public class SysUser
  2.     {
  3.         public string Id { get; set; }
  4.         public string Name { get; set; }
  5.         public string Password { get; set; }
  6.         public DateTime CreationTime { get; set; }
  7.         public bool IsDeleted { get; set; }
  8.     }
复制代码
租户配置表

然后我们新建一张租户的配置信息表用来后续初始化配置
  1.     public class SysUserTenantConfig
  2.     {
  3.         public string Id { get; set; }
  4.         public string UserId { get; set; }
  5.         /// <summary>
  6.         /// 添加ShardingTenantOptions的Json包
  7.         /// </summary>
  8.         public string ConfigJson { get; set; }
  9.         public DateTime CreationTime { get; set; }
  10.         public bool IsDeleted { get; set; }
  11.     }
复制代码
定义租户配置
  1. //为了满足上述需求我们需要对数据库和订单分片方式进行区分
  2.     public class ShardingTenantOptions
  3.     {
  4.         /// <summary>
  5.         /// 默认数据源名称
  6.         /// </summary>
  7.         public  string DefaultDataSourceName { get; set;}
  8.         /// <summary>
  9.         /// 默认数据库地址
  10.         /// </summary>
  11.         public  string DefaultConnectionString { get; set; }
  12.         /// <summary>
  13.         /// 数据库类型
  14.         /// </summary>
  15.         public DbTypeEnum DbType { get; set; }
  16.         /// <summary>
  17.         /// 分片模式 取模还是按月
  18.         /// </summary>
  19.         public OrderShardingTypeEnum OrderShardingType { get; set; }
  20.         /// <summary>
  21.         /// 按月分片其实时间
  22.         /// </summary>
  23.         public DateTime BeginTimeForSharding { get; set; }
  24.         /// <summary>
  25.         /// 分片迁移的命名空间
  26.         /// </summary>
  27.         public string MigrationNamespace { get; set; }
  28.     }
  29.     public enum DbTypeEnum
  30.     {
  31.         MSSQL = 1,
  32.         MYSQL = 2
  33.     }
  34.   public enum OrderShardingTypeEnum
  35.   {
  36.      Mod=1,
  37.      ByMonth=2
  38.     }
复制代码
租户持久化DbContext

新建一个dbcontext用来存储我们的租户信息,当然你也可以使用文件或者redis之类的都行
  1.     public class IdentityDbContext:DbContext
  2.     {
  3.         public IdentityDbContext(DbContextOptions<IdentityDbContext> options):base(options)
  4.         {
  5.             
  6.         }
  7.         protected override void OnModelCreating(ModelBuilder modelBuilder)
  8.         {
  9.             base.OnModelCreating(modelBuilder);
  10.             modelBuilder.ApplyConfiguration(new SysUserMap());
  11.             modelBuilder.ApplyConfiguration(new SysUserTenantConfigMap());
  12.         }
  13.     }
复制代码
这样我们就完成了租户信息的存储
租户管理者

  我们拥有了租户信息持久化的数据后需要对租户信息的使用进行配置
首先我们新建一个接口可以用来管理租户信息
  1.     public interface ITenantManager
  2.     {
  3.         /// <summary>
  4.         /// 获取所有的租户
  5.         /// </summary>
  6.         /// <returns></returns>
  7.         List<string> GetAll();
  8.         /// <summary>
  9.         /// 获取当前租户
  10.         /// </summary>
  11.         /// <returns></returns>
  12.         TenantContext GetCurrentTenantContext();
  13.         /// <summary>
  14.         /// 添加租户信息
  15.         /// </summary>
  16.         /// <param name="tenantId"></param>
  17.         /// <param name="shardingRuntimeContext"></param>
  18.         /// <returns></returns>
  19.         bool AddTenantSharding(string tenantId, IShardingRuntimeContext shardingRuntimeContext);
  20.         /// <summary>
  21.         /// 创建租户环境
  22.         /// </summary>
  23.         /// <param name="tenantId"></param>
  24.         /// <returns></returns>
  25.         TenantScope CreateScope(string tenantId);
  26.     }
  27.     //租户的默认管理实现
  28.     public class DefaultTenantManager:ITenantManager
  29.     {
  30.         private readonly ITenantContextAccessor _tenantContextAccessor;
  31.         private readonly ConcurrentDictionary<string, IShardingRuntimeContext> _cache = new();
  32.         public DefaultTenantManager(ITenantContextAccessor tenantContextAccessor)
  33.         {
  34.             _tenantContextAccessor = tenantContextAccessor;
  35.         }
  36.         public List<string> GetAll()
  37.         {
  38.             return _cache.Keys.ToList();
  39.         }
  40.         public TenantContext GetCurrentTenantContext()
  41.         {
  42.             return _tenantContextAccessor.TenantContext;
  43.         }
  44.         public bool AddTenantSharding(string tenantId, IShardingRuntimeContext shardingRuntimeContext)
  45.         {
  46.             return _cache.TryAdd(tenantId, shardingRuntimeContext);
  47.         }
  48.         public TenantScope CreateScope(string tenantId)
  49.         {
  50.             if (!_cache.TryGetValue(tenantId, out var shardingRuntimeContext))
  51.             {
  52.                 throw new InvalidOperationException("未找到对应租户的配置");
  53.             }
  54.             _tenantContextAccessor.TenantContext = new TenantContext(shardingRuntimeContext);
  55.             return new TenantScope(_tenantContextAccessor);
  56.         }
  57.     }
  58.     //当前租户上下文访问者
  59.     public interface ITenantContextAccessor
  60.     {
  61.         TenantContext? TenantContext { get; set; }
  62.     }
  63.    //当前租户上下文访问者实现
  64.     public class TenantContextAccessor:ITenantContextAccessor
  65.     {
  66.         private static readonly AsyncLocal<TenantContext?> _tenantContext = new AsyncLocal<TenantContext?>();
  67.         public TenantContext? TenantContext
  68.         {
  69.             get => _tenantContext.Value;
  70.             set => _tenantContext.Value = value;
  71.         }
  72.     }
  73.    //租户上下文
  74.     public class TenantContext
  75.     {
  76.         private readonly IShardingRuntimeContext _shardingRuntimeContext;
  77.         public TenantContext(IShardingRuntimeContext shardingRuntimeContext)
  78.         {
  79.             _shardingRuntimeContext = shardingRuntimeContext;
  80.         }
  81.         public IShardingRuntimeContext GetShardingRuntimeContext()
  82.         {
  83.             return _shardingRuntimeContext;
  84.         }
  85.     }
  86.     //用来切换实现当前操作租户环境
  87.     public class TenantScope:IDisposable
  88.     {
  89.         public TenantScope(ITenantContextAccessor tenantContextAccessor)
  90.         {
  91.             TenantContextAccessor = tenantContextAccessor;
  92.         }
  93.         public ITenantContextAccessor TenantContextAccessor { get; }
  94.         public void Dispose()
  95.         {
  96.             TenantContextAccessor.TenantContext = null;
  97.         }
  98.     }
复制代码
构思ShardingCore如何不通过依赖注入使用

其实ShardingCore可以默认不在依赖注入中进行依赖注入,首先我们看下普通情况下ShardingCore如何实现非依赖注入获取分片上下文
  1. var shardingRuntimeContext = new ShardingRuntimeBuilder<DefaultShardingDbContext>()
  2.     .UseRouteConfig(o =>
  3.     {
  4.         o.AddShardingTableRoute<SysUserTableRoute>();
  5.     }).UseConfig(o =>
  6.     {
  7.         o.ThrowIfQueryRouteNotMatch = false;
  8.         o.UseShardingQuery((conStr, builder) =>
  9.         {
  10.             builder.UseMySql(conStr, new MySqlServerVersion(new Version()))
  11.                 .UseLoggerFactory(efLogger)
  12.                 .UseQueryTrackingBehavior(QueryTrackingBehavior.NoTracking);
  13.         });
  14.         o.UseShardingTransaction((connection, builder) =>
  15.         {
  16.             builder
  17.                 .UseMySql(connection, new MySqlServerVersion(new Version()))
  18.                 .UseLoggerFactory(efLogger)
  19.                 .UseQueryTrackingBehavior(QueryTrackingBehavior.NoTracking);
  20.         });
  21.         o.AddDefaultDataSource("ds0",
  22.             "server=127.0.0.1;port=3306;database=dbdbd0;userid=root;password=root;");
  23.         o.UseShardingMigrationConfigure(b =>
  24.         {
  25.             b.ReplaceService<IMigrationsSqlGenerator, ShardingMySqlMigrationsSqlGenerator>();
  26.         });
  27.     }).ReplaceService<ITableEnsureManager, MySqlTableEnsureManager>(ServiceLifetime.Singleton)
  28.     .Build();
复制代码
这样我们就获得了IShardingRuntimeContext,将不同的IShardingRuntimeContext放到不同的数据库中我们就可以实现不同的租户了
订单表
  1.     public class Order
  2.     {
  3.         public string Id { get; set; }
  4.         public string Name { get; set; }
  5.         public DateTime CreationTime { get; set; }
  6.         public bool IsDeleted { get; set; }
  7.     }
复制代码
租户DbContext
  1.     public class TenantDbContext:AbstractShardingDbContext,IShardingTableDbContext
  2.     {
  3.         public TenantDbContext(DbContextOptions<TenantDbContext> options) : base(options)
  4.         {
  5.         }
  6.         protected override void OnModelCreating(ModelBuilder modelBuilder)
  7.         {
  8.             base.OnModelCreating(modelBuilder);
  9.             modelBuilder.ApplyConfiguration(new OrderMap());
  10.         }
  11.         public IRouteTail RouteTail { get; set; }
  12.     }
复制代码
创建订单路由

订单按月分片路由

注意这边我们简单的通过采用一个静态字段来实现
  1.     public class OrderMonthTableRoute:AbstractSimpleShardingMonthKeyDateTimeVirtualTableRoute<Order>
  2.     {
  3.         private readonly ShardingTenantOptions _shardingTenantOptions;
  4.         public OrderMonthTableRoute(ShardingTenantOptions shardingTenantOptions)
  5.         {
  6.             _shardingTenantOptions = shardingTenantOptions;
  7.         }
  8.         public override void Configure(EntityMetadataTableBuilder<Order> builder)
  9.         {
  10.             builder.ShardingProperty(o => o.CreationTime);
  11.         }
  12.         public override bool AutoCreateTableByTime()
  13.         {
  14.             return true;
  15.         }
  16.         public override DateTime GetBeginTime()
  17.         {
  18.             return _shardingTenantOptions.BeginTimeForSharding;
  19.         }
  20.     }
复制代码
订单取模分片路由
  1. public class OrderModTableRoute:AbstractSimpleShardingModKeyStringVirtualTableRoute<Order>
  2. {
  3.     private readonly ShardingTenantOptions _shardingTenantOptions;
  4.     public OrderModTableRoute(ShardingTenantOptions shardingTenantOptions) : base(2, 5)
  5.     {
  6.         _shardingTenantOptions = shardingTenantOptions;
  7.     }
  8.     public override void Configure(EntityMetadataTableBuilder<Order> builder)
  9.     {
  10.         builder.ShardingProperty(o => o.Id);
  11.     }
  12. }
复制代码
实现多数据库的code-first迁移

具体参考之前的博客EFCore高级Saas系统下单DbContext如何支持不同数据库的迁移
https://www.cnblogs.com/xuejiaming/p/16510482.html
分片创建者
  1. public interface IShardingBuilder
  2. {
  3.     IShardingRuntimeContext Build(ShardingTenantOptions tenantOptions);
  4. }
  5. public class DefaultShardingBuilder:IShardingBuilder
  6. {
  7.     public static readonly ILoggerFactory efLogger = LoggerFactory.Create(builder =>
  8.     {
  9.         builder.AddFilter((category, level) =>
  10.             category == DbLoggerCategory.Database.Command.Name && level == LogLevel.Information).AddConsole();
  11.     });
  12.     private readonly IServiceProvider _serviceProvider;
  13.     public DefaultShardingBuilder(IServiceProvider serviceProvider)
  14.     {
  15.         _serviceProvider = serviceProvider;
  16.     }
  17.     public IShardingRuntimeContext Build(ShardingTenantOptions tenantOptions)
  18.     {
  19.         var shardingRuntimeBuilder = new ShardingRuntimeBuilder<TenantDbContext>()
  20.             .UseRouteConfig(o =>
  21.             {
  22.                 if (tenantOptions.OrderShardingType == OrderShardingTypeEnum.Mod)
  23.                 {
  24.                     o.AddShardingTableRoute<OrderModTableRoute>();
  25.                 }
  26.                 if (tenantOptions.OrderShardingType == OrderShardingTypeEnum.ByMonth)
  27.                 {
  28.                     o.AddShardingTableRoute<OrderMonthTableRoute>();
  29.                 }
  30.             }).UseConfig(o =>
  31.             {
  32.                 o.ThrowIfQueryRouteNotMatch = false;
  33.                 o.UseShardingQuery((conStr, builder) =>
  34.                 {
  35.                     if (tenantOptions.DbType == DbTypeEnum.MYSQL)
  36.                     {
  37.                         builder.UseMySql(conStr, new MySqlServerVersion(new Version()))
  38.                             .UseMigrationNamespace(new MySqlMigrationNamespace());
  39.                     }
  40.                     if (tenantOptions.DbType == DbTypeEnum.MSSQL)
  41.                     {
  42.                         builder.UseSqlServer(conStr)
  43.                             .UseMigrationNamespace(new SqlServerMigrationNamespace());
  44.                     }
  45.                     builder.UseLoggerFactory(efLogger)
  46.                         .UseQueryTrackingBehavior(QueryTrackingBehavior.NoTracking)
  47.                         .ReplaceService<IMigrationsAssembly,MultiDatabaseMigrationsAssembly>();
  48.                 });
  49.                 o.UseShardingTransaction((connection, builder) =>
  50.                 {
  51.                     if (tenantOptions.DbType == DbTypeEnum.MYSQL)
  52.                     {
  53.                         builder
  54.                             .UseMySql(connection, new MySqlServerVersion(new Version()));
  55.                             //.UseMigrationNamespace(new MySqlMigrationNamespace());//迁移只会用connection string创建所以可以不加
  56.                     }
  57.                     if (tenantOptions.DbType == DbTypeEnum.MSSQL)
  58.                     {
  59.                         builder.UseSqlServer(connection);
  60.                         //.UseMigrationNamespace(new SqlServerMigrationNamespace());
  61.                     }
  62.                     builder.UseLoggerFactory(efLogger)
  63.                         .UseQueryTrackingBehavior(QueryTrackingBehavior.NoTracking);
  64.                 });
  65.                 o.AddDefaultDataSource(tenantOptions.DefaultDataSourceName,tenantOptions.DefaultConnectionString);
  66.                 //注意这个迁移必须要十分重要
  67.                 //注意这个迁移必须要十分重要
  68.                 //注意这个迁移必须要十分重要
  69.                 //注意这个迁移必须要十分重要
  70.                 o.UseShardingMigrationConfigure(b =>
  71.                 {
  72.                     if (tenantOptions.DbType == DbTypeEnum.MYSQL)
  73.                     {
  74.                         b.ReplaceService<IMigrationsSqlGenerator, ShardingMySqlMigrationsSqlGenerator>();
  75.                     }
  76.                     if (tenantOptions.DbType == DbTypeEnum.MSSQL)
  77.                     {
  78.                         b.ReplaceService<IMigrationsSqlGenerator, ShardingSqlServerMigrationsSqlGenerator>();
  79.                     }
  80.                 });
  81.             }).AddServiceConfigure(s =>
  82.             {
  83.                 //IShardingRuntimeContext内部的依赖注入
  84.                 s.AddSingleton(tenantOptions);
  85.             });
  86.         
  87.         if (tenantOptions.DbType == DbTypeEnum.MYSQL)
  88.         {
  89.             shardingRuntimeBuilder.ReplaceService<ITableEnsureManager, MySqlTableEnsureManager>(ServiceLifetime
  90.                 .Singleton);
  91.         }
  92.         if (tenantOptions.DbType == DbTypeEnum.MSSQL)
  93.         {
  94.             shardingRuntimeBuilder.ReplaceService<ITableEnsureManager, SqlServerTableEnsureManager>(ServiceLifetime
  95.                 .Singleton);
  96.         }
  97.         return shardingRuntimeBuilder.Build(_serviceProvider);
  98.     }
  99. }
复制代码
到此为止基本上我们已经完成了多租户的大部分配置了,jwt部分就不在这边赘述了因为之前有实现过
Startup

主要关键的启动点我们应该怎么配置呢
启动初始化租户

首先我们需要针对程序启动后进行租户的初始化操作
  1.     public static class TenantExtension
  2.     {
  3.         public static void InitTenant(this IServiceProvider serviceProvider)
  4.         {
  5.             var tenantManager = serviceProvider.GetRequiredService<ITenantManager>();
  6.             var shardingBuilder = serviceProvider.GetRequiredService<IShardingBuilder>();
  7.             
  8.             using (var scope = serviceProvider.CreateScope())
  9.             {
  10.                 var identityDbContext = scope.ServiceProvider.GetRequiredService<IdentityDbContext>();
  11.                 identityDbContext.Database.Migrate();
  12.                 var sysUserTenantConfigs = identityDbContext.Set<SysUserTenantConfig>().ToList();
  13.                 if (sysUserTenantConfigs.Any())
  14.                 {
  15.                     foreach (var sysUserTenantConfig in sysUserTenantConfigs)
  16.                     {
  17.                         var shardingTenantOptions = JsonConvert.DeserializeObject<ShardingTenantOptions>(sysUserTenantConfig.ConfigJson);
  18.                         var shardingRuntimeContext = shardingBuilder.Build(shardingTenantOptions);
  19.                         
  20.                         tenantManager.AddTenantSharding(sysUserTenantConfig.UserId, shardingRuntimeContext);
  21.                     }
  22.                 }
  23.             }
  24.             var tenantIds = tenantManager.GetAll();
  25.             foreach (var tenantId in tenantIds)
  26.             {
  27.                 using(tenantManager.CreateScope(tenantId))
  28.                 using (var scope = serviceProvider.CreateScope())
  29.                 {
  30.                     var shardingRuntimeContext = tenantManager.GetCurrentTenantContext().GetShardingRuntimeContext();
  31.                     //开启定时任务
  32.                     shardingRuntimeContext.UseAutoShardingCreate();
  33.                     var tenantDbContext = scope.ServiceProvider.GetService<TenantDbContext>();
  34.                     //
  35.                     tenantDbContext.Database.Migrate();
  36.                     //补偿表
  37.                     shardingRuntimeContext.UseAutoTryCompensateTable();
  38.                 }
  39.             }
  40.         }
  41.     }
复制代码
请求租户中间件

为了让我们的所有请求都可以使用指定对应的租户数据库
  1.     public class TenantSelectMiddleware
  2.     {
  3.         private readonly RequestDelegate _next;
  4.         private readonly ITenantManager _tenantManager;
  5.         public TenantSelectMiddleware(RequestDelegate next,ITenantManager tenantManager)
  6.         {
  7.             _next = next;
  8.             _tenantManager = tenantManager;
  9.         }
  10.         /// <summary>
  11.         /// 1.中间件的方法必须叫Invoke,且为public,非static。
  12.         /// 2.Invoke方法第一个参数必须是HttpContext类型。
  13.         /// 3.Invoke方法必须返回Task。
  14.         /// 4.Invoke方法可以有多个参数,除HttpContext外其它参数会尝试从依赖注入容器中获取。
  15.         /// 5.Invoke方法不能有重载。
  16.         /// </summary>
  17.         /// Author : Napoleon
  18.         /// Created : 2020/1/30 21:30
  19.         public async Task Invoke(HttpContext context)
  20.         {
  21.             if (context.Request.Path.ToString().StartsWith("/api/tenant", StringComparison.CurrentCultureIgnoreCase))
  22.             {
  23.                 if (!context.User.Identity.IsAuthenticated)
  24.                 {
  25.                     await _next(context);
  26.                     return;
  27.                 }
  28.                 var tenantId = context.User.Claims.FirstOrDefault((o) => o.Type == "uid")?.Value;
  29.                 if (string.IsNullOrWhiteSpace(tenantId))
  30.                 {
  31.                     await DoUnAuthorized(context, "not found tenant id");
  32.                     return;
  33.                 }
  34.                 using (_tenantManager.CreateScope(tenantId))
  35.                 {
  36.                     await _next(context);
  37.                 }
  38.             }
  39.             else
  40.             {
  41.                 await _next(context);
  42.             }
  43.         }
  44.         private async Task DoUnAuthorized(HttpContext context, string msg)
  45.         {
  46.             context.Response.StatusCode = 403;
  47.             await context.Response.WriteAsync(msg);
  48.         }
  49.     }
复制代码
编写登录注册操作

startup处配置
  1.     [Route("api/[controller]/[action]")]
  2.     [ApiController]
  3.     [AllowAnonymous]
  4.     public class PassportController : ControllerBase
  5.     {
  6.         private readonly IServiceProvider _serviceProvider;
  7.         private readonly IdentityDbContext _identityDbContext;
  8.         private readonly ITenantManager _tenantManager;
  9.         private readonly IShardingBuilder _shardingBuilder;
  10.         public PassportController(IServiceProvider serviceProvider, IdentityDbContext identityDbContext,
  11.             ITenantManager tenantManager, IShardingBuilder shardingBuilder)
  12.         {
  13.             _serviceProvider = serviceProvider;
  14.             _identityDbContext = identityDbContext;
  15.             _tenantManager = tenantManager;
  16.             _shardingBuilder = shardingBuilder;
  17.         }
  18.         [HttpPost]
  19.         public async Task<IActionResult> Register(RegisterRequest request)
  20.         {
  21.             if (await _identityDbContext.Set<SysUser>().AnyAsync(o => o.Name == request.Name))
  22.                 return BadRequest("user not exists");
  23.             var sysUser = new SysUser()
  24.             {
  25.                 Id = Guid.NewGuid().ToString("n"),
  26.                 Name = request.Name,
  27.                 Password = request.Password,
  28.                 CreationTime = DateTime.Now
  29.             };
  30.             var shardingTenantOptions = new ShardingTenantOptions()
  31.             {
  32.                 DbType = request.DbType,
  33.                 OrderShardingType = request.OrderShardingType,
  34.                 BeginTimeForSharding = request.BeginTimeForSharding.Value,
  35.                 DefaultDataSourceName = "ds0",
  36.                 DefaultConnectionString = GetDefaultString(request.DbType, sysUser.Id)
  37.             };
  38.             var sysUserTenantConfig = new SysUserTenantConfig()
  39.             {
  40.                 Id = Guid.NewGuid().ToString("n"),
  41.                 UserId = sysUser.Id,
  42.                 CreationTime = DateTime.Now,
  43.                 ConfigJson = JsonConvert.SerializeObject(shardingTenantOptions)
  44.             };
  45.             await _identityDbContext.AddAsync(sysUser);
  46.             await _identityDbContext.AddAsync(sysUserTenantConfig);
  47.             await _identityDbContext.SaveChangesAsync();
  48.             var shardingRuntimeContext = _shardingBuilder.Build(shardingTenantOptions);
  49.             _tenantManager.AddTenantSharding(sysUser.Id, shardingRuntimeContext);
  50.             using (_tenantManager.CreateScope(sysUser.Id))
  51.             using (var scope = _serviceProvider.CreateScope())
  52.             {
  53.                 var runtimeContext = _tenantManager.GetCurrentTenantContext().GetShardingRuntimeContext();
  54.                 runtimeContext.UseAutoShardingCreate(); //启动定时任务
  55.                 var tenantDbContext = scope.ServiceProvider.GetService<TenantDbContext>();
  56.                 tenantDbContext.Database.Migrate();
  57.                 runtimeContext.UseAutoTryCompensateTable();
  58.             }
  59.             return Ok();
  60.         }
  61.         [HttpPost]
  62.         public async Task<IActionResult> Login(LoginRequest request)
  63.         {
  64.             var sysUser = await _identityDbContext.Set<SysUser>()
  65.                 .FirstOrDefaultAsync(o => o.Name == request.Name && o.Password == request.Password);
  66.             if (sysUser == null)
  67.                 return BadRequest("name or password error");
  68.             //秘钥,就是标头,这里用Hmacsha256算法,需要256bit的密钥
  69.             var securityKey =
  70.                 new SigningCredentials(new SymmetricSecurityKey(Encoding.ASCII.GetBytes("123123!@#!@#123123")),
  71.                     SecurityAlgorithms.HmacSha256);
  72.             //Claim,JwtRegisteredClaimNames中预定义了好多种默认的参数名,也可以像下面的Guid一样自己定义键名.
  73.             //ClaimTypes也预定义了好多类型如role、email、name。Role用于赋予权限,不同的角色可以访问不同的接口
  74.             //相当于有效载荷
  75.             var claims = new Claim[]
  76.             {
  77.                 new Claim(JwtRegisteredClaimNames.Iss, "https://localhost:5000"),
  78.                 new Claim(JwtRegisteredClaimNames.Aud, "api"),
  79.                 new Claim("id", Guid.NewGuid().ToString("n")),
  80.                 new Claim("uid", sysUser.Id),
  81.             };
  82.             SecurityToken securityToken = new JwtSecurityToken(
  83.                 signingCredentials: securityKey,
  84.                 expires: DateTime.Now.AddHours(2), //过期时间
  85.                 claims: claims
  86.             );
  87.             var token = new JwtSecurityTokenHandler().WriteToken(securityToken);
  88.             return Ok(token);
  89.         }
  90.         private string GetDefaultString(DbTypeEnum dbType, string userId)
  91.         {
  92.             switch (dbType)
  93.             {
  94.                 case DbTypeEnum.MSSQL:
  95.                     return $"Data Source=localhost;Initial Catalog=DB{userId};Integrated Security=True;";
  96.                 case DbTypeEnum.MYSQL:
  97.                     return $"server=127.0.0.1;port=3306;database=DB{userId};userid=root;password=L6yBtV6qNENrwBy7;";
  98.                 default: throw new NotImplementedException();
  99.             }
  100.         }
  101.     }
  102.     public class RegisterRequest
  103.     {
  104.         public string Name { get; set; }
  105.         public string Password { get; set; }
  106.         public DbTypeEnum DbType { get; set; }
  107.         public OrderShardingTypeEnum OrderShardingType { get; set; }
  108.         public DateTime? BeginTimeForSharding { get; set; }
  109.     }
  110.     public class LoginRequest
  111.     {
  112.         public string Name { get; set; }
  113.         public string Password { get; set; }
  114.     }
复制代码
启动配置
  1. var builder = WebApplication.CreateBuilder(args);
  2. // Add services to the container.
  3. builder.Services.AddControllers();
  4. builder.Services.AddAuthentication();
  5. #region 用户系统配置
  6. builder.Services.AddDbContext<IdentityDbContext>(o =>
  7.     o.UseSqlServer("Data Source=localhost;Initial Catalog=IdDb;Integrated Security=True;"));
  8. //生成密钥
  9. var keyByteArray = Encoding.ASCII.GetBytes("123123!@#!@#123123");
  10. var signingKey = new SymmetricSecurityKey(keyByteArray);
  11. //认证参数
  12. builder.Services.AddAuthentication("Bearer")
  13.     .AddJwtBearer(o =>
  14.     {
  15.         o.TokenValidationParameters = new TokenValidationParameters
  16.         {
  17.             ValidateIssuerSigningKey = true,
  18.             IssuerSigningKey = signingKey,
  19.             ValidateIssuer = true,
  20.             ValidIssuer = "https://localhost:5000",
  21.             ValidateAudience = true,
  22.             ValidAudience = "api",
  23.             ValidateLifetime = true,
  24.             ClockSkew = TimeSpan.Zero,
  25.             RequireExpirationTime = true,
  26.         };
  27.     });
  28. #endregion
  29. builder.Services.AddSingleton<ITenantManager, DefaultTenantManager>();
  30. builder.Services.AddSingleton<ITenantContextAccessor, TenantContextAccessor>();
  31. builder.Services.AddSingleton<IShardingBuilder, DefaultShardingBuilder>();
  32. #region 配置ShardingCore
  33. var provider = builder.Configuration.GetValue("Provider", "UnKnown");
  34. //Add-Migration InitialCreate -Context TenantDbContext -OutputDir Migrations\SqlServer -Args "--provider SqlServer"
  35. //Add-Migration InitialCreate -Context TenantDbContext -OutputDir Migrations\MySql -Args "--provider MySql"
  36. builder.Services.AddDbContext<TenantDbContext>((sp, b) =>
  37. {
  38.     var tenantManager = sp.GetRequiredService<ITenantManager>();
  39.     var currentTenantContext = tenantManager.GetCurrentTenantContext();
  40.     //如果有上下文那么创建租户dbcontext否则就是启动命令Add-Migration
  41.     if (currentTenantContext != null)
  42.     {
  43.         var shardingRuntimeContext = currentTenantContext.GetShardingRuntimeContext();
  44.         b.UseDefaultSharding<TenantDbContext>(shardingRuntimeContext);
  45.     }
  46.     if (args.IsNotEmpty())
  47.     {
  48.         //命令启动时为了保证Add-Migration正常运行
  49.         if (provider == "MySql")
  50.         {
  51.             b.UseMySql("server=127.0.0.1;port=3306;database=TenantDb;userid=root;password=L6yBtV6qNENrwBy7;",
  52.                     new MySqlServerVersion(new Version()))
  53.                 .UseMigrationNamespace(new MySqlMigrationNamespace())
  54.                 .ReplaceService<IMigrationsAssembly, MultiDatabaseMigrationsAssembly>();
  55.             return;
  56.         }
  57.         if (provider == "SqlServer")
  58.         {
  59.             b.UseSqlServer("Data Source=localhost;Initial Catalog=TenantDb;Integrated Security=True;")
  60.                 .UseMigrationNamespace(new SqlServerMigrationNamespace())
  61.                 .ReplaceService<IMigrationsAssembly, MultiDatabaseMigrationsAssembly>();
  62.             return;
  63.         }
  64.     }
  65. });
  66. #endregion
  67. var app = builder.Build();
  68. //初始化启动配置租户信息
  69. app.Services.InitTenant();
  70. app.UseAuthorization();
  71. //在认证后启用租户选择中间件
  72. app.UseMiddleware<TenantSelectMiddleware>();
  73. app.MapControllers();
  74. app.Run();
复制代码
添加迁移脚本

持久化identity迁移

多租户SqlServer版本

多租户MySql版本

启动程序

启动程序我们发现IdentityDbContext已经创建好了,并且支持了自动迁移

创建A租户
  1. {
  2.     "Name":"A",
  3.     "Password":"A",
  4.     "DbType":1,
  5.     "OrderShardingType":2,
  6.     "BeginTimeForSharding":"2022-01-01",
  7.     "MigrationNamespace":"ShardingCoreMultiTenantSys.Migrations.SqlServer"
  8. }
复制代码
注意:MigrationNamespace应该自动生成,这边只是为了演示方便没写

完成
创建B租户
  1. {
  2.     "Name":"B",
  3.     "Password":"B",
  4.     "DbType":2,
  5.     "OrderShardingType":1,
  6.     "BeginTimeForSharding":"2022-01-01",
  7.     "MigrationNamespace":"ShardingCoreMultiTenantSys.Migrations.Myql"
  8. }
复制代码
完美创建

创建C租户
  1. {
  2.     "Name":"C",
  3.     "Password":"C",
  4.     "DbType":1,
  5.     "OrderShardingType":2,
  6.     "BeginTimeForSharding":"2022-06-01",
  7.     "MigrationNamespace":"ShardingCoreMultiTenantSys.Migrations.SqlServer"
  8. }
复制代码
C租户完美创建并且和A租户采用一样的分片规则不一样的分片起始时间

分别对abc进行crud

首先获取token,然后插入
A租户

B租户

C租户

最后完成

最后的最后

附上demo:ShardingCoreMultiTenantSys https://github.com/xuejmnet/ShardingCoreMultiTenantSys
您都看到这边了确定不点个star或者赞吗,一款.Net不得不学的分库分表解决方案,简单理解为sharding-jdbc在.net中的实现并且支持更多特性和更优秀的数据聚合,拥有原生性能的97%,并且无业务侵入性,支持未分片的所有efcore原生查询

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有账号?立即注册

x
回复

使用道具 举报

0 个回复

倒序浏览

快速回复

您需要登录后才可以回帖 登录 or 立即注册

本版积分规则

民工心事

金牌会员
这个人很懒什么都没写!
快速回复 返回顶部 返回列表