Spring+Mybatis多数据源的一种实现方式，支持事务

时间：2023-03-09 04:41:15
Spring+Mybatis多数据源的一种实现方式，支持事务

最近一个项目用到了多个数据库，所以需要实现动态切换数据源来查询数据，http://www.cnblogs.com/lzrabbit/p/3750803.html这篇文章让我受益匪浅，提供了一种自动切换数据源的思路，但这种方式不支持事务，所以我进一步改进了这个方案，下面直入正题

多数据源配置：

#============================================================================

# DataBaseOne

#============================================================================

jdbc.one.driver=com.mysql.jdbc.Driver

jdbc.one.url=jdbc:mysql://127.0.0.1:3306/DataBaseOne?useUnicode=true&characterEncoding=UTF-8&allowMultiQueries=true

jdbc.one.username=root

jdbc.one.password=root

=============================================================================

#============================================================================

# DataBaseTwo

#============================================================================

jdbc.two.driver=com.mysql.jdbc.Driver

jdbc.two.url=jdbc:mysql://127.0.0.1:3306/DataBaseTwo?useUnicode=true&characterEncoding=UTF-8&allowMultiQueries=true

jdbc.two.username=root

jdbc.two.password=root

=============================================================================

#============================================================================

# DataBaseThree

#============================================================================

jdbc.three.driver=com.mysql.jdbc.Driver

jdbc.three.url=jdbc:mysql://127.0.0.1:3306/DataBaseThree?useUnicode=true&characterEncoding=UTF-8&allowMultiQueries=true

jdbc.three.username=root

jdbc.mysql.password=root

=============================================================================

#============================================================================

# 通用配置

#============================================================================

jdbc.initialSize=5

jdbc.minIdle=5

jdbc.maxIdle=20

jdbc.maxActive=100

jdbc.maxWait=100000

jdbc.defaultAutoCommit=false

jdbc.removeAbandoned=true

jdbc.removeAbandonedTimeout=600

jdbc.testWhileIdle=true

jdbc.timeBetweenEvictionRunsMillis=60000

jdbc.numTestsPerEvictionRun=20

jdbc.minEvictableIdleTimeMillis=300000

Spring中使用多数据源：

