利用spring aop实现数据源的切换

最近项目中使用了数据库分库，所以用到了用到了数据源切换。写了一个简单的demo，供参考。

数据源切换的原理，采用的spring的aop和AbstractRoutingDataSource。

<?xml version="1.0" encoding="UTF-8"?> <beans xmlns="http://www.springframework.org/schema/beans" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns:aop="http://www.springframework.org/schema/aop" xmlns:tx="http://www.springframework.org/schema/tx" xmlns:context="http://www.springframework.org/schema/context" xsi:schemaLocation="http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans-2.5.xsd http://www.springframework.org/schema/context http://www.springframework.org/schema/context/spring-context-2.5.xsd http://www.springframework.org/schema/aop http://www.springframework.org/schema/aop/spring-aop-2.5.xsd http://www.springframework.org/schema/tx http://www.springframework.org/schema/tx/spring-tx-2.5.xsd" default-autowire="byName" default-lazy-init="true"> <bean id="propertyConfigurer" class="org.springframework.beans.factory.config.PropertyPlaceholderConfigurer"> <property name="locations"> <list> <value>classpath*:config/init.properties </value> </list> </property> </bean> <!-- 配置数据源 --> <bean id="BaseDataSource" class="com.mchange.v2.c3p0.ComboPooledDataSource" destroy-method="close"> <property name="driverClass" value="com.mysql.jdbc.Driver" /> <property name="user" value="${datasource.username}" /> <property name="password" value="${datasource.password}" /> <property name="maxPoolSize" value="${datasource.maxPoolSize}" /> <property name="minPoolSize" value="${datasource.minPoolSize}" /> <property name="initialPoolSize" value="${datasource.initialPoolSize}" /> <property name="acquireIncrement" value="5" /> <property name="idleConnectionTestPeriod" value="7200" /> <property name="maxIdleTime" value="600" /> <property name="maxConnectionAge" value="0" /> <property name="maxStatements" value="20000" /> <property name="maxStatementsPerConnection" value="300" /> <property name="numHelperThreads" value="20" /> <property name="maxAdministrativeTaskTime" value="30" /> <property name="acquireRetryAttempts" value="30" /> <property name="acquireRetryDelay" value="1000" /> <property name="autoCommitOnClose" value="false" /> <property name="automaticTestTable" value="Test" /> <!--property name="preferredTestQuery" value="select id from test where id=1"/ --> <property name="breakAfterAcquireFailure" value="false" /> <property name="checkoutTimeout" value="10000" /> <property name="factoryClassLocation" value="" /> <property name="forceIgnoreUnresolvedTransactions" value="false" /> <property name="propertyCycle" value="300" /> <property name="testConnectionOnCheckout" value="false" /> <property name="testConnectionOnCheckin" value="false" /> <property name="usesTraditionalReflectiveProxies" value="false" /> </bean> <bean id="dataSource" class="org.pay.common.RoutingDataSource"> <property name="targetDataSources"> <map key-type="java.lang.String"> <entry key="dataSourceTest" value-ref="dataSourceTest" /> <entry key="dataSourceTeachmis" value-ref="dataSourceTeachmis" /> </map> </property> <property name="defaultTargetDataSource" ref="dataSourceTest" /> </bean> <bean id="dataSourceTest" class="com.mchange.v2.c3p0.ComboPooledDataSource" destroy-method="close" parent="BaseDataSource"> <property name="jdbcUrl" value="${datasource.url}" /> </bean> <bean id="dataSourceTeachmis" class="com.mchange.v2.c3p0.ComboPooledDataSource" destroy-method="close" parent="BaseDataSource"> <property name="jdbcUrl" value="${datasource.teach.url}" /> </bean> <!-- 在做大字段处理的时候一定要用lobHandler 但是lobHandler针对于MySQL、DB2、MS SQLServer、Oracle10g可以 --> <bean id="lobHandler" class="org.springframework.jdbc.support.lob.DefaultLobHandler" lazy-init="true" /> <bean id="transactionManager" class="org.springframework.jdbc.datasource.DataSourceTransactionManager"> <property name="dataSource" ref="dataSource" /> </bean> <bean id="jdbcTemplate" class="org.springframework.jdbc.core.JdbcTemplate"> <property name="dataSource" ref="dataSource" /> </bean> <!-- 数据库切换aop配置 --> <aop:config proxy-target-class="true"> <aop:advisor pointcut="execution(* org.pay.dao..*.*(..))" advice-ref="daoAdvice" order="100"/> <aop:advisor pointcut="execution(* org.pay.dao1..*.*(..))" advice-ref="daoAdvice" order="100"/> </aop:config> <bean id="daoAdvice" class="org.pay.common.DAOAdvice"> <property name="packageName"> <map> <entry key="dataSourceTest"> <value>org.pay.dao</value> </entry> <entry key="dataSourceTeachmis"> <value>org.pay.dao1</value> </entry> </map> </property> </bean> <!-- 数据库事务配置 --> <aop:config proxy-target-class="true"> <aop:advisor pointcut="execution(* org.pay.dao..*DAO.*(..))" advice-ref="txAdvice" order="100"/> </aop:config> <tx:advice id="txAdvice" transaction-manager="transactionManager"> <tx:attributes> <tx:method name="select*" rollback-for="Exception" /> <tx:method name="*" read-only="true" /> </tx:attributes> </tx:advice> <!-- 使Spring关注Annotation --> <context:annotation-config /> <!-- 让Spring通过自动扫描来查询和管理Bean --> <context:component-scan base-package="org.pay" /> </beans>

