使用带有 Hibernate 和 Spring 的单数据库多模式的多租户引导将数据保存到错误的模式

Question

我正在尝试使用数据为我的多租户（单个数据库、多个模式）系统播种，但遇到了一个问题，当我对单个数据库使用相同的代码时，该问题不存在。 我完全期望在我的研究过程中我错过了一些明显的东西。

每个模式都将包含完全相同的表结构。

这是我的租户上下文

public class TenantContext {

    public static final String DEFAULT_TENANT_IDENTIFIER = "public";

    private static final ThreadLocal<String> TENANT_IDENTIFIER = new ThreadLocal<>();

    public static void setTenant(String tenantIdentifier) {
        TENANT_IDENTIFIER.set(tenantIdentifier);
    }

    public static void reset(String tenantIdentifier) {
        TENANT_IDENTIFIER.remove();
    }

    @Component
    public static class TenantIdentifierResolver implements CurrentTenantIdentifierResolver {
        @Override
        public String resolveCurrentTenantIdentifier() {
            String currentTenantId = TENANT_IDENTIFIER.get();
            return currentTenantId != null ?
                    currentTenantId :
                    DEFAULT_TENANT_IDENTIFIER;
        }

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

还有我的 HibernateConfig

@Configuration
public class HibernateConfig {

    @Autowired
    private JpaProperties jpaProperties;

    @Bean
    public JpaVendorAdapter jpaVendorAdapter() {
        return new HibernateJpaVendorAdapter();
    }

    @Bean
    public LocalContainerEntityManagerFactoryBean entityManagerFactory(DataSource dataSource,
             MultiTenantConnectionProvider multiTenantConnectionProvider, CurrentTenantIdentifierResolver currentTenantIdentifierResolver) {

        Map<String, Object> jpaPropertiesMap = new HashMap<>();
        jpaPropertiesMap.putAll(jpaProperties.getProperties());
        jpaPropertiesMap.put(Environment.MULTI_TENANT, MultiTenancyStrategy.SCHEMA);
        jpaPropertiesMap.put(Environment.MULTI_TENANT_CONNECTION_PROVIDER, multiTenantConnectionProvider);
        jpaPropertiesMap.put(Environment.MULTI_TENANT_IDENTIFIER_RESOLVER, TenantContext.TenantIdentifierResolver.class);

        LocalContainerEntityManagerFactoryBean entityManagerFactoryBean = new LocalContainerEntityManagerFactoryBean();
        entityManagerFactoryBean.setDataSource(dataSource);
        entityManagerFactoryBean.setPackagesToScan(UppStudentAppBeApplication.class.getPackage().getName());
        entityManagerFactoryBean.setJpaVendorAdapter(jpaVendorAdapter());
        entityManagerFactoryBean.setJpaPropertyMap(jpaPropertiesMap);

        return entityManagerFactoryBean;
    }
}

还有我的 TenantConenctionProvider

@Component
public class TenantConnectionProvider implements MultiTenantConnectionProvider {

    private static Logger logger = LoggerFactory.getLogger(TenantConnectionProvider.class);

    @Autowired
    private DataSource dataSource;

    public TenantConnectionProvider(DataSource dataSource) {
        this.dataSource = dataSource;
    }

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

    @Override
    public void releaseAnyConnection(Connection connection) throws SQLException {
        connection.close();
    }

    @Override
    public Connection getConnection(String tenantIdentifier) throws SQLException {
        logger.info("Get connection for tenant  " + String.join(":", tenantIdentifier ));
        final Connection connection = getAnyConnection();
        try {
            //connection.createStatement().execute( String.format("SET SCHEMA \"%s\";", tenantIdentifier));
            connection.setSchema(tenantIdentifier);
        } catch ( SQLException e ) {
            throw new HibernateException(
                    "Could not alter JDBC connection to specified schema [" +
                            tenantIdentifier + "]",
                    e
            );
        }
        return connection;
    }

