Hibernate 與 Spring 多數據源的配置

Hibernate 與 Spring 多數據源的配置 
Spring2.0.1以後的版本已經支持配置多數據源，並且可以在運行的時候動態加載不同的數據源。通過繼承AbstractRoutingDataSource就可以實現多數據源的動態轉換。目前做的項目就是需要訪問12個數據源，每個數據源的表結構都是相同的，所以要求數據源的變動對於編碼人員來說是透明，也就是說同樣SQL語句在不同的環境下操作的數據庫是不一樣的。具體的配置如下： 
一、首先需要寫一個靜態的鍵值對照類：

代碼

package cn.com.xinli.ccp.dynamicds;
public class DataSourceMap {
public static final String Admin="Admin";
public static final String Yxh = "Yxh";
}

 

 

這個類主要在使用的時候當作獲得數據源的標誌使用。 
二、建立一個獲得和設置上下文的類：

代碼

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：

代碼

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的配置文件配置數據源

代碼

<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配置文件配置多數據源映射關係

代碼

<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結合的配置一樣

代碼

<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的代碼這裏就省略了。 
七、配置結束，可以使用了。

代碼

 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，就可以訪問不同的數據庫了