关系型DB(MySQL,MyBatis )

Spring+mybatis动态切换数据源

2018-11-29  本文已影响0人  我叫11

<font size="1">版权声明:本文为博主原创文章,未经博主允许不得转载。https://blog.csdn.net/qq_36523638/article/details/84571304</font>

Spring+mybatis动态切换数据源

这是基于mybatis的动态切换数据源

数据源连接池使用的为Druid
bean工厂使用的是mybatis的SqlSessionFactoryBean

1、配置第一个数据源:

#配置第一个数据源(具体配置视情况而定):
<bean id="dataSourceA" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close">
        <property name="driverClassName" value="${jdbc.driver}"/>
        <property name="url" value="${jdbc.url}"/>
        <property name="username" value="${jdbc.username}"/>
        <property name="password" value="${jdbc.password}"/>
        <!--<property name="filters" value="stat,log4j"/>-->
        <property name="maxActive" value="20"/>
        <property name="initialSize" value="1"/>
        <property name="maxWait" value="60000"/>
        <property name="minIdle" value="1"/>
        <property name="timeBetweenEvictionRunsMillis" value="3000"/>
        <property name="minEvictableIdleTimeMillis" value="300000"/>
        <property name="validationQuery" value="SELECT 'x' FROM DUAL"/>
        <property name="testWhileIdle" value="true"/>
        <property name="testOnBorrow" value="false"/>
        <property name="testOnReturn" value="false"/>
        <property name="poolPreparedStatements" value="true"/>
        <property name="maxPoolPreparedStatementPerConnectionSize" value="20"/>
    </bean>

2、配置第二个数据源:

#配置第二个数据源(具体配置视情况而定):
<bean id="dataSourceB" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close">
        <property name="driverClassName" value="${jdbc.driver}"/>
        <property name="url" value="${jdbc.url}"/>
        <property name="username" value="${jdbc.username}"/>
        <property name="password" value="${jdbc.password}"/>
    </bean>
#配置第三个数据源
#配置第四个数据源
...
...
#可以配置多个数据源,只要不超出连接池的限制就可以

3、配置动态的数据源(关键)这里的开关:MultipleDataSource在第5步说明

 <bean id="multipleDataSource" class="包名.MultipleDataSource">
        <!-- 这里可以设置一个默认的数据源 >
        <property name="defaultTargetDataSource" ref="dataSource" />
        <property name="targetDataSources">
            <map>
                <entry key="dataSourceA" value-ref="dataSourceA" />
                <entry key="dataSourceB" value-ref="dataSourceB" />
            </map>
        </property>
    </bean>

4、然后在配置sqlSessionFactory的配置文件中添加属性:

<property name="dataSource" ref="multipleDataSource"/>

添加完后是这样:

<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
        <property name="dataSource" ref="multipleDataSource"/>
        <一些插件的配置>
        ...
        </>
</bean>

5、包名.MultipleDataSource(动态切换开关):

public class MultipleDataSource extends AbstractRoutingDataSource {

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

    public static void setDataSourceKey(String dataSource) {
        dataSourceKey.set(dataSource);
    }

    @Override
    protected Object determineCurrentLookupKey() {
        return dataSourceKey.get();
    }

    public static void removeDataSourceKey() {
        dataSourceKey.remove();
    }
}

6、最后自己写一个切面AOP来做数据源的切换:

/**
 * 数据源切换AOP
 */
@Aspect
@Order(Ordered.LOWEST_PRECEDENCE-1)
@Component
public class MultipleDataSourceInterceptor {
    /**
     * 拦截器对所有的业务实现类请求之前进行数据源切换 特别注意,由于用到了多数据源,Mapper的调用最好只在*ServiceImpl,不然调用到非默认数据源的表时,会报表不存在的异常
     *
     * @throws
     */
    @Pointcut("execution(* com..*Impl.*(..)) || execution(* com..*Mapper.*(..))")
    public void log(){
    }
    @Before("log()")
    public void setDataSoruce(JoinPoint joinPoint)
            throws Throwable {
        //int i=8;
        //int j=i/0;
        Class<?> clazz = joinPoint.getTarget().getClass();
        String className = clazz.getName();
        if (ClassUtils.isAssignable(clazz, Proxy.class)) {
            className = joinPoint.getSignature().getDeclaringTypeName();
        }

        // 对包名含有serverwpms的设置为wpms数据源,否则默认为后台的数据源
        if (className.contains(".serverwpms.")) {
                MultipleDataSource.setDataSourceKey("dataSourceA");
        }else if (className.contains(".server.") || className.contains(".mapper.")) {
                MultipleDataSource.setDataSourceKey("dataSourceB");
        }
    }

    /**
     * 当操作完成时,释放当前的数据源 如果不释放,频繁点击时会发生数据源冲突,本是另一个数据源的表,结果跑到另外一个数据源去,报表不存在
     *
     * @param joinPoint
     * @throws Throwable
     */
    @After("log()")
    public void removeDataSoruce(JoinPoint joinPoint)
            throws Throwable {
        MultipleDataSource.removeDataSourceKey();
    }
}

aop在springmvc的配置文件中定义下:

    <!-- 定义aspect类 -->
    <bean name="multipleDataSourceInterceptor" class="包名.MultipleDataSourceInterceptor"/>

<font size="4" color="red"> ----到这,系统中如果没有用到事务,多数据源的配置就完成了,aop会根据你想要的需求去动态切换数据源,如果用到事务的话,仅仅到这是不够的</font>

查阅资料说事务管理器是从connection级别来处理的,尤其是非常方便的DataSourceTransactionManager管理器,方法上用一个@Transactional注解就可以完成事务管理。
DataSourceTransactionManager管理器会对connection进行缓存(大概是这个意思),导致我们的aop MultipleDataSourceInterceptor虽然将数据源A的字符串切换为了数据源B,但是根本上的connection是并没有切换回来的,这时候如果本该对数据源A操作的sql去操作数据源B了,会报表不存在的错误。

<font size = "4" color="blue">这时候应该怎么解决呢,其实主要是事务管理器执行的优先级比我们的aop优先级高罢了,每次都是事务管理器先执行,然后aop才去切换数据源,发现并没有切换过来,手动设置下aop的优先级高于事务管理器就好了,而事务管理器的优先级默认为Order(1),也就是我之前所见过的最高的优先级了</font>

<font size="5" color="green"></font>

上面<font size="5" color="red">@Order(Ordered.LOWEST_PRECEDENCE-1)</font>为关键,意思是给aop定一个优先级,这个优先级为多少呢, Ordered.LOWEST_PRECEDENCE为最低的order,大家都知道,Order越低,优先级越高,事务管理器默认优先级为1,这个aop的order比最低的Order还要小一个单位,代表优先级是最高的。

注:事务管理器我是这么配置的:

<!-- 配置事务管理器,注意这里的dataSource和SqlSessionFactoryBean的dataSource要一致,不然事务就没有作用了 -->
    <bean id="transactionManager" class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
        <property name="dataSource" ref="multipleDataSource"/>
    </bean>

    <!-- 用注解来实现事务管理 -->
    <tx:annotation-driven transaction-manager="transactionManager"/>
上一篇下一篇

猜你喜欢

热点阅读