具有多个数据源但只有一组存储库的 Spring Data + JPA

Question

我今天一直在研究这个，我开始认为我想做的事情可能不可能，所以我向你寻求帮助，哦，强大的 Stackoverflow。

我正在用 Java 构建一个 RESTful 服务平台，使用 Spring Data 3.1.2 + JPA 作为我的持久层（如文档中的 here 所述）。我的数据模型对象都实现为扩展 Spring JpaRepository 接口的接口。如本例所示，我已将所有东西都连接起来并与单个数据源很好地工作（请注意，显示的数据源是 Derby，但这仅用于开发目的；在生产中，我们将使用 Oracle）：

<jpa:repositories base-package="com.my.cool.package.repository"/>

<bean id="emf" class="org.springframework.orm.jpa.LocalContainerEntityManagerFactoryBean">
  <property name="dataSource" ref="dataSource" />
  <property name="loadTimeWeaver">
    <bean class="org.springframework.instrument.classloading.InstrumentationLoadTimeWeaver" />
  </property>
  <property name="packagesToScan" value="com.my.cool.package" />
  <property name="jpaVendorAdapter" ref="jpaVendorAdapter" />
</bean>

<bean name="transactionManager" class="org.springframework.orm.jpa.JpaTransactionManager">
  <property name="entityManagerFactory" ref="emf" />
</bean>
<tx:annotation-driven transaction-manager="transactionManager" />

<bean id="dataSource" class="org.springframework.jdbc.datasource.DriverManagerDataSource">
  <property name="driverClassName" value="org.apache.derby.jdbc.EmbeddedDriver" />
  <property name="url" value="jdbc:derby:derbyDB" />
  <property name="username" value="dev" />
  <property name="password" value="notARealPassword" />
</bean>

<bean id="jpaVendorAdapter" class="org.springframework.orm.jpa.vendor.HibernateJpaVendorAdapter">
  <property name="databasePlatform" value="org.hibernate.dialect.DerbyTenSevenDialect" />
</bean>

问题是此应用程序需要连接到多个 (Oracle) 数据库。每个传入请求中包含的凭据将包含一个字段，该字段告诉应用程序要访问哪个数据库才能满足该请求。每个数据库的模式都是相同的，因此不需要为每个数据库单独的存储库接口。

经过大量谷歌搜索后，很明显这是一种常见情况。也就是说：

• multiple databases with Spring Data JPA
• Spring + Hibernate + JPA + multiple databases
• how to setup spring data jpa with multiple datasources

这是（前？）Spring 开发人员的一篇博文，实际上与手头的主题无关，但有人在 cmets 中提出，作者回复了一些信息：

http://blog.springsource.org/2011/04/26/advanced-spring-data-jpa-specifications-and-querydsl/#comment-198835

似乎正在出现的主题是解决这个问题的方法是定义多个 EntityManagerFactories，并将每个连接到适当的存储库，如下所示：

<jpa:repositories base-package="com.my.cool.package.repository1" entity-manager-factory-ref="firstEntityManagerFactory" />
<jpa:repositories base-package="com.my.cool.package.repository2" entity-manager-factory-ref="secondEntityManagerFactory" />

但是，正如我所提到的，我想在所有数据源中重复使用我的存储库，所以这种方法似乎行不通。

我知道在我的代码中没有办法绕过从请求中获取相关信息的逻辑，并使用它来确定要使用的数据源（或 EntityManagerFactory）。我正在努力解决的部分是如何获取该数据源/EntityManagerFactory 的句柄并将其“注入”到我的存储库对象中。有什么想法吗？

Answer 1

如果您真的以多租户的方式使用不同的 DataSourcees（本质上是将请求分配给 DataSource 并在整个请求中坚持使用它），您应该看看 @987654323 @。它本质上提供了一种方法来保持DataSourcees 的Map 以及一个回调方法来返回一个用于查找最终使用的DataSource 的键。该方法的实现通常会查找一些线程绑定键并将其返回（或者甚至将其映射到 DataSource 映射键）。您只需首先确保某个 Web 组件将该键绑定到线程。

如果你有它，你的 Spring 配置只需为 AbstractRoutingDataSource 的子类设置一个 bean，并将 DataSources 的映射通过管道传输到它。您的 Spring Data JPA 设置保持默认方式。 EntityManagerFactoryBean 指的是 AbstractRoutingDataSource，而您只有一个 <jpa:repositories /> 元素。