<?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:context="http://www.springframework.org/schema/context"

       xmlns:aop="http://www.springframework.org/schema/aop"

       xsi:schemaLocation="http://www.springframework.org/schema/beans

    http://www.springframework.org/schema/beans/spring-beans-3.0.xsd

    http://www.springframework.org/schema/context

    http://www.springframework.org/schema/context/spring-context-3.0.xsd

      http://www.springframework.org/schema/aop

          http://www.springframework.org/schema/aop/spring-aop-3.0.xsd">

    <!-- 自己根据情况补全其他配置，以下只提供了数据源配置 -->

    <!-- 多数据源配置 -->

    <bean id="propertyConfigurer" class="org.springframework.beans.factory.config.PropertyPlaceholderConfigurer">

        <property name="location" value="classpath:jdbc.properties"/>

    </bean>

    <!-- 第一个数据源dataSourceOne -->

    <bean id="dataSourceOne" class="org.apache.commons.dbcp.BasicDataSource" destroy-method="close">

        <property name="driverClassName" value="${jdbc.one.driver}"/>

        <property name="url" value="${jdbc.one.url}"/>

        <property name="username" value="${jdbc.one.username}"/>

        <property name="password" value="${jdbc.one.password}"/>

        <property name="initialSize" value="${jdbc.initialSize}"/>

        <property name="minIdle" value="${jdbc.minIdle}"/>

        <property name="maxIdle" value="${jdbc.maxIdle}"/>

        <property name="maxActive" value="${jdbc.maxActive}"/>

        <property name="maxWait" value="${jdbc.maxWait}"/>

        <property name="defaultAutoCommit" value="${jdbc.defaultAutoCommit}"/>

        <property name="removeAbandoned" value="${jdbc.removeAbandoned}"/>

        <property name="removeAbandonedTimeout" value="${jdbc.removeAbandonedTimeout}"/>

        <property name="testWhileIdle" value="${jdbc.testWhileIdle}"/>

        <property name="timeBetweenEvictionRunsMillis" value="${jdbc.timeBetweenEvictionRunsMillis}"/>

        <property name="numTestsPerEvictionRun" value="${jdbc.numTestsPerEvictionRun}"/>

        <property name="minEvictableIdleTimeMillis" value="${jdbc.minEvictableIdleTimeMillis}"/>

    </bean>

    <!-- 第二个数据源dataSourceTwo -->

    <bean id="dataSourceTwo" class="org.apache.commons.dbcp.BasicDataSource" destroy-method="close">

        <property name="driverClassName" value="${jdbc.two.driver}"/>

        <property name="url" value="${jdbc.two.url}"/>

        <property name="username" value="${jdbc.two.username}"/>

        <property name="password" value="${jdbc.two.password}"/>

        <property name="initialSize" value="${jdbc.initialSize}"/>

        <property name="minIdle" value="${jdbc.minIdle}"/>

        <property name="maxIdle" value="${jdbc.maxIdle}"/>

        <property name="maxActive" value="${jdbc.maxActive}"/>

        <property name="maxWait" value="${jdbc.maxWait}"/>

        <property name="defaultAutoCommit" value="${jdbc.defaultAutoCommit}"/>

        <property name="removeAbandoned" value="${jdbc.removeAbandoned}"/>

        <property name="removeAbandonedTimeout" value="${jdbc.removeAbandonedTimeout}"/>

        <property name="testWhileIdle" value="${jdbc.testWhileIdle}"/>

        <property name="timeBetweenEvictionRunsMillis" value="${jdbc.timeBetweenEvictionRunsMillis}"/>

        <property name="numTestsPerEvictionRun" value="${jdbc.numTestsPerEvictionRun}"/>

        <property name="minEvictableIdleTimeMillis" value="${jdbc.minEvictableIdleTimeMillis}"/>

    </bean>

    <!-- 第三个数据源dataSourceThree -->

    <bean id="dataSourceThree" class="org.apache.commons.dbcp.BasicDataSource" destroy-method="close">

        <property name="driverClassName" value="${jdbc.three.driver}"/>

        <property name="url" value="${jdbc.three.url}"/>

        <property name="username" value="${jdbc.three.username}"/>

        <property name="password" value="${jdbc.three.password}"/>

        <property name="initialSize" value="${jdbc.initialSize}"/>

        <property name="minIdle" value="${jdbc.minIdle}"/>

        <property name="maxIdle" value="${jdbc.maxIdle}"/>

        <property name="maxActive" value="${jdbc.maxActive}"/>

        <property name="maxWait" value="${jdbc.maxWait}"/>

        <property name="defaultAutoCommit" value="${jdbc.defaultAutoCommit}"/>

        <property name="removeAbandoned" value="${jdbc.removeAbandoned}"/>

        <property name="removeAbandonedTimeout" value="${jdbc.removeAbandonedTimeout}"/>

        <property name="testWhileIdle" value="${jdbc.testWhileIdle}"/>

        <property name="timeBetweenEvictionRunsMillis" value="${jdbc.timeBetweenEvictionRunsMillis}"/>

        <property name="numTestsPerEvictionRun" value="${jdbc.numTestsPerEvictionRun}"/>

        <property name="minEvictableIdleTimeMillis" value="${jdbc.minEvictableIdleTimeMillis}"/>

    </bean>

    <!-- 使用自己实现的数据源实现类MultipleDataSource，这个类随意放在哪个包下都行 -->

    <bean id="multipleDataSource" class="com.cnblogs.datasource.MultipleDataSource">

        <!-- 设置默认的数据源 -->

        <property name="defaultTargetDataSource" ref="dataSourceOne"/>

        <property name="targetDataSources">

            <map>

                <!-- 这个key是对应数据源的别称，通过这个key可以找到对应的数据源，value-ref就是上面数据源的id -->

                <entry key="dataSourceOneKey" value-ref="dataSourceOne"/>

                <entry key="dataSourceTwoKey" value-ref="dataSourceTwo"/>

                <entry key="dataSourceThreeKey" value-ref="dataSourceThree"/>

            </map>

        </property>

    </bean>

    <!-- 让spring使用我们配置的多数据源 -->

    <bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">

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

    </bean>

    <!-- mybatis.spring自动映射 -->

    <bean class="org.mybatis.spring.mapper.MapperScannerConfigurer">

        <property name="basePackage" value="com.cnblogs.mapper"/>

    </bean>

    <!-- 自动扫描，多个包以 逗号分隔 -->

    <context:component-scan base-package="com.cnblogs.**"/>

    <!-- AOP配置（事务控制） -->

    <aop:config>

        <!--pointcut元素定义一个切入点，execution中的第一个星号 用以匹配方法的返回类型， 这里星号表明匹配所有返回类型。 com.abc.service.*.*(..)表明匹配com.abc.service包下的所有类的所有方法 -->

        <aop:pointcut id="myPointcut"

            expression="execution(* com.cnblogs.service.*.*(..))" />

        <!--将定义好的事务处理策略应用到上述的切入点 -->

        <aop:advisor advice-ref="txAdvice" pointcut-ref="myPointcut" />

    </aop:config>

