spring 多数据源切换

1、实现spring的接口

package xxxx;

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

public class DynamicDataSource extends AbstractRoutingDataSource {

    @Override
    protected Object determineCurrentLookupKey() {
        return DataSourceSwitcher.getDataSourceType();
    }
}

2、绑定线程

package xxxxx;

import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

import org.apache.commons.lang.StringUtils;

public class DataSourceSwitcher {

    // 客服系统数据源
    public static final String DATA_SOURCE_HCRM = "dataSource_hcrm";

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


    /**
     * @Description: 设置数据源类型 @param dataSourceType 数据库类型 @return void @throws
     */
    public static void setDataSourceType(String dataSourceType) {
        contextHolder.set(dataSourceType);
    }

    /**
     * @Description: 获取数据源类型 @param @return String @throws
     */
    public static String getDataSourceType() {
        String dataSource = contextHolder.get();
        if (StringUtils.isEmpty(dataSource)) {
            return DataSourceSwitcher2.DATA_SOURCE_HCRM;
        } 
        return dataSource;
    }

    /**
     * @Description: 清除数据源类型 @param @return void @throws
     */
    public static void clearDataSourceType() {
        contextHolder.remove();
    }
}

3、操作数据库之前切换数据源

public int insert(String dataSourceCode, String sqlMapId, Object object){
        DataSourceSwitcher.setDataSourceType(dataSourceCode);
        return myBatisDAO.insert(sqlMapId, object);
    }

4、配置动态数据源

 <bean id="data_source_2_0" parent="dataSource" init-method="init"
    destroy-method="close">
    <property name="url" value="${jdbc-2_0.url}" />
    <property name="username" value="${jdbc-2_0.user}" />
    <property name="password" value="${jdbc-2_0.password}" />
</bean>

<!-- 数据源动态映射 -->
<!-- 这种动态数据源在分布式事务时 目前做不到，这种情况可能要用JPA分布式事务-->
<bean id="dynamicDataSource" class="xxx.DynamicDataSource">
    <property name="targetDataSources">
        <map>
            <entry key="xxxx" value-ref="xxxx" />
            <entry key="data_source_2_0" value-ref="data_source_2_0" />
        </map>
    </property>
    
</bean>

<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
    <property name="dataSource" ref="dynamicDataSource" />
    <property name="configLocation" value="classpath:db/sql_map_service_config.xml" />
</bean>
<!-- 复杂的操作可以用到 和 DataSourceSwitcher.switchDataSource 配合适用,切换数据库-->
<!-- 如果涉及到事务,需要spring的声明事务，目前的想到的方法是加入拦截器 在开启事务之前切换数据源，aop建议切入到dao层 -->
<!-- 也就是说 ：在一次业务处理中, 如果涉及到多数据库操作，同时又需要加入事务，建议把对一个数据库操作操作放到一个dao中,切dao层-->

如果没有数据库的事务管理，已经可以实现数据库的动态切换了。但是如果涉及到数据库的事务管理，需要在数据库事务开启切换数据库，
可以定义一个aop处理类在数据库事务开启之前切换数据库

public class DataSourceAspect implements MethodBeforeAdvice,AfterReturningAdvice 
{

    @Override
    public void afterReturning(Object returnValue, Method method,
            Object[] args, Object target) throws Throwable {
        // TODO Auto-generated method stub
        DataSourceContextHolder.clearDataSourceType();
    }

    @Override
    public void before(Method method, Object[] args, Object target)
            throws Throwable {
    
        if (method.isAnnotationPresent(DataSource.class)) 
        {
            //DataSource datasource = method.getAnnotation(DataSource.class);
            DataSourceContextHolder.setDataSourceType("XXXXXXXXXX");
        }
        else
        {
            DataSourceContextHolder.setDataSourceType("xxxxxxxxxxxxxxxx");
        }
        
    }
}

设置数据库事务切面和切换数据库切面执行的顺序

<aop:config>
        <aop:pointcut id="abc" expression="execution(* com.test.service.*.*(..))" />
        <aop:advisor pointcut-ref="abc"
            advice-ref="txAdvice" order="2" />
        <aop:advisor advice-ref="dataSourceAspect" pointcut-ref="abc" order="1"/>
    </aop:config>

总结：