动态更改Spring数据源

Question

我有一个Spring应用程序，我想动态更改数据源，也就是说，当输入DS URL时，Spring Beans和所有依赖项将自动更新。我知道这有点奇怪，但无论如何我都想实现它。 我的Spring配置如下：

<bean id="majorDataSource" class="org.postgresql.ds.PGSimpleDataSource">
    <property name="serverName" value="${jdbc.serverName}" />
    <property name="portNumber" value="${jdbc.portNumber}" />
    <property name="user" value="${jdbc.username}" />
    <property name="password" value="${jdbc.password}" />
    <property name="databaseName" value="${jdbc.databaseName}" />
</bean>
<bean id="transactionManager" class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
    <property name="dataSource" ref="majorDataSource"/>
</bean>

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

<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
    <property name="dataSource" ref="majorDataSource"/>
    <property name="configLocation" value="classpath:sqlmap-config.xml"/>
</bean>

问题如下：

1. JDBC URL 存储在属性中，可能在运行时更改。

2. 一旦 URL 被更改，我需要重新创建数据源，以及可能的依赖对象。我无法找到一种优雅地在 Spring 中完成这个任务的方法？

我知道 Spring 可以根据一个键动态路由数据源，但是数据源 URL 在 Spring 中是预定义的，不能在运行时更改。这不符合我的情况。

- Simon Wang

Answer 1

您可以通过扩展Spring的AbstractRoutingDataSource并覆盖方法determineCurrentLookupKey()来使用它，该方法应返回引用要使用的数据源的Spring bean的键。

请查看这篇spring source博客文章，它将向您展示如何使用此功能的示例。

基本上，为了回答您的问题，您需要在XML配置中将两个数据源定义为不同的Spring bean。无需动态创建一个，Spring会加载两个，并根据您在determineCurrentLookupKey()方法中的条件动态使用其中之一。

这将导致类似于以下内容的结果： XML配置

<!-- first data source -->
<bean id="majorDataSource" class="org.postgresql.ds.PGSimpleDataSource">
    <property name="serverName" value="${jdbc.major.serverName}" />
    <property name="portNumber" value="${jdbc.major.portNumber}" />
    <property name="user" value="${jdbc.major.username}" />
    <property name="password" value="${jdbc.major.password}" />
    <property name="databaseName" value="${jdbc.major.databaseName}" />
</bean>
<!-- second data source -->
<bean id="minorDataSource" class="org.postgresql.ds.PGSimpleDataSource">
    <property name="serverName" value="${jdbc.minor.serverName}" />
    <property name="portNumber" value="${jdbc.minor.portNumber}" />
    <property name="user" value="${jdbc.minor.username}" />
    <property name="password" value="${jdbc.minor.password}" />
    <property name="databaseName" value="${jdbc.minor.databaseName}" />
</bean>
<!-- facade data source -->
<bean id="dataSource" class="blog.datasource.CustomerRoutingDataSource">
   <property name="targetDataSources">
      <map>
         <entry key="MINOR" value-ref="minorDataSource"/>
         <entry key="MAJOR" value-ref="majorDataSource"/>
      </map>
   </property>
   <property name="defaultTargetDataSource" ref="majorDataSource"/>
</bean>
<!-- wiring up -->
<bean id="transactionManager" class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
    <property name="dataSource" ref="dataSource"/>
</bean>
<tx:annotation-driven transaction-manager="transactionManager"/>
<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
    <property name="dataSource" ref="dataSource"/>
    <property name="configLocation" value="classpath:sqlmap-config.xml"/>
</bean>

Java

public class MyRoutingDataSource extends AbstractRoutingDataSource {
   @Override
   protected Object determineCurrentLookupKey() {
      // get the current url
      HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.currentRequestAttributes()).getRequest();
      if (request.getRequestURL().toString().endsWith("/minor"))
          return "MINOR";
      else
          return "MAJOR";
   }
}

- Alex

---

我使用主要和次要数据库进行故障转移。当主要数据库失败时，我会将次要数据库提升为主要数据库。然后，我需要设置第三个数据库作为备用，其URL将是动态的，无法在此预定义。不过，也许我可以强制使用静态IP地址来使用主要和次要数据库，这样我就不需要动态更改URL了。 - Simon Wang

你可以使用多租户策略。例如：http://docs.jboss.org/hibernate/orm/4.2/devguide/en-US/html/ch16.html - Ali.Mojtahed

Answer 2

我不确定这是否是正确的做法，但您可以尝试像这样做。

<bean id="majorDataSource" class="org.postgresql.ds.PGSimpleDataSource">
    <property name="serverName" value="dummydata" />
    <property name="portNumber" value="dummydata" />
    <property name="user" value="dummydata" />
    <property name="password" value="dummydata" />
    <property name="databaseName" value="dummydata" />
</bean>
<bean id="transactionManager" class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
    <property name="dataSource" ref="majorDataSource"/>
</bean>

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

而在Java类中

public class TestTransaction {

   @Autowired
   private DataSourceTransactionManager manager;

   private PlatformTransactionManager transactionManager;

   public testExecution(DataSource ds) {
       manager.setDataSource(ds);
       transactionManager = manager;
       TransactionDefinition def = new DefaultTransactionDefinition();
       TransactionStatus status = transactionManager.getTransaction(def);
       try {
           jdbcTemplate.update();
           transactionManager.commit(status);
       } catch (Exception ex) {
           transactionManager.rollback(status);
       }
   }
}

请建议这种方法是否可行，因为我对Spring还很陌生。

- Jashanpreet Singh