Hibernate @TenantId与@JoinColumn协同失效问题求助

我们的应用已切换为多租户数据库，采用Java 21 + Hibernate 6.5.2，通过@TenantId注解区分客户端（未使用Spring）。数据库每张表都包含tenant_id列，所有实体都添加了带@TenantId注解的tenant_id字段，Hibernate能自动过滤该字段，正常持久化和获取当前租户的对象，但**@TenantId在@JoinColumn关联查询时不生效**。

示例关联代码：

@ManyToOne(fetch = FetchType.EAGER)
@JoinColumn(name = "some_type", referencedColumnName = "name")
private SomeType someType;

SomeType实体同样带有@TenantId注解的tenant_id字段。问题在于，EntityManager查询时只会根据referencedColumnName（即name字段）获取子对象，不会自动加上tenant_id过滤条件，导致其他租户存在相同name值时返回多条记录，抛出异常。

我们可以修改数据库改为关联SomeType的主键，但更倾向于保留用name字段关联（方便手动查看表数据），希望找到无需大规模重构的解决方案，之前尝试过@Where注解和@FilterDef都无效。

1. 自定义SQL拦截器自动添加租户过滤 #

实现Hibernate的StatementInspector接口，在SQL执行前为关联表自动追加tenant_id过滤条件：

public class TenantStatementInspector implements StatementInspector {
    private final Supplier<String> tenantIdSupplier;

    public TenantStatementInspector(Supplier<String> tenantIdSupplier) {
        this.tenantIdSupplier = tenantIdSupplier;
    }

    @Override
    public String inspect(String sql) {
        String tenantId = tenantIdSupplier.get();
        if (tenantId == null || !sql.contains("JOIN some_type")) {
            return sql;
        }
        // 为some_type表追加租户过滤
        return sql.replace("JOIN some_type", "JOIN some_type WHERE some_type.tenant_id = '" + tenantId + "'");
    }
}

在创建EntityManagerFactory时配置该拦截器：

EntityManagerFactory emf = Persistence.createEntityManagerFactory("your-pu", Map.of(
        "hibernate.session_factory.statement_inspector", 
        new TenantStatementInspector(() -> getCurrentTenantId())
));

如果有多个类似关联表，可扩展拦截器的表名匹配逻辑。

2. 显式在HQL中添加租户过滤条件 #

查询主实体时，手动指定关联对象的租户匹配条件，示例HQL：

SELECT e FROM Entity e 
JOIN FETCH e.someType st 
WHERE e.tenant_id = :tenantId AND st.tenant_id = :tenantId

这种方式需要修改所有涉及该关联的查询，但过滤逻辑精准可控。

3. 结合@Where注解+参数自动注入 #

在SomeType实体上添加@Where注解：

@Entity
@Where(clause = "tenant_id = :tenantId")
public class SomeType {
    @TenantId
    private String tenantId;
    // 其他字段
}

之前尝试无效可能是因为未绑定参数，可通过拦截器自动为所有查询注入tenantId参数：

public class TenantParameterInterceptor implements EmptyInterceptor {
    @Override
    public void onPrepareStatement(String sql) {
        if (sql.contains(":tenantId")) {
            // 这里需要结合上下文获取当前租户ID，注入参数
            // 可通过ThreadLocal存储当前租户ID
            String tenantId = TenantContext.getTenantId();
            // 注意：实际实现需通过Hibernate的参数绑定API，此处为简化示例
        }
    }
}

配置拦截器到EntityManagerFactory即可自动注入参数。

4. 给name+tenant_id加唯一约束 #

虽然不能解决查询过滤问题，但可从数据库层面避免跨租户同name导致的查询冲突：

ALTER TABLE some_type ADD CONSTRAINT uk_some_type_name_tenant UNIQUE (name, tenant_id);

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