跨多个租户查询 tables（相同的 table 名称）

Question

我有一个系统，其中租户数量未知（同一数据库服务器上的不同数据库实例）。我有用户登录并选择正确租户的工作代码，我可以读取该租户的配置 table。

我希望应用程序在启动时循环遍历所有租户，读取配置并对其执行操作。在迁移到 Spring Data JPA（由 hibernate 支持）之前，这很容易，因为我分别连接到每个数据库实例。

我认为我不能使用 Spring 的 @Transactional，因为它只建立一个连接。

我希望对同一个 bean 使用同一个存储库接口，因为这在我一次只需要访问一个租户时有效。

我确实有一个 class MultiTenantConnectionProviderImpl extends AbstractDataSourceBasedMultiTenantConnectionProviderImpl 可以为我提供给定租户的数据源，但我不确定如何在 @Service class 的方法中使用它？

Answer 1

我认为我接近于一种解决方案，但我对它不是很满意。我很乐意提出一个更好的答案。

已编辑：事实证明这不太有效，因为 Spring 或 Hibernate 似乎只调用一次当前租户标识符解析器，而不是每次调用 @Transactional 方法时都调用

它涉及更改 CurrentTenantIdentifierResolver 实现，不仅查看当前用户（如果已设置）以获取他们当前的租户 ID（由实现者决定如何设置）...它还需要查看线程局部变量以查看是否已设置覆盖。

使用这种方法，我可以临时设置 tenantID...使用指定的多租户事务管理器调用服务方法，然后获取数据。

我的测试服务：

@Service
public class TestService
{
    @Transactional(transactionManager = "sharedTxMgr")
    public void doSomeGets()
    {
        List<String> tenants = getListSomehow();
        try(MultitenancyTemporaryOverride tempOverride = new MultitenancyTemporaryOverride())
        {
            for(String tenant : tenants)
            {
                tempOverride.setCurrentTenant(tenant);
                doTenantSpecificWork();
            }
        }
        catch (Exception e)
        {
            logger.error(e);
        }
    }

    @Transactional(transactionManager = "tenantSpecificTxMgr")
    public void doTenantSpecificWork()
    {
        //do some work here, which only applies to the tenant
    }
}

我的 class 包装设置 ThreadLocal，实现 AutoCloseable 以帮助确保清理变量

public class MultitenancyTemporaryOverride implements AutoCloseable
{
    static final ThreadLocal<String> tenantOverride = new ThreadLocal<>();

    public void setCurrentTenant(String tenantId)
    {
        tenantOverride.set(tenantId);
    }

    public String getCurrentTenant()
    {
        return tenantOverride.get();
    }

    @Override
    public void close() throws Exception
    {
        tenantOverride.remove();
    }

}

我使用本地线程的租户解析器实现

public class CurrentTenantIdentifierResolverImpl implements CurrentTenantIdentifierResolver
{

    @Override
    public String resolveCurrentTenantIdentifier()
    {
        Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
        logger.debug(ToStringBuilder.reflectionToString(authentication));
        String database = "shared";
        if (authentication != null && authentication.getPrincipal() instanceof MyUser)
        {
            MyUser user = (MyUser) authentication.getPrincipal();
            database = user.getTenantId();
        }
        String temporaryOverride = MultitenancyTemporaryOverride.tenantOverride.get();
        if(temporaryOverride != null)
        {
            database = temporaryOverride;
        }
        return database;
    }

Answer 2

我不确定我是否应该删除我以前的答案，编辑它或什么。因此，如果 MOD 可以让我知道正确的程序，我将很乐意遵守。

事实证明，我对 @Transactional 无法正常工作的看法是正确的。我最终使用和 AbstractRoutingDataSource 的自定义实现来替换我的 MultiTenantConnectionProviderImpl 和 CurrentTenantResolverImpl。我使用这个新数据源而不是设置 hibernate.multiTenancy hibernate.multi_tenant_connection_provider 和 hibernate.tenant_identifier_resolver

我的临时覆盖 class 看起来像这样：

public class MultitenancyTemporaryOverride implements AutoCloseable
{    
    static final ThreadLocal<String> tenantOverride = new NamedThreadLocal<>("temporaryTenantOverride");

    public void setCurrentTenant(String tenantId)
    {
        tenantOverride.set(tenantId);
    }

    public String getCurrentTenant()
    {
        return tenantOverride.get();
    }

    @Override
    public void close() throws Exception
    {
        tenantOverride.remove();
    }
}

我的 TenantRoutingDataSource 如下所示：

@Component
public class TenantRoutingDataSource extends AbstractDataSource implements InitializingBean
{

    @Override
    public Connection getConnection() throws SQLException
    {
        return determineTargetDataSource().getConnection();
    }

    @Override
    public Connection getConnection(String username, String password) throws SQLException
    {
        return determineTargetDataSource().getConnection(username, password);
    }

    @Override
    public void afterPropertiesSet() throws Exception
    {
    }

    protected String determineCurrentLookupKey()
    {
        Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
        String database = "shared";
        if (authentication != null && authentication.getPrincipal() instanceof MyUser)
        {
            MyUser user = (MyUser) authentication.getPrincipal();
            database = user.getTenantId();
        }
        String temporaryOverride = MultitenancyTemporaryOverride.tenantOverride.get();
        if (temporaryOverride != null)
        {
            database = temporaryOverride;
        }
        return database;
    }

    protected DataSource determineTargetDataSource()
    {
        return selectDataSource(determineCurrentLookupKey());
    }

    public DataSource selectDataSource(String tenantIdentifier)
    {
        //I use C3P0 for my connection pool
        PooledDataSource pds = C3P0Registry.pooledDataSourceByName(tenantIdentifier);
        if (pds == null)
            pds = getComboPooledDataSource(tenantIdentifier);
        return pds;
    }

    private ComboPooledDataSource getComboPooledDataSource(String tenantIdentifier)
    {
        ComboPooledDataSource cpds = new ComboPooledDataSource(tenantIdentifier);
        cpds.setJdbcUrl("A JDBC STRING HERE");
        cpds.setUser("MyDbUsername");
        cpds.setPassword("MyDbPassword");
        cpds.setInitialPoolSize(10);
        cpds.setMaxConnectionAge(10000);
        try
        {
            cpds.setDriverClass("com.informix.jdbc.IfxDriver");
        }
        catch (PropertyVetoException e)
        {
            throw new RuntimeException("Weird error when setting the driver class", e);
        }
        return cpds;
    }
}

然后我在创建实体管理器工厂 bean 时将我的自定义数据源提供给它。

@Service
public class TestService
{
    public void doSomeGets()
    {
        List<String> tenants = getListSomehow();
        try(MultitenancyTemporaryOverride tempOverride = new MultitenancyTemporaryOverride())
        {
            for(String tenant : tenants)
            {
                tempOverride.setCurrentTenant(tenant);
                //do some work here, which only applies to the tenant
            }
        }
        catch (Exception e)
        {
            logger.error(e);
        }
    }
}

跨多个租户查询 tables（相同的 table 名称）

Query tables across multiple tenants (same table name)

java

hibernate

multi-tenant

spring-data-jpa