    @Override
    public void releaseConnection(String tenantIdentifier, Connection connection) throws SQLException {
        try {
            //connection.createStatement().execute( String.format("SET SCHEMA \"%s\";", TenantContext.DEFAULT_TENANT_IDENTIFIER) );
            connection.setSchema(TenantContext.DEFAULT_TENANT_IDENTIFIER);
        } catch ( SQLException e ) {
            throw new HibernateException(
                    "Could not alter JDBC connection to specified schema [" +
                            tenantIdentifier + "]",
                    e
            );
        }
        releaseAnyConnection(connection);
    }

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

    @Override
    public boolean isUnwrappableAs(Class unwrapType) {
        return false;
    }

    @Override
    public <T> T unwrap(Class<T> unwrapType) {
        return null;
    }
}

我将我的种子称为 class，它使用 flyway 迁移构建了我的租户和模式。

然后我尝试遍历保存的租户切换 TenantContext。 当调试似乎起作用时。 但是，当我尝试对 repo 做任何事情时，我收到以下错误。

ohengine.jdbc.spi.SqlExceptionHelper: 错误: 列 Campus0_.createdat 不存在
提示：也许您的意思是引用列“campus0_.created_at”。
Position：32

正如我之前所说，当它是单个数据库和模式时，它以前工作得很好。 我不是 100% 确定我哪里出错了。 我应该如何注册模式？ 如果是这样，我如何在不重新部署的情况下加入新租户？ 我应该在这个阶段使用使用 repo 中的模式的自定义查询吗？

提前感谢您的任何帮助或建议。

编辑 1所以我现在通过检查 hibernate 属性克服了我最初的障碍，因此通过如下更改 hibernate 配置

@Bean
public LocalContainerEntityManagerFactoryBean entityManagerFactory(DataSource dataSource,
             MultiTenantConnectionProvider multiTenantConnectionProvider,
             HibernateProperties hibernateProperties) {

        Map<String, Object> jpaPropertiesMap = hibernateProperties.determineHibernateProperties(jpaProperties.getProperties(), new HibernateSettings());
        //jpaPropertiesMap.putAll(jpaProperties.getProperties());
        jpaPropertiesMap.put(Environment.MULTI_TENANT, MultiTenancyStrategy.SCHEMA);
        jpaPropertiesMap.put(Environment.MULTI_TENANT_CONNECTION_PROVIDER, multiTenantConnectionProvider);
        jpaPropertiesMap.put(Environment.MULTI_TENANT_IDENTIFIER_RESOLVER, TenantContext.TenantIdentifierResolver.class);

        LocalContainerEntityManagerFactoryBean entityManagerFactoryBean = new LocalContainerEntityManagerFactoryBean();
        entityManagerFactoryBean.setDataSource(dataSource);
        entityManagerFactoryBean.setPackagesToScan(UppStudentAppBeApplication.class.getPackage().getName());
        entityManagerFactoryBean.setJpaVendorAdapter(jpaVendorAdapter());
        entityManagerFactoryBean.setJpaPropertyMap(jpaPropertiesMap);

        return entityManagerFactoryBean;
}

这现在已经消除了上述命名错误。 但是现在它保存到我的默认架构而不是 TenantIdentifierResolver 中设置的架构。

Answer 1

你实现AsyncHandlerInterceptor - Spring 的拦截器。 也应该在WebMvcConfigurer中注册。

@Component
public class TenantRequestInterceptor implements AsyncHandlerInterceptor{

private SecurityDomain securityDomain;

public TenantRequestInterceptor(SecurityDomain securityDomain) {
    this.securityDomain = securityDomain;
}

 @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) {
     return Optional.ofNullable(request)
             .map(req -> securityDomain.getTenantIdFromJwt(req))
             .map(tenant -> setTenantContext(tenant))
             .orElse(false);
    }

    @Override
    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) {
        TenantContext.reset();
    }
    
    private boolean setTenantContext(String tenant) {
        TenantContext.setCurrentTenant(tenant);
        return true;
    }
}

这很重要，因为在这里您使用租户填充 TenantContext。 您是否调试过方法getConnection(String tenantIdentifier)作为tenantIdentifier 的值是什么？