文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

Springboot如何实现自定义mybatis拦截器

2023-06-22 07:02

关注

这篇文章将为大家详细讲解有关Springboot如何实现自定义mybatis拦截器,小编觉得挺实用的,因此分享给大家做个参考,希望大家阅读完这篇文章后可以有所收获。

实践的准备 :
 

整合mybatis ,然后故意写了3个查询方法, 1个是list 列表数据,2个是 单条数据 。

Springboot如何实现自定义mybatis拦截器

我们通过自己写一个MybatisInterceptor实现 mybatis框架的 Interceptor来做文章:
 

MybatisInterceptor.java :

import org.apache.ibatis.executor.Executor;import org.apache.ibatis.mapping.*;import org.apache.ibatis.plugin.*;import org.apache.ibatis.session.ResultHandler;import org.apache.ibatis.session.RowBounds;import org.springframework.stereotype.Component; import java.lang.reflect.Method;import java.util.*;   @Component@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 MybatisInterceptor implements Interceptor {     @Override    public Object intercept(Invocation invocation) throws Throwable {        //获取执行参数        Object[] objects = invocation.getArgs();        MappedStatement ms = (MappedStatement) objects[0];        //解析执行sql的map方法,开始自定义规则匹配逻辑        String mapperMethodAllName = ms.getId();        int lastIndex = mapperMethodAllName.lastIndexOf(".");        String mapperClassStr = mapperMethodAllName.substring(0, lastIndex);        String mapperClassMethodStr = mapperMethodAllName.substring((lastIndex + 1));        Class<?> mapperClass = Class.forName(mapperClassStr);        Method[] methods = mapperClass.getMethods();        Class<?> returnType;        for (Method method : methods) {            if (method.getName().equals(mapperClassMethodStr)) {                returnType = method.getReturnType();                if (returnType.isAssignableFrom(List.class)) {                    System.out.println("返回类型是 List");                    System.out.println("针对List 做一些操作");                } else if (returnType.isAssignableFrom(Set.class)) {                    System.out.println("返回类型是 Set");                    System.out.println("针对Set 做一些操作");                } else{                    BoundSql boundSql = ms.getSqlSource().getBoundSql(objects[1]);                    String oldSql = boundSql.getSql().toLowerCase(Locale.CHINA).replace("[\\t\\n\\r]", " ");                    if (!oldSql.contains("LIMIT")){                        String newSql = boundSql.getSql().toLowerCase(Locale.CHINA).replace("[\\t\\n\\r]", " ") + " LIMIT 1";                        BoundSql newBoundSql = new BoundSql(ms.getConfiguration(), newSql,                                boundSql.getParameterMappings(), boundSql.getParameterObject());                        MappedStatement newMs = newMappedStatement(ms, new MyBoundSqlSqlSource(newBoundSql));                        for (ParameterMapping mapping : boundSql.getParameterMappings()) {                            String prop = mapping.getProperty();                            if (boundSql.hasAdditionalParameter(prop)) {                                newBoundSql.setAdditionalParameter(prop, boundSql.getAdditionalParameter(prop));                            }                        }                        Object[] queryArgs = invocation.getArgs();                        queryArgs[0] = newMs;                        System.out.println("打印新SQL语句" + newSql);                    }                 }            }        }        //继续执行逻辑        return invocation.proceed();    }      @Override    public Object plugin(Object o) {        //获取代理权        if (o instanceof Executor) {            //如果是Executor(执行增删改查操作),则拦截下来            return Plugin.wrap(o, this);        } else {            return o;        }    }         class MyBoundSqlSqlSource implements SqlSource {        private BoundSql boundSql;         public MyBoundSqlSqlSource(BoundSql boundSql) {            this.boundSql = boundSql;        }         @Override        public BoundSql getBoundSql(Object parameterObject) {            return boundSql;        }     }     private MappedStatement newMappedStatement(MappedStatement ms, SqlSource newSqlSource) {        MappedStatement.Builder builder = new                MappedStatement.Builder(ms.getConfiguration(), ms.getId(), newSqlSource, ms.getSqlCommandType());        builder.resource(ms.getResource());        builder.fetchSize(ms.getFetchSize());        builder.statementType(ms.getStatementType());        builder.keyGenerator(ms.getKeyGenerator());        if (ms.getKeyProperties() != null && ms.getKeyProperties().length > 0) {            builder.keyProperty(ms.getKeyProperties()[0]);        }        builder.timeout(ms.getTimeout());        builder.parameterMap(ms.getParameterMap());        builder.resultMaps(ms.getResultMaps());        builder.resultSetType(ms.getResultSetType());        builder.cache(ms.getCache());        builder.flushCacheRequired(ms.isFlushCacheRequired());        builder.useCache(ms.isUseCache());        return builder.build();    }      @Override    public void setProperties(Properties properties) {        //读取mybatis配置文件中属性    }  }

简单代码端解析:

① 拿出执行参数,里面有很多东西给我们用的,我本篇主要是用MappedStatement。大家可以发挥自己的想法。 打debug可以自己看看里面的东西。

//获取执行参数Object[] objects = invocation.getArgs();MappedStatement ms = (MappedStatement) objects[0];

Springboot如何实现自定义mybatis拦截器

 Springboot如何实现自定义mybatis拦截器

 ② 这里我主要是使用了从MappedStatement里面拿出来的id,做切割分别切出来 目前执行的sql的mapper是哪个,然后方法是哪个。

String mapperMethodAllName = ms.getId();int lastIndex = mapperMethodAllName.lastIndexOf(".");String mapperClassStr = mapperMethodAllName.substring(0, lastIndex);String mapperClassMethodStr = mapperMethodAllName.substring((lastIndex + 1));

③ 这就是我自己随便写的一下,我直接根据切割出来的mapper类找出里面的方法,主要是为了拿出每个方法的返回类型,因为 我这篇实践内容就是,判断出单个pojo接受的mapper方法,加个LIMIT 1 , 其他的 LIST 、SET 、 Page 等等这些,我暂时不做扩展。

Class<?> mapperClass = Class.forName(mapperClassStr);Method[] methods = mapperClass.getMethods();Class<?> returnType;

 ④这一段代码就是该篇的拓展逻辑了,从MappedStatement里面拿出 SqlSource,然后再拿出BoundSql,然后一顿操作 给加上 LIMIT 1  , 然后new 一个新的 MappedStatement,一顿操作塞回去invocation 里面

BoundSql boundSql = ms.getSqlSource().getBoundSql(objects[1]);String oldSql = boundSql.getSql().toLowerCase(Locale.CHINA).replace("[\\t\\n\\r]", " ");if (!oldSql.contains("LIMIT")){    String newSql = boundSql.getSql().toLowerCase(Locale.CHINA).replace("[\\t\\n\\r]", " ") + " LIMIT 1";    BoundSql newBoundSql = new BoundSql(ms.getConfiguration(), newSql,            boundSql.getParameterMappings(), boundSql.getParameterObject());    MappedStatement newMs = newMappedStatement(ms, new MyBoundSqlSqlSource(newBoundSql));    for (ParameterMapping mapping : boundSql.getParameterMappings()) {        String prop = mapping.getProperty();        if (boundSql.hasAdditionalParameter(prop)) {            newBoundSql.setAdditionalParameter(prop, boundSql.getAdditionalParameter(prop));        }    }    Object[] queryArgs = invocation.getArgs();    queryArgs[0] = newMs;    System.out.println("打印新SQL语句:" + newSql);}

OK,最后简单来试试效果 :
 

Springboot如何实现自定义mybatis拦截器

 Springboot如何实现自定义mybatis拦截器

最后再重申一下,本篇文章内容只是一个抛砖引玉,随便想的一些实践效果,大家可以按照自己的想法发挥。

最后再补充一个 解决 mybatis自定义拦截器和 pagehelper 拦截器 冲突导致失效的问题出现的

解决方案:

加上一个拦截器配置类 

MyDataSourceInterceptorConfig.java :

import org.apache.ibatis.session.SqlSessionFactory;import org.springframework.beans.factory.annotation.Autowired;import org.springframework.context.ApplicationListener;import org.springframework.context.event.ContextRefreshedEvent;import org.springframework.stereotype.Component; import java.util.List; @Componentpublic class MyDataSourceInterceptorConfig implements ApplicationListener<ContextRefreshedEvent> {     @Autowired    private MybatisInterceptor mybatisInterceptor;     @Autowired    private List<SqlSessionFactory> sqlSessionFactories;     @Override    public void onApplicationEvent(ContextRefreshedEvent contextRefreshedEvent) {        for (SqlSessionFactory factory : sqlSessionFactories) {            factory.getConfiguration().addInterceptor(mybatisInterceptor);        }    }}

最后最后再补充多一些,

文中 针对拦截的是Executor 这种接口的插件,

其实 还可以使用ParameterHandler、ResultSetHandler、StatementHandler。

每当执行这四种接口对象的方法时,就会进入拦截方法,然后我们可以根据不同的插件去拿不同的参数。

类似:

@Signature(method = "prepare", type = StatementHandler.class, args = {Connection.class,Integer.class})

然后可以做个转换,也是可以取出相关的 BoundSql 等等 :

if (!(invocation.getTarget() instanceof RoutingStatementHandler)){                return invocation.proceed();            }            RoutingStatementHandler statementHandler = (RoutingStatementHandler) invocation.getTarget();            BoundSql boundSql = statementHandler.getBoundSql();            String sql = boundSql.getSql().toUpperCase();

ParameterHandler、ResultSetHandler、StatementHandler、Executor ,不同的插件拦截,有不同的使用场景,想深入的看客们,可以深一下。

关于“Springboot如何实现自定义mybatis拦截器”这篇文章就分享到这里了,希望以上内容可以对大家有一定的帮助,使各位可以学到更多知识,如果觉得文章不错,请把它分享出去让更多的人看到。

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     807人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     351人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     314人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     433人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     221人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