Springboot自定義mybatis攔截器實現(xiàn)擴展

更新時間：2021年12月31日 16:37:51 作者：小目標青年

本文主要介紹了Springboot自定義mybatis攔截器實現(xiàn)擴展，文中通過示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下

前言

相信大家對攔截器并不陌生，對mybatis也不陌生。

有用過pagehelper的，那么對mybatis攔截器也不陌生了，按照使用的規(guī)則觸發(fā)sql攔截，幫我們自動添加分頁參數(shù) 。

那么今天，我們的實踐自定義mybatis攔截器也是如此，本篇文章實踐的效果：

針對一些使用單個實體類去接收返回結(jié)果的 mapper方法，我們攔截檢測，如果沒寫 LIMIT 1 ，我們將自動幫忙填充，達到查找單條數(shù)據(jù) 效率優(yōu)化的效果。

ps：當然，跟著該篇學會了這個之后，那么可以擴展的東西就多了，大家按照自己的想法或是項目需求都可以自己發(fā)揮。我該篇的實踐僅僅作為拋磚引玉吧。

正文

實踐的準備：

整合mybatis ，然后故意寫了3個查詢方法， 1個是list 列表數(shù)據(jù)，2個是單條數(shù)據(jù) 。

我們通過自己寫一個MybatisInterceptor實現(xiàn) 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.*;
 
 
 
/**
 * @Author JCccc
 * @Description
 * @Date 2021/12/14 16:56
 */
@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 {
        //獲取執(zhí)行參數(shù)
        Object[] objects = invocation.getArgs();
        MappedStatement ms = (MappedStatement) objects[0];
        //解析執(zhí)行sql的map方法,開始自定義規(guī)則匹配邏輯
        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);
                    }
 
                }
            }
        }
        //繼續(xù)執(zhí)行邏輯
        return invocation.proceed();
    }
 
 
    @Override
    public Object plugin(Object o) {
        //獲取代理權
        if (o instanceof Executor) {
            //如果是Executor（執(zhí)行增刪改查操作），則攔截下來
            return Plugin.wrap(o, this);
        } else {
            return o;
        }
    }
 
    /**
     * 定義一個內(nèi)部輔助類，作用是包裝 SQL
     */
    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配置文件中屬性
    }
 
 
}

簡單代碼端解析：

① 拿出執(zhí)行參數(shù)，里面有很多東西給我們用的，我本篇主要是用MappedStatement。大家可以發(fā)揮自己的想法。打debug可以自己看看里面的東西。

//獲取執(zhí)行參數(shù)
Object[] objects = invocation.getArgs();
MappedStatement ms = (MappedStatement) objects[0];

② 這里我主要是使用了從MappedStatement里面拿出來的id，做切割分別切出來目前執(zhí)行的sql的mapper是哪個，然后方法是哪個。

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

③ 這就是我自己隨便寫的一下，我直接根據(jù)切割出來的mapper類找出里面的方法，主要是為了拿出每個方法的返回類型，因為我這篇實踐內(nèi)容就是，判斷出單個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，最后簡單來試試效果：

最后再重申一下，本篇文章內(nèi)容只是一個拋磚引玉，隨便想的一些實踐效果，大家可以按照自己的想法發(fā)揮。

最后再補充一個解決 mybatis自定義攔截器和 pagehelper 攔截器沖突導致失效的問題出現(xiàn)的

解決方案：

加上一個攔截器配置類

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;
 
/**
 * @Author JCccc
 * @Description
 * @Date 2021/12/14 16:56
 */
@Component
public 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。

每當執(zhí)行這四種接口對象的方法時，就會進入攔截方法，然后我們可以根據(jù)不同的插件去拿不同的參數(shù)。

類似：

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

然后可以做個轉(zhuǎn)換，也是可以取出相關的 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攔截器實現(xiàn)擴展的文章就介紹到這了,更多相關Springboot mybatis攔截器內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章:

Java中增強for循環(huán)代碼示例
這篇文章主要給大家介紹了Java中增強for循環(huán)的相關資料,for/in循環(huán)就是JDK5.0中所謂的增強For循環(huán),它能對數(shù)組和集合進行遍歷，使用它會使用你的代碼短小而精煉的多,需要的朋友可以參考下
2023-10-10
Spring+Vue整合UEditor富文本實現(xiàn)圖片附件上傳的方法
這篇文章主要介紹了Spring+Vue整合UEditor富文本實現(xiàn)圖片附件上傳的方法，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧
2019-07-07
使用Java實現(xiàn)5種負載均衡算法實例
負載均衡指由多臺服務器以對稱的方式組成一個服務器集合,每臺服務器都具有等價的地位,都可以單獨對外提供服務而無須其他服務器的輔助,這篇文章主要給大家介紹了關于使用Java實現(xiàn)5種負載均衡算法的相關資料,需要的朋友可以參考下
2021-09-09
一文徹底弄懂零拷貝原理以及java實現(xiàn)
零拷貝(英語: Zero-copy) 技術是指計算機執(zhí)行操作時,CPU不需要先將數(shù)據(jù)從某處內(nèi)存復制到另一個特定區(qū)域,下面這篇文章主要給大家介紹了關于零拷貝原理以及java實現(xiàn)的相關資料,需要的朋友可以參考下
2021-08-08
Java實現(xiàn)有限狀態(tài)機的推薦方案分享
有限狀態(tài)機又稱有限狀態(tài)自動機,簡稱狀態(tài)機,是表示有限個狀態(tài)以及在這些狀態(tài)之間的轉(zhuǎn)移和動作等行為的數(shù)學模型,這篇文章主要給大家介紹了關于Java實現(xiàn)有限狀態(tài)機的推薦方案,需要的朋友可以參考下
2021-11-11
Java設計模式之觀察者模式
本文詳細講解了Java設計模式之觀察者模式，文中通過示例代碼介紹的非常詳細。對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下
2022-09-09
Java輸入學號、姓名、年齡并對其進行輸出的實現(xiàn)方法
這篇文章主要給大家介紹了關于Java輸入學號、姓名、年齡并對其進行輸出的實現(xiàn)方法,在計算機編程中,輸出學號和姓名是一個常見的任務,文中通過代碼介紹的非常詳細,需要的朋友可以參考下
2023-09-09
SpringBoot配置logback.xml 多環(huán)境的操作步驟
最近在研究springboot的日志，所以記錄一下，做一下總結(jié)，今天重點給大家介紹SpringBoot配置logback.xml 多環(huán)境的操作步驟，要實現(xiàn)多環(huán)境的配置，主要是依賴于springboot的application.yml文件去實現(xiàn)，感興趣的朋友跟隨小編一起看看吧
2021-05-05
MyBatis-Plus中公共字段的統(tǒng)一處理的實現(xiàn)
在開發(fā)中經(jīng)常遇到多個實體類有共同的屬性字段,這些字段屬于公共字段，本文主要介紹了MyBatis-Plus中公共字段的統(tǒng)一處理的實現(xiàn)，具有一定的參考價值，感興趣的可以了解一下
2023-08-08
MYSQL批量插入數(shù)據(jù)的實現(xiàn)代碼
非常的實現(xiàn)原理，代碼較多，建議大家仔細看看。
2008-10-10