Spring2连接多数据库

Spring2.0.1以后的版本已经支持配置多数据源，并且可以在运行的时候动态加载不同的数据源。通过继承 AbstractRoutingDataSource就可以实现多数据源的动态转换。目前做的项目就是需要访问2个数据源，每个数据源的表结构都是相同的，所以要求数据源的变动对于编码人员来说是透明，也就是说同样SQL语句在不同的环境下操作的数据库是不一样的。具体的流程如下：

一、建立一个获得和设置上下文的类

package com.lvye.base.dao.impl.jdbc;

/**连接哪个数据源的 环境变量*/

public class JdbcContextHolder {

       private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>(); 

       public static void setJdbcType(String jdbcType) { 

             contextHolder.set(jdbcType);  

       }  

       public static void setSlave(){

             setJdbcType("slave");

       }

       public static void setMaster(){

              clearJdbcType();

       }

       public static String getJdbcType() { 

            return (String) contextHolder.get();  

       }  

       public static void clearJdbcType() { 

            contextHolder.remove();  

       }  

}

二、建立动态数据源类，这个类必须继承AbstractRoutingDataSource

package com.lvye.base.dao.impl.jdbc;

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

public class DynamicDataSource extends AbstractRoutingDataSource{

       protected Object determineCurrentLookupKey() {

              return JdbcContextHolder.getJdbcType();

       }

}

这个类实现了determineCurrentLookupKey方法，该方法返回一个Object，一般是返回字符串。该方法中直接 使用了JdbcContextHolder.getJdbcType();方法获得上下文环境并直接返回。

三、编写spring的配置文件配置数据源

<beans>

       <bean id="master" class="com.mchange.v2.c3p0.ComboPooledDataSource"  destroy-method="close">

              <property name="driverClass"><value>com.mysql.jdbc.Driver</value></property>

              <property name="jdbcUrl">

                     <value>jdbc:mysql://192.168.18.143:3306/wenhq?useUnicode=true&amp;characterEncoding=utf-8</value>

              </property>

              <property name="user"><value>root</value></property>

              <property name="password"><value></value></property>

       </bean>

        <bean id="slave" class="com.mchange.v2.c3p0.ComboPooledDataSource" destroy-method="close">

              <property name="driverClass"><value>com.mysql.jdbc.Driver</value></property>

              <property name="jdbcUrl">

                     <value>jdbc:mysql://192.168.18.144:3306/ wenhq?useUnicode=true&amp;characterEncoding=utf-8</value>

              </property>

              <property name="user"><value>root</value></property>

              <property name="password"><value></value></property>

       </bean>

   <bean id="mySqlDataSource" class="com.lvye.base.dao.impl.jdbc.DynamicDataSource"> 

          <property name="targetDataSources"> 

             <map> 

                <entry key="slave" value-ref="slave"/> 

             </map> 

          </property> 

          <property name="defaultTargetDataSource" ref="master"/> 

   </bean> 

   <bean id="jdbcTemplate" class="org.springframework.jdbc.core.JdbcTemplate">

              <property name="dataSource" ref="mySqlDataSource" />

   </bean>

   <bean id="transactionManager" class="org.springframework.jdbc.datasource.DataSourceTransactionManager">

              <property name="dataSource" ref="mySqlDataSource" />

   </bean>

</beans>

在这个配置中可以看到首先配置两个真实的数据库连接，使用的msyql数据 库；master和slave是按照mysql配置的主从关系的数据库，数据会自动实时同步。mySqlDataSource会根据上下文选择不同的数据源。

在这个配置中第一个property属性配置目标数据源， <entry key="slave" value-ref=" slave "/>中key的值必须要和JdbcContextHolder类中设置的参数值相同，如果有多个值，可以配置多个< entry>标签。第二个property属性配置默认的数据源，我们一般默认为主数据库。

有些朋友喜欢使用hibernate，只需要把上面的jdbcTemplate替换为hibernate的就可以了。




四、多数据库连接配置完毕，简单测试

public void testSave() throws Exception{  

        dbcContextHolder.setSlave();//设置从数据源  

        Test test = new Test();  

        test.setTest("www.wenhq.com.cn");             

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

       JdbcContextHolder.setMaster();//设置主数据源

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

}   




五、实现读写分离，上面的测试通过了，现在就简单了

我的程序是使用jdbc实现的保存数据，只是使用了c3p0的数据库连接池而已。

把所有访问数据库的方法包装一下，统一调用。

把执行更新的sql发送到主数据库了

       public void execute(String sql) {

              JdbcContextHolder.setMaster();

              log.debug("execute-sql:" + sql);

              jdbcTemplate.execute(sql);

       }                

把查询的发送到从数据库，需要注意的是像LAST_INSERT_ID这类的查询需要特殊处理，必须发送到主数据库，建议增加专门的方法，用于获取自增长的主键。

       public List findObject(String queryString, Class clazz) {

              JdbcContextHolder.setSlave();

              log.debug("findObject-sql:" + queryString);

              List list = jdbcTemplate.queryForList(queryString);

              try {

                     list = StringBase.convertList(list, clazz);// 将List转化为List<clazz>

              } catch (Exception e) {

                     log.error("List convert List<Object> error:" + e);

              }

              return list;

       }










在网上找到的配置多数据源的方法。

1.扩展 org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource类

实现代码



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

public class DynamicDataSource extends AbstractRoutingDataSource {

    @Override
    protected Object determineCurrentLookupKey() {
        // TODO Auto-generated method stub
        return DbContextHolder.getDbType(); 
    }

}





2.实现一个线程安全的DbContextHolder类。



代码
public class DbContextHolder {
        private static final ThreadLocal contextHolder = new ThreadLocal();   
       
        public static void setDbType(String dbType)
        {   
            contextHolder.set(dbType);  
        }   
       
       
       
        public static String getDbType()
        {   
            String str=(String) contextHolder.get();
            if(null==str || "".equals(str))
                str="1";
            return str;  
        }   
       
        public static void clearDbType()
        {   
            contextHolder.remove();  
        } 
}





3.修改spring的配置文件。



代码
<!--动态选择数据源--> 
    <bean id="dataSource" class="tecamo.core.utils.DynamicDataSource"> 
        <property name="targetDataSources"> 
            <map key-type="java.lang.String"> 
                <entry key="1" value-ref="ds1"/> 
                <entry key="2" value-ref="ds2"/>
            </map> 
        </property> 
        <property name="defaultTargetDataSource" ref="ds1"/> 
    </bean> 
   
   
   
   
   
    <!-- 数据源定义 -->
    <bean id="ds1" class="org.apache.commons.dbcp.BasicDataSource" destroy-method="close">
        <property name="driverClassName" value="${jdbc.driverClassName}"/>
        <property name="url" value="${jdbc.url}"/>
        <property name="username" value="${jdbc.username}"/>
        <property name="password" value="${jdbc.password}"/>
    </bean>
   
    <bean id="ds2" class="org.apache.commons.dbcp.BasicDataSource" destroy-method="close">
        <property name="driverClassName" value="${jdbcOrcale.driverClassName}"/>
        <property name="url" value="${jdbcOrcale.url}"/>
        <property name="username" value="${jdbcOrcale.username}"/>
        <property name="password" value="${jdbcOrcale.password}"/>
    </bean>






4.切换数据源方法。

在serverice方法之前调用

DbContextHolder.setDbType("2");表示使用第二个数据源。