使用Autofac多租户时，如何在AuthenticationHandler注入租户作用域服务？

一、调整中间件顺序，确保租户识别早于认证处理 #

Autofac.AspNetCore.Multitenant的租户识别逻辑依赖UseAutofacMultitenantRequestMiddleware中间件执行，若该中间件在认证中间件之后运行，就会导致AuthenticationHandler实例化时租户还未被识别。

你需要在Program.cs中先注册租户识别中间件，再注册认证中间件：

var builder = WebApplication.CreateBuilder(args);

// 1. 配置Autofac多租户
builder.Host.UseServiceProviderFactory(new AutofacMultitenantServiceProviderFactory());
builder.Host.ConfigureContainer<ContainerBuilder>(containerBuilder =>
{
    // 注册租户识别策略（从Header获取租户ID）
    containerBuilder.RegisterType<HeaderTenantIdentificationStrategy>()
                   .As<ITenantIdentificationStrategy>()
                   .SingleInstance();

    // 注册租户特定服务（比如数据库上下文）
    containerBuilder.Register(ctx =>
    {
        ctx.Resolve<ITenantIdentificationStrategy>().IdentifyTenant(out var tenantId);
        // 根据租户ID创建对应DbContext
        var options = new DbContextOptionsBuilder<AppDbContext>()
            .UseSqlServer(GetConnectionStringForTenant(tenantId?.ToString()))
            .Options;
        return new AppDbContext(options);
    }).AsSelf().InstancePerTenant();
});

var app = builder.Build();

// 2. 先启用Autofac多租户请求中间件
app.UseAutofacMultitenantRequestMiddleware();

// 3. 再启用认证中间件
app.UseAuthentication();
app.UseAuthorization();

// ... 其他中间件和端点配置

调整后，租户会在AuthenticationHandler实例化前完成识别，此时你可以直接在Handler的构造函数中注入租户作用域服务（如AppDbContext）。

二、在HandleAuthenticateAsync中手动获取租户作用域服务 #

如果无法调整中间件顺序，或需要在认证逻辑中动态获取服务，可以通过HttpContext.RequestServices获取租户作用域的服务实例：

public class CustomAuthenticationHandler : AuthenticationHandler<AuthenticationSchemeOptions>
{
    public CustomAuthenticationHandler(IOptionsMonitor<AuthenticationSchemeOptions> options, 
                                       ILoggerFactory logger, 
                                       UrlEncoder encoder, 
                                       ISystemClock clock) 
        : base(options, logger, encoder, clock)
    {
    }

    protected override async Task<AuthenticateResult> HandleAuthenticateAsync()
    {
        // 从HttpContext的RequestServices中获取租户作用域的DbContext
        using var dbContext = Context.RequestServices.GetRequiredService<AppDbContext>();
        
        // 执行租户数据库操作，比如验证凭证
        var credentialValid = await ValidateCredentialFromTenantDb(dbContext, Request.Headers["Authorization"]);
        
        if (credentialValid)
        {
            var claims = new List<Claim> { new Claim(ClaimTypes.Name, "user") };
            var identity = new ClaimsIdentity(claims, Scheme.Name);
            var principal = new ClaimsPrincipal(identity);
            var ticket = new AuthenticationTicket(principal, Scheme.Name);
            return AuthenticateResult.Success(ticket);
        }
        
        return AuthenticateResult.Fail("Invalid credential");
    }

    private async Task<bool> ValidateCredentialFromTenantDb(AppDbContext dbContext, StringValues authHeader)
    {
        // 实现租户数据库中的凭证验证逻辑
        return true;
    }
}

关键注意点 #

• RequestServices是当前租户的作用域服务容器，从中获取的服务会自动关联当前租户的数据库。
• 必须确保UseAutofacMultitenantRequestMiddleware已执行，否则RequestServices可能未切换到租户作用域，获取的服务会是默认实例而非租户特定实例。

内容的提问来源于stack exchange，提问作者Niels Harremoes