Spring 动态设置数据源

Spring 动态设置数据源
Spring数据结构HibernateDB2DAO.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.}   
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.        <!-- to override, use the "SpringDatasourceConfig" snippet in your project -->     
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">  
        <!-- to override, use the "SpringDatasourceConfig" snippet in your project -->  
        <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，就可以访问不同的数据库
补充：使用filter设置数据源

Java代码 
1.public void doFilter(ServletRequest request, ServletResponse response,FilterChain chain) throws IOException, ServletException       
2.    {       
3.       HttpServletRequest httpRequest = (HttpServletRequest) request;       
4.       Object dataSourceName=httpRequest.getSession().getAttribute("currentDataSource");       
5.           if(dataSourceName!=null&&!"".equals(dataSourceName))       
6.           {       
7.              String name = dataSourceName.toString();       
8.              //System.out.println("使数据源"+name);  
9.              CustomerContextHolder.setCustomerType(name);  
10.           }       
11.           else       
12.           {    
13.               CustomerContextHolder.setCustomerType("publicDataSource");  
14.              //System.out.println("使用默认数据源");       
15.           }       
16.       chain.doFilter(request, response);  
17.       CustomerContextHolder.clearCustomerType();  
18.    }