java - Hibernate Multi-Tenancy : change tenant in session

Question

我们正在为多个消费者开发 SaaS 解决方案。该解决方案基于 Spring、Wicket 和 Hibernate。我们的数据库包含来自多个客户的数据。我们决定按如下方式对数据库建模:

• 公开
  所有客户之间的共享数据，例如用户帐户，因为我们不知道用户属于哪个客户
• 客户_1
• 客户_2
• ...

为了使用此设置，我们使用具有以下 TenantIdentifierResolver 的 Multi-Tenancy 设置:

public class TenantProviderImpl implements CurrentTenantIdentifierResolver {
    private static final ThreadLocal<String> tenant = new ThreadLocal<>();

    public static void setTenant(String tenant){
        TenantProviderImpl.tenant.set(tenant);
    }

    @Override
    public String resolveCurrentTenantIdentifier() {
        return tenant.get();
    }

    @Override
    public boolean validateExistingCurrentSessions() {
        return false;
    }

    /**
     * Initialize a tenant by storing the tenant identifier in both the HTTP session and the ThreadLocal
     *
     * @param   String  tenant  Tenant identifier to be stored
     */
    public static void initTenant(String tenant) {
        HttpServletRequest req = ((ServletWebRequest) RequestCycle.get().getRequest()).getContainerRequest();
        req.getSession().setAttribute("tenant", tenant);
        TenantProviderImpl.setTenant(tenant);
    }
}

initTenant 方法由 servlet 过滤器为每个请求调用。此过滤器在连接之前处理对数据库开放。

我们还实现了一个 AbstractDataSourceBasedMultiTenantConnectionProviderImpl，它被设置为我们的hibernate.multi_tenant_connection_provider。它在每个请求之前发出一个 SET search_path 查询。对于通过上述 servlet 过滤器的请求，这就像魅力一样。

现在是我们真正的问题:我们的应用程序有一些入口点没有通过 servlet 过滤器，例如一些 SOAP 端点。也有执行的定时作业没有通过 servlet 过滤器。这证明是一个问题。

作业/端点以某种方式接收一个值，该值可用于识别应与哪个客户相关联作业/端点请求。此唯一值通常映射到我们的 public 数据库架构中。因此，我们需要查询在我们知道哪个客户关联之前。因此，Spring 初始化了一个完整的 Hibernate session 。这 session 具有我们的默认租户 ID，未映射到特定客户。然而，在我们解决了独特的问题之后我们希望 session 更改租户标识符的客户值(value)。虽然这似乎不受支持，但那里不是 HibernateSession.setTenantIdentifier(String) 这样的东西，而有一个SharedSessionContract.getTenantIdentifier()。

我们认为我们有以下方法的解决方案:

org.hibernate.SessionFactory sessionFactory = getSessionFactory();
org.hibernate.Session session = null;
try
{
    session = getSession();
    if (session != null)
    {
       if(session.isDirty())
       {
          session.flush();
       }
       if(!session.getTransaction().wasCommitted())
       {
          session.getTransaction().commit();
       }

       session.disconnect();
       session.close();
       TransactionSynchronizationManager.unbindResource(sessionFactory);
    }
}
catch (HibernateException e)
{
    // NO-OP, apparently there was no session yet
}
TenantProviderImpl.setTenant(tenant);
session = sessionFactory.openSession();
TransactionSynchronizationManager.bindResource(sessionFactory, new SessionHolder(session));
return session;

然而，此方法似乎在作业/端点的上下文中不起作用并导致 HibernateException，例如“ session 结束!”或“交易未成功开始”。

我们有点迷茫，因为我们已经尝试寻找解决方案很长一段时间了。是不是我们误会了什么？我们误解了什么？我们如何解决上述问题？

回顾:HibernateSession-s 不是由用户请求创建的，而是由定时作业等创建的，不会传递我们的 servlet过滤器，因此在 Hibernate session 启动之前没有关联的租户标识符。他们有独特的值(value)观我们可以通过查询数据库将其转换为租户标识符。我们怎样才能告诉一个现有的 Hibernate session 更改它的租户标识符，从而发出新的 SET search_path 语句？

Answer 1

我们从未找到解决此问题的真正方法，但是 chimmi链接到 Jira-ticket 是其他人要求这样的功能:https://hibernate.atlassian.net/browse/HHH-9766

根据这张票，我们想要的行为目前不受支持。不过，我们找到了一种解决方法，因为我们实际想要使用此功能的次数有限，因此我们可以使用默认的 Java 并发实现在单独的线程中运行这些操作。

通过在单独的线程中运行操作，将创建一个新 session (因为 session 是线程绑定(bind)的)。将租户设置为跨线程共享的变量对我们来说非常重要。为此，我们在 CurrentTenantIdentifierResolver 中有一个静态变量。

为了在单独的线程中运行一个操作，我们实现了一个Callable。这些可调用对象作为具有 prototype 作用域的 Spring-beans 实现，因此每次请求( Autowiring )时都会创建一个新实例。我们已经实现了我们自己的 Callable 抽象实现，它最终确定了 Callable 接口(interface)定义的 call() 方法，并且实现开始一个新的 HibernateSession。代码看起来有点像这样:

public abstract class OurCallable<TYPE> implements Callable<TYPE> {
    private final String tenantId;

    @Autowired
    private SessionFactory sessionFactory;

    // More fields here

    public OurCallable(String tenantId) {
        this.tenantId = tenantId;
    }

    @Override
    public final TYPE call() throws Exception {
        TenantProvider.setTenant(tenantId);
        startSession();

        try {
            return callInternal();
        } finally {
            stopSession();
        }
    }

    protected abstract TYPE callInternal();

    private void startSession(){
        // Implementation skipped for clarity
    }

    private void stopSession(){
        // Implementation skipped for clarity
    }
}