如何获取Token声明值并注入DbContext实现多租户全局查询

我来帮你搞定这个多租户全局查询的问题，要从Token里拿到CompanyId并注入到你的DbContext里，咱们可以按下面的步骤来操作：

1. 先注册IHttpContextAccessor #

ASP.NET Core默认不会自动注册这个服务，所以得先在Startup.cs（或者.NET 6+的Program.cs）里添加注册：

services.AddHttpContextAccessor();

2. 修改DbContext的注入逻辑 #

你当前的注入代码里companyId是硬编码的1，现在要改成从Token声明里读取。咱们可以通过IHttpContextAccessor获取当前请求的HttpContext，然后从中提取CompanyId声明：

services.AddScoped(provider => {
    var optionsBuilder = new DbContextOptionsBuilder<MyDbContext>();
    var options = optionsBuilder.UseSqlServer(Configuration.GetConnectionString("MyDbDatabase")).Options;
    
    // 获取HttpContextAccessor实例
    var httpContextAccessor = provider.GetRequiredService<IHttpContextAccessor>();
    var httpContext = httpContextAccessor.HttpContext;
    
    if (httpContext == null)
    {
        // 处理无HttpContext的场景（比如后台任务），可根据业务需求调整逻辑
        throw new InvalidOperationException("无法获取当前请求上下文，无法读取CompanyId");
    }
    
    // 从用户Claims中提取CompanyId，注意这里的Claim名称要和你Token里的一致
    var companyIdClaim = httpContext.User.Claims.FirstOrDefault(c => c.Type == "CompanyId");
    if (companyIdClaim == null || !int.TryParse(companyIdClaim.Value, out var companyId))
    {
        // 处理Claim缺失或格式错误的情况
        throw new UnauthorizedAccessException("Token中未包含有效的CompanyId声明");
    }
    
    return new MyDbContext(options, companyId);
});

3. 确保DbContext构造函数正确接收CompanyId #

你的MyDbContext需要保留接收DbContextOptions和companyId的构造函数，并把companyId存为私有字段供全局查询过滤器使用，就像你已经写的那样：

public class MyDbContext : DbContext
{
    private readonly int _companyId;

    public MyDbContext(DbContextOptions<MyDbContext> options, int companyId) : base(options)
    {
        _companyId = companyId;
    }

    protected override void OnModelCreating(ModelBuilder modelBuilder)
    {
        base.OnModelCreating(modelBuilder);
        // 应用全局查询过滤器
        SetGlobalQuery<BaseEntity>(modelBuilder);
    }

    public void SetGlobalQuery<T>(ModelBuilder builder) where T : BaseEntity
    {
        builder.Entity<T>().HasQueryFilter(e => e.CompanyId == _companyId);
    }
}

一些注意事项 #

• 要确保Token里的CompanyId声明名称和你代码里的一致（比如上面的"CompanyId"），如果你的Token用了自定义命名空间的Claim类型（比如http://schemas.yourdomain.com/claims/companyid），记得替换成对应的字符串。
• 如果你的应用有非HTTP请求场景（比如定时后台任务），HttpContext会是null，这时候需要单独处理这类场景的CompanyId注入逻辑，比如通过配置文件或其他方式传递。
• 建议在提取Claim时做必要的校验，避免因Token无效或Claim缺失导致程序异常。

内容的提问来源于stack exchange，提问作者Luis Henrique Kraus