Spring-具有多租户 Hibernate 的数据 JPA

Question

我正在尝试让 Spring-Data JPA 与带有自定义 MultiTenantConnectionProvider 的 Hibernate 一起工作。

我下面的配置中的所有内容似乎都有效。每次我尝试调用 Repository 方法时，我的 MultiTenantConnectionProviderImpl class 都会被调用。

主要问题是无法提供租户标识符。 Spring-Data 提供的 Repository 接口负责获取 Hibernate Session。

有什么方法可以提供 Spring-Data 租户标识符吗？或者我们可以在某个地方拦截 Hibernate Session 的创建，以便我们可以适当地调用 sessionFactory.withOptions().tenantIdentifier(itendintifier).openSession();

这是我的 Spring 配置 XML 文件。我尽量保持简单。

<?xml version="1.0" encoding="UTF-8"?>
<beans xmlns="http://www.springframework.org/schema/beans"
       xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
       xmlns:context="http://www.springframework.org/schema/context"
       xmlns:tx="http://www.springframework.org/schema/tx" xmlns:jpa="http://www.springframework.org/schema/data/jpa"
       xsi:schemaLocation="http://www.springframework.org/schema/beans
       http://www.springframework.org/schema/beans/spring-beans.xsd
       http://www.springframework.org/schema/context
       http://www.springframework.org/schema/context/spring-context.xsd
       http://www.springframework.org/schema/tx
       http://www.springframework.org/schema/tx/spring-tx-3.2.xsd http://www.springframework.org/schema/data/jpa http://www.springframework.org/schema/data/jpa/spring-jpa.xsd">

    <context:annotation-config/>
    <context:component-scan base-package="com.company"/>
    <jpa:repositories base-package="com.company.repositories"/>
    <bean id="entityManagerFactory" class="org.springframework.orm.jpa.LocalContainerEntityManagerFactoryBean">
        <property name="packagesToScan" value="com.company.entities"/>
        <property name="dataSource" ref="dataSource"/>
        <property name="jpaVendorAdapter">
            <bean class="org.springframework.orm.jpa.vendor.HibernateJpaVendorAdapter">
                <property name="showSql" value="true"/>
            </bean>
        </property>
    </bean>


    <bean id="transactionManager" class="org.springframework.orm.hibernate4.HibernateTransactionManager">
        <property name="sessionFactory" ref="sessionFactory"/>
    </bean>

    <bean id="sessionFactory" class="org.springframework.orm.hibernate4.LocalSessionFactoryBean">
        <property name="hibernateProperties">
            <props>
                <prop key="hibernate.dialect">org.hibernate.dialect.SQLServer2012Dialect</prop>
                <prop key="hibernate.format_sql">true</prop>
                <prop key="hibernate.multi_tenant_connection_provider">com.company.hibernate.MultiTenantConnectionProviderImpl</prop>
                <prop key="hibernate.multiTenancy">DATABASE</prop>
            </props>
        </property>
    </bean>


    <tx:annotation-driven transaction-manager="transactionManager"/>

    <!--Vendor specific properties here-->
    <bean id="dataSource" class="org.springframework.jdbc.datasource.DriverManagerDataSource">
        <property name="driverClassName" value="net.sourceforge.jtds.jdbc.Driver"/>
        <property name="url" value="jdbc:jtds:sqlserver://localhost:1433/myDatabase"/>
        <property name="username" value="username"/>
        <property name="password" value="password"/>
    </bean>

</beans>

Answer 1

使用CurrentTenantIdentifierResolver:

<bean id="sessionFactory" class="org.springframework.orm.hibernate4.LocalSessionFactoryBean">
    <property name="hibernateProperties">
        <map>
            <entry key="hibernate.dialect" value="org.hibernate.dialect.SQLServer2012Dialect">
            <entry key="hibernate.format_sql" value="true">
            <entry key="hibernate.multi_tenant_connection_provider" value="com.company.hibernate.MultiTenantConnectionProviderImpl">
            <entry key="hibernate.multiTenancy" value="DATABASE">
            <!-- tenant resolver as spring bean -->
            <entry key="hibernate.tenant_identifier_resolver" value-ref="currentTenantIdentifierResolver"/>
        </map>
    </property>
</bean>

<bean id="currentTenantIdentifierResolver"
    class="com.xxx.CurrentTenantResolver">
</bean>

简单的租户标识符解析器如下所示：

public class CurrentTenantResolver implements CurrentTenantIdentifierResolver {

    public String resolveCurrentTenantIdentifier() {
        // retrieve tenant from logged in user
        User usr = (User) SecurityContextHolder.getContext().getAuthentication().getPrincipal() ;       
        return usr.getTenantName();

    }

    public boolean validateExistingCurrentSessions() {
        return true;
    }

}

记住上面的 class 是一个 spring bean，所以你可以像普通的 spring bean 一样自动装配任何 spring bean(service/dao)。

每次 spring 需要会话时，hibernate 将从该 bean 中检索租户标识符。

Spring-具有多租户 Hibernate 的数据 JPA

Spring-Data JPA with Multi-Tenancy Hibernate

hibernate

spring-data

spring-data-jpa

multi-tenant

discriminator