ABP框架扩展TenantResolveContributorBase实现OrganizationId租户过滤问题

一、修复当前租户解析器的错误 #

你遇到的错误是因为ITenantResolveContext是抽象接口，在HTTP请求场景下实际传入的是HttpTenantResolveContext，且ABP的租户解析上下文不直接暴露TenantId属性，而是通过方法设置。修改你的CustomOrganizationTenantResolveContributor代码如下：

using Volo.Abp.MultiTenancy;
using Volo.Abp.AspNetCore.MultiTenancy;

public class CustomOrganizationTenantResolveContributor : TenantResolveContributorBase
{
    public const string ContributorName = "CustomOrganizationTenant";

    public override string Name => ContributorName;

    public override Task ResolveAsync(ITenantResolveContext context)
    {
        // 转型为HTTP场景专属上下文
        if (context is HttpTenantResolveContext httpContext)
        {
            var organizationIdStr = httpContext.HttpContext.Request.Headers["OrganizationId"].FirstOrDefault();
            if (!string.IsNullOrEmpty(organizationIdStr) && Guid.TryParse(organizationIdStr, out var organizationId))
            {
                context.Handled = true;
                // 使用SetTenantId方法设置租户ID，而非直接赋值
                context.SetTenantId(organizationId);
            }
        }
        
        return Task.CompletedTask;
    }
}

注意：如果你的目的是把OrganizationId当作租户ID复用多租户机制，上面的修改可以解决错误；但如果是要在租户之外新增一个独立的组织维度数据隔离，这种方式并不适配真实业务场景，下面给出更合理的实现方案。

二、正确实现「OrganizationId全局自动过滤」的方案 #

如果需求是在多租户基础上，新增组织维度的数据隔离（租户下包含多个组织，数据按组织自动过滤），不要复用租户ID，应单独实现组织的解析与全局过滤逻辑：

1. 定义OrganizationId访问器 #

在Domain层创建接口和实现，用于全局获取当前请求的OrganizationId：

public interface IOrganizationIdAccessor
{
    Guid? OrganizationId { get; }
    void SetOrganizationId(Guid? organizationId);
}

public class OrganizationIdAccessor : IOrganizationIdAccessor
{
    private static readonly AsyncLocal<Guid?> _currentOrganizationId = new AsyncLocal<Guid?>();

    public Guid? OrganizationId => _currentOrganizationId.Value;

    public void SetOrganizationId(Guid? organizationId)
    {
        _currentOrganizationId.Value = organizationId;
    }
}

2. 创建OrganizationId解析中间件 #

在HttpApi项目中创建中间件，从请求Header中提取OrganizationId并设置到访问器：

using Microsoft.AspNetCore.Http;
using Volo.Abp.DependencyInjection;

public class OrganizationIdMiddleware : IMiddleware, ITransientDependency
{
    private readonly IOrganizationIdAccessor _organizationIdAccessor;

    public OrganizationIdMiddleware(IOrganizationIdAccessor organizationIdAccessor)
    {
        _organizationIdAccessor = organizationIdAccessor;
    }

    public async Task InvokeAsync(HttpContext context, RequestDelegate next)
    {
        var organizationIdStr = context.Request.Headers["OrganizationId"].FirstOrDefault();
        if (!string.IsNullOrEmpty(organizationIdStr) && Guid.TryParse(organizationIdStr, out var organizationId))
        {
            _organizationIdAccessor.SetOrganizationId(organizationId);
        }
        else
        {
            _organizationIdAccessor.SetOrganizationId(null);
        }

        await next(context);
    }
}

在HttpApi模块中注册并启用中间件：

public class YourHttpApiModule : AbpModule
{
    public override void ConfigureServices(ServiceConfigurationContext context)
    {
        context.Services.AddTransient<OrganizationIdMiddleware>();
    }

    public override void OnApplicationInitialization(ApplicationInitializationContext context)
    {
        var app = context.GetApplicationBuilder();
        // 放在多租户中间件之后执行
        app.UseMiddleware<OrganizationIdMiddleware>();
    }
}

3. 实现全局组织数据过滤 #

实现ABP的查询过滤器，为所有带OrganizationId的实体自动添加过滤条件：

using System.Linq.Expressions;
using Volo.Abp.Data;
using Volo.Abp.EntityFrameworkCore;

// 定义接口标记需要组织过滤的实体
public interface IHasOrganization
{
    Guid OrganizationId { get; set; }
}

// 全局查询过滤器实现
public class OrganizationQueryFilter : IQueryFilter, ITransientDependency
{
    private readonly IOrganizationIdAccessor _organizationIdAccessor;

    public OrganizationQueryFilter(IOrganizationIdAccessor organizationIdAccessor)
    {
        _organizationIdAccessor = organizationIdAccessor;
    }

    public Expression<Func<TEntity, bool>> CreateFilter<TEntity>() where TEntity : class
    {
        var organizationId = _organizationIdAccessor.OrganizationId;
        // 仅对实现IHasOrganization的实体应用过滤
        if (organizationId == null || !typeof(IHasOrganization).IsAssignableFrom(typeof(TEntity)))
        {
            return e => true;
        }

        return e => ((IHasOrganization)e).OrganizationId == organizationId;
    }
}

在DbContext中应用该过滤器：

public class YourDbContext : AbpDbContext<YourDbContext>
{
    private readonly IQueryFilter _organizationQueryFilter;

    public YourDbContext(DbContextOptions<YourDbContext> options, IQueryFilter organizationQueryFilter) 
        : base(options)
    {
        _organizationQueryFilter = organizationQueryFilter;
    }

    protected override void OnModelCreating(ModelBuilder builder)
    {
        base.OnModelCreating(builder);

        // 为所有实现IHasOrganization的实体绑定过滤器
        builder.Entity<IHasOrganization>(b =>
        {
            b.HasQueryFilter(_organizationQueryFilter.CreateFilter<IHasOrganization>());
        });
    }
}

4. 实体标记组织过滤接口 #

让需要按组织过滤的实体实现IHasOrganization接口：

public class YourEntity : FullAuditedAggregateRoot<Guid>, IHasOrganization
{
    public Guid OrganizationId { get; set; }
    // 其他业务字段...
}

三、注册自定义租户解析器（若需复用多租户机制） #

如果你确实需要将OrganizationId当作租户ID使用，需在Domain模块中注册自定义解析器：

[DependsOn(typeof(AbpAspNetCoreMultiTenancyModule))]
public class YourDomainModule : AbpModule
{
    public override void ConfigureServices(ServiceConfigurationContext context)
    {
        Configure<AbpTenantResolveOptions>(options =>
        {
            // 将自定义解析器添加到租户解析器列表
            options.TenantResolvers.Add(new CustomOrganizationTenantResolveContributor());
        });
    }
}

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