RoutingDataSource.java

package org.pay.common; import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource; public class RoutingDataSource extends AbstractRoutingDataSource { @Override protected Object determineCurrentLookupKey() { // TODO Auto-generated method stub return DataSourceHolder.getCustomerType(); } }

DataSourceHolder.java

package org.pay.common; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; public class DataSourceHolder { protected static final Log logger = LogFactory .getLog(java.sql.Connection.class); @SuppressWarnings("unchecked") private static final ThreadLocal customerType = new ThreadLocal(); @SuppressWarnings("unchecked") public static void setCustomerType(String type) { customerType.set(type); } public static String getCustomerType() { String data = (String) customerType.get(); if (data != null) { logger.debug("DataSource Name:[" + data + "]"); } else { logger.debug("DataSource Name:default"); } return data; } public static void clearCustomerType() { customerType.remove(); } }

DAOAdvice.java

package org.pay.common; import java.lang.reflect.Method; import java.util.Iterator; import java.util.Map; import org.springframework.aop.MethodBeforeAdvice; public class DAOAdvice implements MethodBeforeAdvice { private Map<String, String> packageName; public Map<String, String> getPackageName() { return packageName; } public void setPackageName(Map<String, String> packageName) { this.packageName = packageName; } public void before(Method method, Object[] args, Object obj) throws Throwable { String daoPackageName = obj.getClass().getPackage().getName(); for(Iterator<String> it = packageName.keySet().iterator();it.hasNext();) { String key = it.next(); String definePackageName = packageName.get(key); System.out.println("definePackageName:==" + definePackageName + "daoPackageName:=="+daoPackageName); if(daoPackageName.startsWith(definePackageName)){ DataSourceHolder.setCustomerType(key); } } } }

说明：RoutingDataSource类通过<bean id="dataSource" class="org.pay.common.RoutingDataSource"> 注入，在DAOAdvice类中设置的datasource的key来获得哪个数据源，DAOAdvice类根据

<aop:config proxy-target-class="true">
        <aop:advisor pointcut="execution(* org.pay.dao..*.*(..))"
            advice-ref="daoAdvice" order="100"/>

这个切面判断dao所在的package来判断使用哪个数据源将其设置到上下文根中（DataSourceHolder），

例如当org.pay.dao中的dao被调用时，取到的是dataSourceTest这个数据源