Spring MVC框架搭建扩展--多数据源配置（一）

本节主要介绍spring mvc+mybatis多数据源配置

1.首先配置2个数据源
在spring-mybatis.xml里

<!-- 配置数据源1 使用的是Druid数据源 -->
    <bean id="fdataSource" class="com.alibaba.druid.pool.DruidDataSource"
        init-method="init" destroy-method="close">
        <property name="url" value="${dataSource.url}" />
        <property name="username" value="${dataSource.username}" />
        <property name="password" value="${dataSource.password}" />

        <!-- 初始化连接大小 -->
        <property name="initialSize" value="0" />
        <!-- 连接池最大使用连接数量 -->
        <property name="maxActive" value="20" />

        <!-- 连接池最小空闲 -->
        <property name="minIdle" value="0" />
        <!-- 获取连接最大等待时间 -->
        <property name="maxWait" value="60000" />
        <property name="poolPreparedStatements" value="true" />
        <property name="maxPoolPreparedStatementPerConnectionSize"
            value="33" />
        <!-- 用来检测有效sql -->
        <property name="validationQuery" value="${validationQuery}" />
        <property name="testOnBorrow" value="false" />
        <property name="testOnReturn" value="false" />
        <property name="testWhileIdle" value="true" />
        <!-- 配置间隔多久才进行一次检测，检测需要关闭的空闲连接，单位是毫秒 -->
        <property name="timeBetweenEvictionRunsMillis" value="60000" />
        <!-- 配置一个连接在池中最小生存的时间，单位是毫秒 -->
        <property name="minEvictableIdleTimeMillis" value="25200000" />
        <!-- 打开removeAbandoned功能 -->
        <property name="removeAbandoned" value="true" />
        <!-- 1800秒，也就是30分钟 -->
        <property name="removeAbandonedTimeout" value="1800" />
        <!-- 关闭abanded连接时输出错误日志 -->
        <property name="logAbandoned" value="true" />
        <!-- 监控数据库 -->
        <property name="filters" value="mergeStat" />
    </bean>

<!-- 数据源2 -->
    <bean id="sdataSource" class="com.alibaba.druid.pool.DruidDataSource"
        init-method="init" destroy-method="close">
        <property name="url" value="${jdbc.url}" />
        <property name="username" value="${jdbc.username}" />
        <property name="password" value="${jdbc.password}" />

        <!-- 初始化连接大小 -->
        <property name="initialSize" value="0" />
        <!-- 连接池最大使用连接数量 -->
        <property name="maxActive" value="20" />

        <!-- 连接池最小空闲 -->
        <property name="minIdle" value="0" />
        <!-- 获取连接最大等待时间 -->
        <property name="maxWait" value="60000" />
        <property name="poolPreparedStatements" value="true" />
        <property name="maxPoolPreparedStatementPerConnectionSize"
            value="33" />
        <!-- 用来检测有效sql -->
        <property name="validationQuery" value="${validationQuery}" />
        <property name="testOnBorrow" value="false" />
        <property name="testOnReturn" value="false" />
        <property name="testWhileIdle" value="true" />
        <!-- 配置间隔多久才进行一次检测，检测需要关闭的空闲连接，单位是毫秒 -->
        <property name="timeBetweenEvictionRunsMillis" value="60000" />
        <!-- 配置一个连接在池中最小生存的时间，单位是毫秒 -->
        <property name="minEvictableIdleTimeMillis" value="25200000" />
        <!-- 打开removeAbandoned功能 -->
        <property name="removeAbandoned" value="true" />
        <!-- 1800秒，也就是30分钟 -->
        <property name="removeAbandonedTimeout" value="1800" />
        <!-- 关闭abanded连接时输出错误日志 -->
        <property name="logAbandoned" value="true" />
        <!-- 监控数据库 -->
        <property name="filters" value="mergeStat" />
    </bean>

2.准备一个类(DynamicDataSource)继承AbstractRoutingDataSource实现determineCurrentLookupKey方法，该方法可以实现数据库的动态切换；一个设置和获取数据源的类(DataSourceContextHolder)

public class DynamicDataSource extends AbstractRoutingDataSource{

    @Override
    protected Object determineCurrentLookupKey() {
        return DataSourceContextHolder.getDataSource();
    }

}

public class DataSourceContextHolder {

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

    /**
     *设置数据源
     */
    public static void setDataSource(String dataSource){
        contextHolder.set(dataSource);
    }

