spring 配置多数据源-优快云博客

本文链接：https://blog.youkuaiyun.com/iteye_7030/article/details/81955713

Spring2.0.1以后的版本已经支持配置多数据源，并且可以在运行的时候动态加载不同的数据源。通过继承AbstractRoutingDataSource就可以实现多数据源的动态转换。目前做的项目就是需要访问12个数据源，每个数据源的表结构都是相同的，所以要求数据源的变动对于编码人员来说是透明，也就是说同样SQL语句在不同的环境下操作的数据库是不一样的。具体的配置如下：
一、首先需要写一个静态的键值对照类：
Java代码
1.package cn.com.xinli.ccp.dynamicds;
2.
3.public class DataSourceMap {
4.    public static final String Admin="Admin";
5.    public static final String Yxh = "Yxh";
6.}
package cn.com.xinli.ccp.dynamicds;

public class DataSourceMap {
public static final String Admin="Admin";
public static final String Yxh = "Yxh";
}

这个类主要在使用的时候当作获得数据源的标志使用。
二、建立一个获得和设置上下文的类：

Java代码
1.package cn.com.xinli.ccp.dynamicds;
2.
3.public class CustomerContextHolder {
4.    private static final ThreadLocal contextHolder =
5.        new ThreadLocal();
6.
7.    public static void setCustomerType(String customerType) {
8.      contextHolder.set(customerType);
9.    }
10.
11.    public static String getCustomerType() {
12.      return (String) contextHolder.get();
13.    }
14.
15.    public static void clearCustomerType() {
16.      contextHolder.remove();
17.    }
18.
19.
20.}
package cn.com.xinli.ccp.dynamicds;

public class CustomerContextHolder {
private static final ThreadLocal contextHolder =
        new ThreadLocal();

public static void setCustomerType(String customerType) {
   contextHolder.set(customerType);
}

public static String getCustomerType() {
   return (String) contextHolder.get();
}

public static void clearCustomerType() {
   contextHolder.remove();
}

}
这个主要负责设置上下文环境和获得上下文环境。
三、建立动态数据源类，这个类必须继承AbstractRoutingDataSource：

Java代码
1.package cn.com.xinli.ccp.dynamicds;
2.
3.import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
4.
5.public class DynamicDataSource extends AbstractRoutingDataSource {
6.
7.    protected Object determineCurrentLookupKey() {
8.        // TODO Auto-generated method stub
9.        return CustomerContextHolder.getCustomerType();
10.    }
11.
12.}
package cn.com.xinli.ccp.dynamicds;

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

public class DynamicDataSource extends AbstractRoutingDataSource {

protected Object determineCurrentLookupKey() {
// TODO Auto-generated method stub
return CustomerContextHolder.getCustomerType();
}

}
这个类实现了determineCurrentLookupKey方法，该方法返回一个Object，一般是返回字符串，也可以是枚举类型。该方法中直接使用了CustomerContextHolder.getCustomerType()方法获得上下文环境并直接返回。
四、编写spring的配置文件配置数据源

Java代码
1.<bean id="parentDataSource"
2.            class="org.springframework.jdbc.datasource.DriverManagerDataSource">
3.            <property name="driverClassName">
4.                <value>COM.ibm.db2.jdbc.net.DB2Driver</value>
5.            </property>
6.            <property name="url">
7.                <value>jdbc:db2:127.0.0.1:TEST</value>
8.            </property>
9.    </bean>
10.
11.    <bean id="adminDataSource" parent="parentDataSource">
12.        <property name="username" value="admin"/>
13.        <property name="password" value="master997mb"/>
14.    </bean>
15.
16.    <bean id="yxhDataSource" parent="parentDataSource">
17.        <property name="username" value="yxh"/>
18.        <property name="password" value="yxh"/>
19.    </bean>
<bean id="parentDataSource"
   class="org.springframework.jdbc.datasource.DriverManagerDataSource">
   <property name="driverClassName">
    <value>COM.ibm.db2.jdbc.net.DB2Driver</value>
   </property>
   <property name="url">
    <value>jdbc:db2:127.0.0.1:TEST</value>
   </property>
</bean>

<bean id="adminDataSource" parent="parentDataSource">
    <property name="username" value="admin"/>
    <property name="password" value="master997mb"/>
</bean>

<bean id="yxhDataSource" parent="parentDataSource">
    <property name="username" value="yxh"/>
    <property name="password" value="yxh"/>
</bean>
在这个配置中可以看到首先有个parentDataSource，这个主要配置一些数据源的公用信息，项目中都是链接DB2数据库；adminDataSource和yxhDataSource是根据不同需要配置的个性化信息，但都必须加parent属性，值为parentDataSource。这样就配置好了2个数据源信息。当然如果链接的多数据源是不同类型的两个数据库，那么parentDataSource就可以不要了，直接配置两个不同的数据源链接就可以了。
五、编写spring配置文件配置多数据源映射关系

