EF多租户实例：快速实现分库分表

前言

来到这篇随笔，我们继续演示如何实现EF多租户。

今天主要是演示多租户下的变形，为下图所示

实施

项目结构

这次我们的示例项目进行了精简，仅有一个API项目，直接包含所有代码。

其中Controller，StoreContext，Entity都完全和以往的示例一模一样，这里就不再过多介绍了。

具有主要区别的是 CombinedConnectionGenerator 和 Startup 

代码解释

1. 首先要关注的是作为入口的 Startup ，还是一个套路，分别在 ConfigureService 注册EF多租户， 在 Configure 配置中间件。

ConfigureService 还是一贯的简单。但是注意这里使用的 AddMySqlPerTable 这个模式。

在混合的模式中，需要已最小的单元作为服务注册。由于这次是数据库和数据表混合模式，所以需要用数据表来注册。

public void ConfigureServices(IServiceCollection services)
{
    services.AddScoped<IConnectionGenerator, CombindedConnectionGenerator>();
    services.AddMySqlPerTable<StoreDbContext>(settings =>
    {
        settings.ConnectionPrefix = "mysql_";
    });
    services.AddControllers();
}

Configure的使用更加简单，只需要添加中间件 TenantInfoMiddleware 即可。

public void Configure(IApplicationBuilder app, IWebHostEnvironment env)
{
    if (env.IsDevelopment())
    {
        app.UseDeveloperExceptionPage();
    }

    app.UseMiddleware<TenantInfoMiddleware>();

    app.UseRouting();

    app.UseEndpoints(endpoints =>
    {
        endpoints.MapControllers();
    });
}

2. 这次需要自己实现 ConnectionGenerator 

关键点有2个，

第一个关键点，由于我们的类库是同时支持多个DbContext的，所以需要有TenantKey去区分。由于有种特殊情况，需要一个 ConnectionGenerator 同时支持多个 DbContext ，所以这里提供了 MatchTenantKey 方法作为补充的判断依据。

可以看出来，我们这里TenantKey 为空，所以一般都不会匹配中。示例中完全是依靠 MatchTenantKey 来做匹配的。

第二个关键点，GetConnection 作为最主要的逻辑方法，通过对TenantName 的数字部分进行取模，最终拼接处ConnectionString的键值

并且通过 Configuration 获取连接字符串

public class CombindedConnectionGenerator : IConnectionGenerator
{
    private readonly IConfiguration configuration;
    public string TenantKey => "";

    public CombindedConnectionGenerator(IConfiguration configuration)
    {
        this.configuration = configuration;
    }


    public string GetConnection(TenantOption option, TenantInfo tenantInfo)
    {
        var span = tenantInfo.Name.AsSpan();
        if (span.Length > 4 && int.TryParse(span[5].ToString(), out var number))
        {
            return configuration.GetConnectionString($"{option.ConnectionPrefix}container{number % 2 + 1}");
        }
        throw new NotSupportedException("tenant invalid");
    }

    public bool MatchTenantKey(string tenantKey)
    {
        return true;
    }
}

检验结果

检验结果我觉得已经没有必要的，都是同样的套路，主要的区别是，之前的只有一个数据库，或者多个数据库

这次的混合模式，主要是一个数据库作为一个container，里面可以同时包含多个product数据表。

Container1

Container2

总结

其实这个例子也是非常简单的，目的是让每个人都能快速应用复杂的分库分表

下一篇文章将会通过多租户实现读写分离。

关于这个文章的所有代码，已经同步到Github

https://github.com/woailibain/kiwiho.EFcore.MultiTenant/tree/master/example/mix_mode/kiwiho.EFcore.MultiTenant.MixMode