ssm使用mysql_ssm实现mysql读写分离

用于以后方便查看,借用: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配置文件,

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值