Java代码
1.<bean id="dataSource" class="cn.com.xinli.ccp.dynamicds.DynamicDataSource">
2.       <property name="targetDataSources">
3.          <map key-type="java.lang.String">
4.             <entry key="Yxh" value-ref="yxhDataSource"/>
5.          </map>
6.       </property>
7.       <property name="defaultTargetDataSource" ref="adminDataSource"/>
8.    </bean>
<bean id="dataSource" class="cn.com.xinli.ccp.dynamicds.DynamicDataSource">
    <property name="targetDataSources">
       <map key-type="java.lang.String">
          <entry key="Yxh" value-ref="yxhDataSource"/>
       </map>
    </property>
    <property name="defaultTargetDataSource" ref="adminDataSource"/>
</bean>
在这个配置中第一个property属性配置目标数据源，<map key-type="java.lang.String">中的key-type必须要和静态键值对照类DataSourceMap中的值的类型相同；<entry key="Yxh" value-ref="yxhDataSource"/>中key的值必须要和静态键值对照类中的值相同，如果有多个值，可以配置多个<entry>标签。第二个property属性配置默认的数据源。
六、配置hibernate。
Hibernate的配置和普通的hibernate、spring结合的配置一样

Java代码
1.<bean id="sessionFactory"
2.        class="org.springframework.orm.hibernate3.LocalSessionFactoryBean">
3.        
4.        <property name="dataSource">
5.            <ref local="dataSource" />
6.        </property>
7.        <property name="mappingResources">
8.            <list>
9.                <value>
10.                    cn/com/xinli/ccp/entity/User.hbm.xml
11.                </value>
12.                <value>
13.                    cn/com/xinli/ccp/entity/Test.hbm.xml
14.                </value>
15.            </list>
16.        </property>
17.        <property name="hibernateProperties">
18.            <props>
19.                <prop key="hibernate.dialect">
20.                    org.hibernate.dialect.DB2Dialect
21.                </prop>
22.
23.                <prop key="hibernate.show_sql">true</prop>
24.                <prop key="hibernate.use_outer_join">true</prop>
25.                <prop key="hibernate.jdbc.batch_size">50</prop>
26.                <prop key="hibernate.jdbc.fetch_size">5</prop>
27.                <prop key="hibernate.connection.pool_size">2</prop>
28.                <prop key="hibernate.connection.autocommit">false</prop>
29.                <prop key="hibernate.cache.use_query_cache">false</prop>
30.                <prop key="hibernate.max_fetch_depth">1</prop>
31.                <prop key="hibernate.bytecode.use_reflection_optimizer">true</prop>
32.            </props>
33.        </property>
34.    </bean>
35.
36.<bean id="mydao" class="cn.com.xinli.ccp.dao.HibernateBaseDao">
37.        <property name="sessionFactory">
38.            <ref local="sessionFactory" />
39.        </property>
40.    </bean>
<bean id="sessionFactory"
  class="org.springframework.orm.hibernate3.LocalSessionFactoryBean">
  
  <property name="dataSource">
   <ref local="dataSource" />
  </property>
  <property name="mappingResources">
   <list>
    <value>
     cn/com/xinli/ccp/entity/User.hbm.xml
    </value>
    <value>
     cn/com/xinli/ccp/entity/Test.hbm.xml
    </value>
   </list>
  </property>
  <property name="hibernateProperties">
   <props>
    <prop key="hibernate.dialect">
     org.hibernate.dialect.DB2Dialect
    </prop>

    <prop key="hibernate.show_sql">true</prop>
    <prop key="hibernate.use_outer_join">true</prop>
    <prop key="hibernate.jdbc.batch_size">50</prop>
    <prop key="hibernate.jdbc.fetch_size">5</prop>
    <prop key="hibernate.connection.pool_size">2</prop>
    <prop key="hibernate.connection.autocommit">false</prop>
    <prop key="hibernate.cache.use_query_cache">false</prop>
    <prop key="hibernate.max_fetch_depth">1</prop>
    <prop key="hibernate.bytecode.use_reflection_optimizer">true</prop>
   </props>
  </property>
</bean>

<bean id="mydao" class="cn.com.xinli.ccp.dao.HibernateBaseDao">
  <property name="sessionFactory">
   <ref local="sessionFactory" />
  </property>
</bean>
关于dao的代码这里就省略了。
七、配置结束，可以使用了。

Java代码
1.public class DaoTest extends TestCase {
2.
3.    public void testSave() throws Exception{
4.        CustomerContextHolder.setCustomerType(DataSourceMap.Admin);//设置数据源
5.        //hibernate创建实体
6.        Test test = new Test();
7.        test.setTest("22222222");
8.
9.        mydao.save(test);//使用dao保存实体
10.
11.        CustomerContextHolder.setCustomerType(DataSourceMap.Yxh);//设置为另一个数据源
12.
13.        mydao.save(test);//使用dao保存实体到另一个库中
14.
15.    }
16.}
public class DaoTest extends TestCase {

public void testSave() throws Exception{
  CustomerContextHolder.setCustomerType(DataSourceMap.Admin);//设置数据源
  //hibernate创建实体
  Test test = new Test();
  test.setTest("22222222");

  mydao.save(test);//使用dao保存实体

  CustomerContextHolder.setCustomerType(DataSourceMap.Yxh);//设置为另一个数据源

  mydao.save(test);//使用dao保存实体到另一个库中

}
}在项目中对于编码人员对多数据源的切换可以做成透明的，操作同样的dao，就可以访问不同的数据库了。