Spring HIbernate 多資料來源切換
資料來源配置
<bean id="sessionFactory"
class="org.springframework.orm.hibernate4.LocalSessionFactoryBean">
<property name="dataSource" ref="multipleDataSource" />
<property name="packagesToScan">
<list>
<!-- 可以加多個包 -->
<value>com.lei.demo.entity</value>
</list>
</property>
<propertyname="hibernateProperties">
<props>
<!-- <propkey="hibernate.hbm2ddl.auto">${hibernate.hbm2ddl.auto}</prop>-->
<propkey="hibernate.dialect">${hibernate.dialect}</prop>
<propkey="hibernate.show_sql">${hibernate.show_sql}</prop>
<!-- <propkey="hibernate.current_session_context_class">thread</prop>-->
</props>
</property>
</bean>
<bean id="dataSource"class="com.alibaba.druid.pool.DruidDataSource"init-method="init" destroy-method="close">
<property name="url"value="${jdbc.url}" />
<property name="username"value="${jdbc.user}" />
<property name="password"value="${jdbc.pass}" />
<property name="filters"value="stat" />
<property name="maxActive"value="20" />
<propertyname="initialSize" value="1" />
<property name="maxWait"value="60000" />
<property name="minIdle"value="1" />
<propertyname="timeBetweenEvictionRunsMillis" value="3000" />
<propertyname="minEvictableIdleTimeMillis" value="300000" />
<propertyname="validationQuery" value="SELECT 'x'" />
<property name="testWhileIdle"value="true" />
<propertyname="testOnBorrow" value="false" />
<propertyname="testOnReturn" value="false" />
<propertyname="poolPreparedStatements" value="true" />
<propertyname="maxPoolPreparedStatementPerConnectionSize" value="20"/>
</bean>
<bean id="dataSource2"class="com.alibaba.druid.pool.DruidDataSource"init-method="init" destroy-method="close">
<property name="url"value="${jdbc2.url}" />
<property name="username"value="${jdbc2.user}" />
<property name="password"value="${jdbc2.pass}" />
<property name="filters"value="stat" />
<property name="maxActive"value="20" />
<propertyname="initialSize" value="1" />
<property name="maxWait"value="60000" />
<property name="minIdle"value="1" />
<propertyname="timeBetweenEvictionRunsMillis" value="3000" />
<propertyname="minEvictableIdleTimeMillis" value="300000" />
<propertyname="validationQuery" value="SELECT 'x'" />
<propertyname="testWhileIdle" value="true" />
<property name="testOnBorrow"value="false" />
<propertyname="testOnReturn" value="false" />
<propertyname="poolPreparedStatements" value="true" />
<propertyname="maxPoolPreparedStatementPerConnectionSize" value="20"/>
</bean>
<bean id="multipleDataSource"class="com.lei.demo.MultipleDataSource">
<propertyname="defaultTargetDataSource" ref="dataSource"/>
<property name="targetDataSources">
<map>
<!-- 注意這裡的value是和上面的DataSource的id對應，key要和下面的CustomerContextHolder中的常量對應 -->
<entryvalue-ref="dataSource" key="dataSource"/>
<entryvalue-ref="dataSource2" key="dataSource2"/>
</map>
</property>
</bean>
2、配置攔截器
<beanclass="org.springframework.aop.aspectj.annotation.AnnotationAwareAspectJAutoProxyCreator"/>
<beanclass="org.springframework.web.servlet.mvc.annotation.AnnotationMethodHandlerAdapter"/>
<beanclass="org.springframework.web.servlet.mvc.method.annotation.RequestMappingHandlerMapping">
<propertyname="interceptors">
<list>
<bean class="com.ug.web.ControllerInterceptor"/>
</list>
</property>
</bean>
<beanclass="org.springframework.web.servlet.mvc.method.annotation.RequestMappingHandlerAdapter"/>
這裡注意：RequestMappingHandlerMapping，否則攔截器中
public boolean preHandle(HttpServletRequest request, HttpServletResponseresponse, Object arg2) throws Exception {
arg2 會是Controller而不是method
3、攔截器
HandlerMethod handlerMethod = (HandlerMethod)arg2;
Method method =handlerMethod.getMethod();
DataSourceSetter annotation =method.getAnnotation(DataSourceSetter.class);
if(annotation == null ||annotation.source().equals("")){
//設定預設資料來源
CustomerContextHolder.setCustomerType(CustomerContextHolder.DATA_SOURCE1);
} else {
CustomerContextHolder.setCustomerType(annotation.source());
}
根據註解，自動設定資料來源
4、資料來源
public abstract class CustomerContextHolder {
public final static String DATA_SOURCE1 = "dataSource";
public final static String DATA_SOURCE2 ="dataSource2";
private static final ThreadLocal<String> contextHolder =new ThreadLocal<String>();
public static void setCustomerType(String customerType) {
contextHolder.set(customerType);
}
public static String getCustomerType() {
return contextHolder.get();
}
public static void clearCustomerType() {
contextHolder.remove();
}
}
public class MultipleDataSource extends AbstractRoutingDataSource {
@Override
protected Object determineCurrentLookupKey() {
return CustomerContextHolder.getCustomerType();
}
}
5、註解
//method級的資料來源
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface DataSourceSetter{
String source() default "";
}
空，則設定預設資料來源
//Control級的資料來源
@Target(ElementType.Type)
@Retention(RetentionPolicy.RUNTIME)
public @interface DataSourceSetter{
String source() default "";
}
空，則設定預設資料來源
放在method上的
@DataSourceSetter(source= CustomerContextHolder.DATA_SOURCE2)
注意：CustomerContextHolder.setCustomerType(annotation.source());
這種方式，設定過一次之後，後面的關聯都是按照此資料來源來進行。所以在多資料來源系統中，必須自動按照配置，每次都需要設定資料來源。否則會因為資料來源沒有切換回來，導致查詢失敗。