如何使用具有相同存储库的多个数据源类型?

时间:2015-10-12 15:49:15

标签: java hibernate jpa spring-boot

我创建了一个多租户逻辑,我将租户号码存储在ThreadLocal中。当数据库具有相同类型(方言)时,一切都很好。在Spring启动初始化期间,JPA存储库使用第一个(和默认)数据源进行初始化(其SQL查询)。

但是,当我配置了Oracle和Postgres数据库时,spring boot仍会尝试使用第一个租户数据源的方言进行查询,这是一个非常大的问题,因为数据库类型的方言不同。

这将创建租户类型的dataSource。如果租户尚未知晓(在启动期间),则需要租户-1。

@Bean
public DataSource dataSource() throws SQLException {
    String tenant1 = "tenant-1";
    String tenant2 = "tenant-2";

    CustomRoutingDataSource customRoutingDataSource = new CustomRoutingDataSource();

    final DataSource tenant1DataSource = createDataSourceForTenant(tenant1);
    final DataSource tenant2DataSource = createDataSourceForTenant(tenant2);

    HashMap<Object, Object> targetDataSources = new HashMap<>();
    targetDataSources.put(tenant1.toUpperCase(), tenant1DataSource);
    targetDataSources.put(tenant2.toUpperCase(), tenant2DataSource);

    customRoutingDataSource.setTargetDataSources(targetDataSources);
    return customRoutingDataSource;
}

这将静态创建数据源 - 稍后当我解决问题时,将被JNDI源的动态提取所取代:

private DataSource createDataSourceForTenant(String tenant) throws SQLException {
    final DriverManagerDataSource dataSource = new DriverManagerDataSource();

    final LocalContainerEntityManagerFactoryBean factoryBean = new LocalContainerEntityManagerFactoryBean();
    factoryBean.setDataSource(dataSource);
    factoryBean.setPackagesToScan(new String[]{"entities"});

    if (tenant.equalsIgnoreCase("tenant-1")) {
        dataSource.setDriverClassName("org.postgresql.Driver");
        dataSource.setUrl("jdbc:postgresql://localhost:5433/importservices");
        dataSource.setUsername("test");
        dataSource.setPassword("test");

        factoryBean.setJpaVendorAdapter(new HibernateJpaVendorAdapter());
        factoryBean.afterPropertiesSet();
    }

    if (tenant.equalsIgnoreCase("tenant-2")) {
        dataSource.setDriverClassName("org.hibernate.dialect.Oracle10gDialect");
        dataSource.setUrl("jdbc:oracle:thin:@//oracle11:1521/deviso");
        dataSource.setUsername("test");
        dataSource.setPassword("test");

        factoryBean.setJpaVendorAdapter(new HibernateJpaVendorAdapter());
        factoryBean.afterPropertiesSet();
    }

    factoryBean.destroy();
    return dataSource;
}

自定义数据库路由。当租户不可用时(启动期间),getTenant返回默认值(1)。

public class CustomRoutingDataSource extends AbstractRoutingDataSource {

    @Override
    protected Object determineCurrentLookupKey() {
        return "TENANT-"+TenantThreadLocal.getTenant();
    }

}

现在,如果我有默认租户= 1,则Spring启动时会初始化并为实体创建特定于PostgreSQL的查询。如果我在运行时作为租户2(Oracle方言)访问它,我得到:

java.sql.SQLSyntaxErrorException: ORA-00933: SQL command not properly ended

因为Spring JPA用PostgreSQL方言初始化了JPA存储库。所以可能性是:

1)每个租户有多个EntityManagers - 不太舒服,因为租户是动态管理的 2) hibernate中的多租户支持 - 实际上我无法实现这一点,因为jadira框架在多租户支持方面存在一些问题,并且在2周之前仍未修复并等待{{3} } 3) ??

1 个答案:

答案 0 :(得分:0)

(刚刚发布的)Jadira 5.0.0.GA中的更改与多租户部署共存。免责声明:我维持Jadira。