用于以后方便查看,借用:https://blog.csdn.net/weixin_43882034/article/details/84996443
1.创建 DyncmicDataSourceHolder 类
packagecn.ljs.dyncmic;importcom.alibaba.dubbo.common.utils.StringUtils;importorg.slf4j.Logger;importorg.slf4j.LoggerFactory;/*** 在读写分离的实现上,
* Spring为我们提供了路由数据源的AbstractRoutingDataSource抽象类,
* 我们只需编写一个新的类继承它,
* 即可在不改变原有的代码的基础上实现读写分离*/
public classDyncmicDataSourceHolder {private static Logger logger= LoggerFactory.getLogger(DyncmicDataSourceHolder.class);/*** 保证线程安全,使用线程threadlocal*/
private static ThreadLocal contextHolder=new ThreadLocal();public static final String DB_MASTER="master";public static final String DB_SLAVE="slave";/***@returnjava.lang.String
* @description 获取连接类型*/
public staticString getDbType(){
String db=contextHolder.get();if(StringUtils.isBlank(db)) {
db=DB_MASTER;
}returndb;
}/*** 设置数据源类型
*@paramstr*/
public static voidsetDbType(String str){
logger.debug("使用数据源类型---"+str);
System.err.println("使用数据源类型---"+str);
contextHolder.set(str);
}/*** 清理连接类型*/
public static voidclearDaType(){
contextHolder.remove();
}
}
2.创建 DynamicDataSource,继承AbstractRoutingDataSource
packagecn.ljs.dyncmic;importorg.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;public class DynamicDataSource extendsAbstractRoutingDataSource {
@OverrideprotectedObject determineCurrentLookupKey() {returnDyncmicDataSourceHolder.getDbType();
}
}
3 创建DyncmicDataSourceInterceptor,mybatis拦截器
packagecn.ljs.dyncmic;importorg.apache.ibatis.executor.Executor;importorg.apache.ibatis.executor.keygen.SelectKeyGenerator;importorg.apache.ibatis.mapping.BoundSql;importorg.apache.ibatis.mapping.MappedStatement;importorg.apache.ibatis.mapping.SqlCommandType;import org.apache.ibatis.plugin.*;importorg.apache.ibatis.session.ResultHandler;importorg.apache.ibatis.session.RowBounds;importorg.slf4j.Logger;importorg.slf4j.LoggerFactory;importorg.springframework.transaction.support.TransactionSynchronizationManager;importjava.util.Locale;importjava.util.Properties;/*** 创建DyncmicDataSourceInterceptor,mybatis拦截器*/@Intercepts({@Signature(type= Executor.class, method = "update", args = {MappedStatement.class,
Object.class}),
@Signature(type= Executor.class, method = "query", args = {MappedStatement.class, Object.class,
RowBounds.class, ResultHandler.class})})public class DyncmicDataSourceInterceptor implementsInterceptor {private static Logger logger = LoggerFactory.getLogger(DyncmicDataSourceInterceptor.class);//写操作的正则表达式
private static final String REGEX = ".*insert\\u0020.*||.*delete\\u0020|.*update\\u0020.*";
@Overridepublic Object intercept(Invocation invocation) throwsThrowable {//判断方法是否是被事务管理的
boolean synchronizationAction =TransactionSynchronizationManager.isActualTransactionActive();//获取sql 中的参数
Object[] objects =invocation.getArgs();
MappedStatement mappedStatement= (MappedStatement) objects[0];//数据源key
String lookupKey =DyncmicDataSourceHolder.DB_MASTER;if (!synchronizationAction) {//读方法
if(mappedStatement.getSqlCommandType().equals(SqlCommandType.SELECT)) {//selectKey为自增id查询主键(SELECT LAST_INSERT_ID)方法
if(mappedStatement.getId().contains(SelectKeyGenerator.SELECT_KEY_SUFFIX)) {
lookupKey=DyncmicDataSourceHolder.DB_MASTER;
}else{//获取sql 语句
BoundSql boundSql = mappedStatement.getSqlSource().getBoundSql(objects[1]);
String sql= boundSql.getSql().toLowerCase(Locale.CHINA).replace("\\t\\n\\r", " ");if (sql.matches(REGEX)) {//insert、update、delete
lookupKey =DyncmicDataSourceHolder.DB_MASTER;
}else {//select
lookupKey =DyncmicDataSourceHolder.DB_SLAVE;
}
}
}
}else{
lookupKey=DyncmicDataSourceHolder.DB_MASTER;
}
logger.debug("设置方法[{}] use [{}] Strategy,SqlCommandType [{}]...", mappedStatement.getId(),
lookupKey, mappedStatement.getSqlCommandType().name());
DyncmicDataSourceHolder.setDbType(lookupKey);returninvocation.proceed();
}
@OverridepublicObject plugin(Object target) {//Executor表示含有增删改查的操作 的对象
if (target instanceofExecutor) {//有增删改查的操作,就调用拦截方法
return Plugin.wrap(target, this);
}else{//无增删改查的操作。不做处理
returntarget;
}
}
@Overridepublic voidsetProperties(Properties properties) {
}
}
4.修改mybatis-config,引入拦截器
/p>
PUBLIC"-//mybatis.org//DTD Config 3.0//EN"
"http://mybatis.org/dtd/mybatis-3-config.dtd">
5,修改mybatis配置文件,