Spring+mybatis多数据源切换笔记
分两种方式，手动和自动切换，
前提：使用mybatis自动生成工具生成相关xml，实体类，接口等，spring框架，依赖的jar包都已引入
1.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:p="http://www.springframework.org/schema/p" xmlns:aop="http://www.springframework.org/schema/aop" xmlns:context="http://www.springframework.org/schema/context" xmlns:tx="http://www.springframework.org/schema/tx" xsi:schemaLocation="http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans.xsd http://www.springframework.org/schema/context http://www.springframework.org/schema/context/spring-context-4.3.xsd http://www.springframework.org/schema/aop http://www.springframework.org/schema/aop/spring-aop-4.3.xsd http://www.springframework.org/schema/tx http://www.springframework.org/schema/tx/spring-tx-4.3.xsd"> <!--6 容器自动扫描IOC组件 --> <context:component-scan base-package="com.vip"></context:component-scan> <!--1 引入属性文件，在配置中占位使用 --> <context:property-placeholder location="classpath*:dbt.properties" /> <!-- 配置数据源Master --> <bean name="dataSourceMaster" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close"> <property name="url" value="${jdbc.url}" /> <property name="username" value="${jdbc.uid}" /> <property name="password" value="${jdbc.pwd}" /> <!-- 初始化连接大小 --> <property name="initialSize" value="0" /> <!-- 连接池最大使用连接数量 --> <property name="maxActive" value="20" /> <!-- 连接池最大空闲 --> <property name="maxIdle" value="20" /> <!-- 连接池最小空闲 --> <property name="minIdle" value="0" /> <!-- 获取连接最大等待时间 --> <property name="maxWait" value="60000" /> </bean> <!-- 配置数据源Slave --> <bean name="dataSourceSlave" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close"> <property name="url" value="${jdbc2.url}" /> <property name="username" value="${jdbc2.uid}" /> <property name="password" value="${jdbc2.pwd}" /> <!-- 初始化连接大小 --> <property name="initialSize" value="0" /> <!-- 连接池最大使用连接数量 --> <property name="maxActive" value="20" /> <!-- 连接池最大空闲 --> <property name="maxIdle" value="20" /> <!-- 连接池最小空闲 --> <property name="minIdle" value="0" /> <!-- 获取连接最大等待时间 --> <property name="maxWait" value="60000" /> </bean> <bean id="dataSource" class="com.vip.inventory.mybaisc.ThreadLocalRountingDataSource"> <property name="defaultTargetDataSource" ref="dataSourceMaster" /> <property name="targetDataSources"> <map key-type="com.vip.inventory.mybaisc.DataSources"> <entry key="MASTER" value-ref="dataSourceMaster" /> <entry key="SLAVE" value-ref="dataSourceSlave" /> <!-- 这里还可以加多个dataSource --> </map> </property> </bean> <!--3 会话工厂bean sqlSessionFactoryBean --> <bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean"> <!-- 数据源 --> <property name="dataSource" ref="dataSource"></property> <!-- 实体类别名 --> <property name="typeAliasesPackage" value="com/vip/inventory/entity"></property> <!-- sql映射文件路径 --> <property name="mapperLocations" value="classpath*:mapper/**/*Mapper.xml"></property> </bean> <!--4 自动扫描对象关系映射 --> <bean class="org.mybatis.spring.mapper.MapperScannerConfigurer"> <!--指定会话工厂，如果当前上下文中只定义了一个则该属性可省去 --> <property name="sqlSessionFactoryBeanName" value="sqlSessionFactory"></property> <!-- 指定要自动扫描接口的基础包，实现接口 --> <property name="basePackage" value="com.vip"></property> </bean> <!--5 声明式事务管理 --> <!--定义事物管理器，由spring管理事务 --> <bean id="transactionManager" class="org.springframework.jdbc.datasource.DataSourceTransactionManager"> <property name="dataSource" ref="dataSource"></property> </bean> <!--支持注解驱动的事务管理，指定事务管理器 --> <tx:annotation-driven transaction-manager="transactionManager" /> <!--7 aspectj支持自动代理实现AOP功能 --> <aop:aspectj-autoproxy proxy-target-class="true"></aop:aspectj-autoproxy> </beans>
2.数据源配置文件：
#mysql jdbc jdbc.driver=com.mysql.jdbc.Driver jdbc.url=jdbc:mysql://ip/test1?useUnicode=true&characterEncoding=UTF-8 jdbc.uid=uid jdbc.pwd=pa #galaxy jdbc jdbc2.driver=com.mysql.jdbc.Driver jdbc2.url=jdbc:mysql://ip/test2?useUnicode=true&characterEncoding=UTF-8 jdbc2.uid=uid jdbc2.pwd=pa
3.新建枚举类：
package com.inventory.mybaisc;
public enum DataSources {
MASTER, SLAVE
}
注意此枚举值必须对应spring配置文件各数据源的别名
4.新建DataSourceTypeManager类，具体执行数据源切换
public class DataSourceTypeManager { private static final ThreadLocal<DataSources> dataSourceTypes = new ThreadLocal<DataSources>(){ @Override protected DataSources initialValue(){ return DataSources.MASTER; } }; public static DataSources get(){ return dataSourceTypes.get(); } public static void set(DataSources dataSourceType){ dataSourceTypes.set(dataSourceType); } public static void reset(){ dataSourceTypes.set(DataSources.MASTER); } }
5.新建ThreadLocalRountingDataSource继承AbstractRoutingDataSource，实现切换
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource; public class ThreadLocalRountingDataSource extends AbstractRoutingDataSource{ @Override protected Object determineCurrentLookupKey() { return DataSourceTypeManager.get(); } }
一：执行具体mybatis接口时手动切换方式：
在service层具体调用insert接口进行数据库操作前加入：
DataSourceTypeManager.set(DataSources.MASTER);
则，在执行insert时会切换数据源到master
insert方法为mybatic自动生成的接口，具体如下：
public interface InventoryMapper { int deleteByPrimaryKey(Long id); int deleteByItemKey(String item); int insert(Inventory record); int insertSelective(Inventory record); Inventory selectByPrimaryKey(Long id); int updateByPrimaryKeySelective(Inventory record); int updateByPrimaryKey(Inventory record); }
二：动态切换mybatic数据源
通过增加一个切面去拦截servcie层在调用mybatis生成的接口时，来切换数据源，从而省去每次调用mybatis生成的接口时都要手动注明数据源
AOP代码如下：
@Aspect // for aop @Component // for auto scan @Order(0) // execute before @Transactional public class DynamicChangeDbSource { @Pointcut("execution(public * com.inventory.ferrari.service.data.ferrari..*.*(..))") public void invanyMethod() { }; @Pointcut("execution(public * com.inventory.ferrari.service.data.galaxy..*.*(..))") public void galaxyanyMethod() { }; @Before("invanyMethod()") public void beforeinv(JoinPoint jp) { Object[] args = jp.getArgs(); if(args==null){ DataSourceTypeManager.set(DataSources.MASTER); //return; } //System.out.println("-------------" + args[0]); DataSourceTypeManager.set(DataSources.MASTER); } @Before("galaxyanyMethod()") public void beforegalaxy(JoinPoint jp) { DataSourceTypeManager.set(DataSources.SLAVE); } }
@Order(0) 注解表示在执行sql前就切换数据源