</beans>

MultipleDataSource.java实现：

package com.cnblogs.datasource;

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

/** 多数据源java实现 */

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();

    }

}

下面详解使用SpringAOP来实现数据源切换，并支持事务控制

上面我们配置的AOP是针对service层的，所以在调用service层的任何方法时都会经过AOP，因此我们就在AOP中先于service调用时把数据源切换，看代码

新建类MultipleDataSourceAspectAdvice，把MultipleDataSourceAspectAdvice.java放在spring能自动注入的包中，比如controller包，或者自己把这个类所在的包加入到component-scan配置中去，总之要要让spring能自动加载

package com.cnblogs.controller;

import org.apache.log4j.Logger;

import org.aspectj.lang.ProceedingJoinPoint;

import org.aspectj.lang.annotation.Around;

import org.aspectj.lang.annotation.Aspect;

import org.aspectj.lang.annotation.Pointcut;

import org.springframework.core.annotation.Order;

import org.springframework.stereotype.Component;

import com.cnblogs.datasource.MultipleDataSource;

import com.cnblogs.service.DataBaseOne;

import com.cnblogs.service.DataBaseTwo;

import com.cnblogs.service.DataBaseThree;

/**

 * 数据库链接自动切换AOP处理

 * Order优先级设置到最高，因为在所有service方法调用前都必须把数据源确定

 * Order数值越小优先级越高

 */

@Component

@Aspect

@Order(1)

public class MultipleDataSourceAspectAdvice {

    private static final Logger LOGGER = Logger.getLogger(MultipleDataSourceAspectAdvice.class);

    public MultipleDataSourceAspectAdvice() {

        LOGGER.info("MultipleDataSourceAspectAdvice 加载成功");

    }

    /**

     * 定义切面

     */

    @Pointcut("execution(* com.cnblogs.service.*.*(..))")

    public void pointCut() {

    }

    // dataSourceOneKey

    // dataSourceTwoKey

    // dataSourceThreeKey

    @Around("pointCut()")

    public Object doAround(ProceedingJoinPoint jp) throws Throwable {

        if (jp.getTarget() instanceof DataBaseOne) {

            LOGGER.debug("使用数据库链接：dataSourceOneKey");

            MultipleDataSource.setDataSourceKey("dataSourceOneKey");

        } else if (jp.getTarget() instanceof DataBaseTwo) {

            LOGGER.debug("使用数据库链接：dataSourceTwoKey");

            MultipleDataSource.setDataSourceKey("dataSourceTwoKey");

        } else if (jp.getTarget() instanceof DataBaseThree) {

            LOGGER.debug("使用数据库链接：dataSourceThreeKey");

            MultipleDataSource.setDataSourceKey("dataSourceThreeKey");

        } else {

            // 默认是dataSourceOneKey

            LOGGER.debug("使用数据库链接：dataSourceOneKey");

            MultipleDataSource.setDataSourceKey("dataSourceOneKey");

        }

        return jp.proceed();

    }

}

到这里我们所以的spring多数据源配置已经完毕，那如何在执行service方法时让service切换到正确的数据库呢？上面的类中定义了有3个类DataBaseOne，DataBaseTwo，DataBaseThree，这3个类其实只是一个interface，没有任何实现方法，我们让具体业务的service都继承至这3个类以区分不同的service对应不同的数据源，因为业务的service我是知道他用的哪个数据源的，比如FooService继承至DataBaseOne，则在使用FooService任何方法时AOP就会先把数据源切换到dataSourceOneKey，以此就达到了自动切换数据源的目的，并且支持事务，下面看代码：

package com.cnblogs.service;

/**

 * DataBaseOne数据库占位类

 * 详情参见 MultipleDataSourceAspectAdvice 类

 */

public interface DataBaseOne {

}

DataBaseTwo，DataBaseThree与这完全相同

至此多数据源就配置完毕，可以安心写业务逻辑了

原理部分博友http://www.cnblogs.com/lzrabbit/p/3750803.html讲的十分清楚，再次感谢

相关文章