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

Question

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

我正在用 Java 构建一个 RESTful 服务平台，使用 Spring Data 3.1.2 + JPA 作为我的持久层(如文档所述 here )。我的数据模型对象都实现为扩展 Spring JpaRepository 接口(interface)的接口(interface)。如本例所示，我已将所有内容都连接起来并与单个数据源很好地工作(请注意，显示的数据源是 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) 数据库。每个传入请求中包含的凭据将包含一个字段，该字段告诉应用程序要转到哪个数据库才能满足该请求。每个数据库的模式都是相同的，因此不需要为每个数据库单独的存储库接口(interface)。

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

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

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

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 的句柄并将其“注入(inject)”到我的存储库对象中。有什么想法吗？

Answer 1

如果您真的使用不同的 DataSource以 Multi-Tenancy 的方式(本质上将请求分配给 DataSource 并在整个请求中坚持使用它)你应该看看 AbstractRoutingDataSource .它本质上提供了一种保存 Map 的方法。的 DataSource es 以及返回用于查找 DataSource 的键的回调方法最终被使用。该方法的实现通常会查找一些线程绑定(bind)的键并返回(或者甚至将其映射到 DataSource 映射键)。您只需首先确保某个 Web 组件将该键绑定(bind)到线程。

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