一、定义用于标记使用那个数据源的注解类
package me.utils.database;
import java.lang.annotation.ElementType;
import java.lang.annotation.Target;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
/**
定义 @DataSource 注解,标记当前使用的数据源
@version 创建时间:2017年4月17日 上午11:02:07
*/
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface DataSource {
String value();
}
二、定义一个切入点,用于在执行数据连接前通过读取 @DataSource 注解,动态选择数据源
package me.utils.database;
import java.lang.reflect.Method;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.reflect.MethodSignature;
import org.aspectj.weaver.ast.Var;
import com.sun.tools.javac.resources.javac;
/**
多数据源切换切入点。解析 @DataSource 注解
@version 创建时间:2017年4月17日 上午11:08:08 */
public class DataSourceAspect {
/**
在dao层方法之前调用,获取 @DataSource 注解标记的数据源名称 */
public void before(JoinPoint point) {
Object target = point.getTarget();
String methodName = point.getSignature().getName();
MethodSignature methodSignature = (MethodSignature) point.getSignature();
Method method = methodSignature.getMethod();
//获取方法上的 @DataSource 注解
DataSource annotation = (DataSource) method.getAnnotation(DataSource.class);
String dataSourceName = null;
if (annotation != null) {
//获得注解标记的数据源名称
dataSourceName = annotation.value();
HandleDataSource.putDataSource(dataSourceName);
}
printMsg("方法={0};使用数据源={1}",methodName,annotation);
}
void printMsg(String pattern, Object… arguments){
System.out.println(java.text.MessageFormat.format(pattern,arguments));
}
}
三、定义一个记录当前线程使用的数据源的对象,管理多线程下的变量存储(线程安全)
package me.utils.database;
/**
*保存当前线程数据源使用的数据源名
@version 创建时间:2017年4月17日 上午11:07:25 / public class HandleDataSource { /*
保存当前线程使用的数据源名称 / public static final ThreadLocal holder = new ThreadLocal(); /*
绑定当前线程数据源路由的key
@param key
*/
public static void putDataSource(String datasource) {
holder.set(datasource);
System.out.println(java.text.MessageFormat.format("线程[{0}],设置使用的数据源={1}"
,Thread.currentThread().getName()
,datasource));
}
/**
获取当前线程的数据源路由的key
@return
*/
public static String getDataSource() {
System.out.println(java.text.MessageFormat.format("线程[{0}],获取使用的数据源={1}"
,Thread.currentThread().getName()
,holder.get()));
return holder.get();
}
}
四、基于 spring 的多数据源选择对象
package me.utils.database;
import java.util.Map;
import java.util.function.BiConsumer;
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
/**
继承 AbstractRoutingDataSource 实现自己的路由数据源
@version 创建时间:2017年4月17日 上午11:05:20 */
public class ChooseDataSource extends AbstractRoutingDataSource {
@Override
protected Object determineCurrentLookupKey() {
//返回当前线程使用的数据源名称
return HandleDataSource.getDataSource();
}
@Override
public void setTargetDataSources(Map targetDataSources) {
super.setTargetDataSources(targetDataSources);
System.out.println("配置读写分离数据源:");
//设置目标数据源
if(targetDataSources!=null){
targetDataSources.forEach((key,val)->{
System.out.println(java.text.MessageFormat.format("数据源:{0}={1}", key,val));
});
}
}
}
五、 mybatis 的配置(其中的顺序很关键)
<!-- 1.数据源配置 (druid 数据源 (参考:https://github.com/alibaba/druid/wiki/DruidDataSource%E9%85%8D%E7%BD%AE%E5%B1%9E%E6%80%A7%E5%88%97%E8%A1%A8) -->
<!-- 1.1 写库 -->
<bean id="dataSource_druid_write" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close">
<!-- 基本属性 url、user、password -->
<property name="url" value="${jdbc.url}" />
<property name="username" value="${jdbc.username}" />
<property name="password" value="${jdbc.password}" />
<!-- 配置初始化大小、最小、最大 -->
<property name="initialSize" value="1" />
<property name="minIdle" value="1" />
<property name="maxActive" value="20" />
<!-- 配置获取连接等待超时的时间 -->
<property name="maxWait" value="60000" />
<!-- 配置间隔多久才进行一次检测,检测需要关闭的空闲连接,单位是毫秒 -->
<property name="timeBetweenEvictionRunsMillis" value="60000" />
<!-- 配置一个连接在池中最小生存的时间,单位是毫秒 -->
<property name="minEvictableIdleTimeMillis" value="300000" />
<property name="validationQuery" value="SELECT 'x'" />
<property name="testWhileIdle" value="true" />
<property name="testOnBorrow" value="false" />
<property name="testOnReturn" value="false" />
<!-- 打开PSCache,并且指定每个连接上PSCache的大小 -->
<property name="poolPreparedStatements" value="true" />
<property name="maxPoolPreparedStatementPerConnectionSize" value="20" />
<!-- 配置监控统计拦截的filters -->
<property name="filters" value="mergeStat" />
<!-- 合并多个DruidDataSource的监控数据 -->
<property name="useGlobalDataSourceStat" value="true" />
<!-- 慢SQL记录监控 -->
<property name="connectionProperties" value="druid.stat.slowSqlMillis=5000" />
</bean>
<!-- 1.2 只读库 -->
<bean id="dataSource_druid_read" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close">
<!-- 基本属性 url、user、password -->
<property name="url" value="${jdbc.url}" />
<property name="username" value="${jdbc.username}" />
<property name="password" value="${jdbc.password}" />
<!-- 配置初始化大小、最小、最大 -->
<property name="initialSize" value="1" />
<property name="minIdle" value="1" />
<property name="maxActive" value="20" />
<!-- 配置获取连接等待超时的时间 -->
<property name="maxWait" value="60000" />
<!-- 配置间隔多久才进行一次检测,检测需要关闭的空闲连接,单位是毫秒 -->
<property name="timeBetweenEvictionRunsMillis" value="60000" />
<!-- 配置一个连接在池中最小生存的时间,单位是毫秒 -->
<property name="minEvictableIdleTimeMillis" value="300000" />
<property name="validationQuery" value="SELECT 'x'" />
<property name="testWhileIdle" value="true" />
<property name="testOnBorrow" value="false" />
<property name="testOnReturn" value="false" />
<!-- 打开PSCache,并且指定每个连接上PSCache的大小 -->
<property name="poolPreparedStatements" value="true" />
<property name="maxPoolPreparedStatementPerConnectionSize" value="20" />
<!-- 配置监控统计拦截的filters -->
<property name="filters" value="mergeStat" />
<!-- 合并多个DruidDataSource的监控数据 -->
<property name="useGlobalDataSourceStat" value="true" />
<!-- 慢SQL记录监控 -->
<property name="connectionProperties" value="druid.stat.slowSqlMillis=5000" />
</bean>
<!-- 1.3 动态读写分离数据源 -->
<bean id="demoDataSource" class="me.utils.database.ChooseDataSource">
<property name="targetDataSources">
<map key-type="java.lang.String">
<!-- 写库 -->
<entry key="write" value-ref="dataSource_druid_write"/>
<!-- 读库 -->
<entry key="read" value-ref="dataSource_druid_read"/>
</map>
</property>
<!-- 设置默认数据源 -->
<property name="defaultTargetDataSource" ref="dataSource_druid_write"/>
</bean>
<!-- 2 注册 mybatis sqlmapper 映射-->
<!-- 2.1 spring 自动注册 mybatis 的映射文件配置 (mappers.mapper 节点内容) -->
<bean class="org.mybatis.spring.mapper.MapperScannerConfigurer">
<!-- 映射文件中,mapper.namespace 属性值“接口”的基础包名 -->
<property name="basePackage" value="me.daos" />
<property name="sqlSessionFactoryBeanName" value="sqlSessionFactory" />
</bean>
<!-- 2.2 sping 自动注册 mappers.mapper 文件,并将对应的接口注册到ioc容器(即:使用接口实例访问数据库) -->
<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
<!-- 使用的数据源 -->
<property name="dataSource" ref="demoDataSource" />
<!-- 如需添加特殊映射文件,可以写到下面的 xml 文件中 -->
<property name="configLocation" value="classpath:mybatis-config/sqlMapConfig.xml"/>
<!-- 自动扫描mapping.xml文件,**表示迭代查找,也可在sqlMapConfig.xml中单独指定xml文件-->
<property name="mapperLocations" value="classpath:mappers/**/*Mapper.xml" />
</bean>
<!-- 2.3 使用 sqlSessionFactory 管理 sqlsession 对象,并注册到ioc容器(即:使用 sqlSession.selectList() 等方式访问数据库)-->
<bean id="sqlSessionTemplate" class="org.mybatis.spring.SqlSessionTemplate">
<constructor-arg index="0" ref="sqlSessionFactory"></constructor-arg>
</bean>
<!-- 3 事务管理 -->
<!-- 3.1 注册(事务管理)transaction manager, use JtaTransactionManager for global tx -->
<bean id="transactionManager" class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
<!-- 管理的数据源 -->
<property name="dataSource" ref="demoDataSource" />
</bean>
<!-- 3.2 开启使用annotation(注解)控制事务,基于类的事务将启用,默认为“基于接口的代理将起作用” -->
<tx:annotation-driven transaction-manager="transactionManager" proxy-target-class="true"/>
<!-- 4 切入点,用于通过 @DataSource 注解获取当前方法使用的数据源名称 -->
<!-- 4.1 注册ioc组件 -->
<bean id="dataSourceAspect" class="me.utils.database.DataSourceAspect" />
<!-- 4.2 配置切入点,在指定包下生效 -->
<aop:config proxy-target-class="true">
<!-- 调整aop 执行排序级别 -->
<aop:aspect id="dataSourceAspect" ref="dataSourceAspect" order="1">
<!-- 标记 aop 起作用的包
星号作用:第一个*标记任何返回值类型;第二个*标记任何类型;第三个*标记任何方法;
双点儿所用:第一个..标记任何子包;第二个..标记任何参数;
-->
<aop:pointcut id="tx" expression="execution(* me.dals..*.*(..)) "/>
<!-- 在调用标有 @DataSource注解 的方法之前,执行 me.utils.database.DataSourceAspect.before() 方法 -->
<aop:before pointcut-ref="tx" method="before" />
</aop:aspect>
</aop:config>
六、调用测试
package me.dals;
import java.util.List;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Repository;
import org.springframework.transaction.annotation.Propagation;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.transaction.interceptor.TransactionAspectSupport;
import me.daos.UserInfoDao;
import me.exceptions.RollbackException;
import me.models.UserInfoModel;
/**
@version 创建时间:2017年4月11日 下午4:16:54 / @Repository public class UserInfoDal { /*
注入 UserInfoDao 实例 / @Autowired UserInfoDao dao; /*
注入 SqlSessionTemplate 实例 */
@Autowired
org.mybatis.spring.SqlSessionTemplate sqlSession;
/**
使用接口形式查询数据库
@return / @me.utils.database.DataSource(value="write") public List getAllUseMapper() { List list = dao.getAll(); return list; } /*
使用 sqlsession 形式查询数据库
@return */
@me.utils.database.DataSource(value="read")
public List getAllUseSqlSession() {
List list = sqlSession.selectList("me.daos.UserInfoDao.getAll");
return list;
}
}