    /**
     *获取数据源
     */
    public static String getDataSource(){
        return contextHolder.get();
    }

    /**
     * 清除数据源名字
     */
    public static void clearDataSource(){
        contextHolder.remove();
    }
}

3.spring-mybatis.xml里配置多数据源

<!-- 多数据源配置 -->
    <bean id="dataSource" class="com.ghca.util.db.DynamicDataSource">
        <property name="defaultTargetDataSource" ref="fdataSource"></property>
        <property name="targetDataSources">
            <map key-type ="java.lang.String">
                <entry key="fdataSource" value-ref="fdataSource" />
                <entry key="sdataSource" value-ref="sdataSource" />
            </map>
        </property>
    </bean>

<!-- myBatis文件 -->
    <bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
        <property name="dataSource" ref="dataSource" />
        <!-- 自动扫描entity目录, 省掉Configuration.xml里的手工配置 -->
        <property name="mapperLocations" value="classpath:mapper/*.xml" />
    </bean>

    <bean class="org.mybatis.spring.mapper.MapperScannerConfigurer">
        <property name="basePackage" value="com.ghca.dao" />
        <property name="sqlSessionFactoryBeanName" value="sqlSessionFactory" />
    </bean>

<!-- 配置事务管理器 -->
    <bean id="transactionManager"
        class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
        <property name="dataSource" ref="dataSource" />
    </bean>

4.通过aop配置来实现自动数据源切换
本文先通过xml方式配置
(1)准备一个Annotation，用于记录数据源名：

@Target({ElementType.METHOD,ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface DataSource {

    String name() default DataSource.fdataSource;

    public static String fdataSource = "fdataSource";

    public static String sdataSource = "sdataSource";
}

(2)准备一个切面：

public class AspectDataSourceExchange{

    public void afterReturning(){
        DataSourceContextHolder.clearDataSource();
    }

    public void before(JoinPoint joinPoint){
        // 从切点上获取目标方法
        MethodSignature methodSignature = (MethodSignature) joinPoint.getSignature();
        Method method = methodSignature.getMethod();
        if (method.isAnnotationPresent(DataSource.class)) {
            DataSource datasource = method.getAnnotation(DataSource.class);
            if (null == datasource || (null != datasource && null == datasource.name())) {
                DataSourceContextHolder.setDataSource(DataSource.fdataSource);
                System.out.println("当前数据源为：" + DataSourceContextHolder.getDataSource());
            }
            else {
                DataSourceContextHolder.setDataSource(datasource.name());
                System.out.println("当前数据源为：" + DataSourceContextHolder.getDataSource());
            }
        }
    }

}

(3)aop配置
spring-mybatis.xml里

<!-- 注册切换数据源bean (切面) -->
    <bean id="aspectDataSourceExchange" class="com.ghca.util.db.AspectDataSourceExchange" />

    <!-- Spring aop事务管理 -->
    <aop:config>
        <aop:aspect id="aspect" ref="aspectDataSourceExchange">
            <aop:pointcut expression="execution(* com.ghca.service.impl.LoginServiceImpl.*(..))"
                id="transactionPointcut" />
            <aop:before method="before" pointcut-ref="transactionPointcut"/>
            <aop:after-returning method="afterReturning"
                pointcut-ref="transactionPointcut"/>
        </aop:aspect>
    </aop:config>

(4)激活代理
在spring-servlet.xml里

<!-- 激活自动代理功能 -->
    <aop:aspectj-autoproxy />

(ps:
1. 配置的位置不对容易导致报错!尽量放在DispatcherServlet对应的xml文件里
2. proxy-target-class默认为false,如果配成true，则需导入CGLIB包
)
5.代码测试
LoginService.java

public interface LoginService {
    //设置数据源
    @DataSource(name = DataSource.sdataSource)
    User findUser(String username);
}

LoginServiceImpl .java

@Service
public class LoginServiceImpl implements LoginService{

    @Autowired
    private UserMapper userMapper;

    public User findUser(String username) {
        User user = userMapper.findUser(username);
        return user;
    }

}

Controller

@Controller
public class LoginController {

    @Autowired
    private LoginService loginService;

    @RequestMapping("/login")
    @ResponseBody
    @DataSource(name = DataSource.sdataSource)
    public Object login(String username){
        User user = loginService.findUser(username);
        if (user == null) {
            return "该用户未注册！";
        }
        return user;
    }
}

测试结果：