快捷導(dǎo)航

利用Mybatis?Plus實(shí)現(xiàn)一個SQL攔截器

更新時間：2023年05月09日 14:08:04 作者：KevinQ

SQL攔截器是一種用于攔截和修改Mybatis執(zhí)行的SQL語句的工具，通過使用SQL攔截器，開發(fā)人員可以在執(zhí)行SQL語句之前或之后對其進(jìn)行修改或記錄，本文就來借助一下Mybatis-Plus實(shí)現(xiàn)一個SQL攔截器吧

起源

最近公司要做多租戶，Mybatis-Plus的多租戶插件很好用，但是有一個場景是：字典表或者某些數(shù)據(jù)表，一些數(shù)據(jù)需要在各個租戶之間共享，而數(shù)據(jù)表本身又同時要實(shí)現(xiàn)多租戶數(shù)據(jù)隔離，比如字典表：性別等“男”/“女”基礎(chǔ)數(shù)據(jù)。

SQL攔截器是一種用于攔截和修改Mybatis執(zhí)行的SQL語句的工具。通過使用SQL攔截器，開發(fā)人員可以在執(zhí)行SQL語句之前或之后對其進(jìn)行修改或記錄，從而更好地控制和優(yōu)化數(shù)據(jù)庫操作。例如MyBatis-Plus的基礎(chǔ)分頁插件、多租戶插件就是SQL攔截器，那么，我們是否可以實(shí)現(xiàn)自己的SQL攔截器呢？

答案當(dāng)然是肯定的。

實(shí)現(xiàn)攔截器接口InnerInterceptor

InnerInterceptor 這個接口是MyBaitsPlus的攔截器接口類，實(shí)現(xiàn)它之后，并且通過MybatisPlusInterceptor配置后，就可以實(shí)現(xiàn)SQL執(zhí)行攔截。

MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();

// 多租戶攔截器
interceptor.addInnerInterceptor(new TenantLineInnerInterceptor(new TenantDatabaseInterceptor(tenantProperties)));
// 多租戶部分表數(shù)據(jù)共享攔截器
interceptor.addInnerInterceptor(new TenantDatabaseShareInterceptor(new TenantShareHandlerImpl(tenantProperties)));

那么如何實(shí)現(xiàn)該接口以及修改SQL呢？

下面是一個樣例：

import com.baomidou.mybatisplus.core.interceptor.InnerInterceptor;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.parser.SqlParserHelper;
import com.baomidou.mybatisplus.extension.plugins.inner.InnerInterceptorChain;
import net.sf.jsqlparser.parser.CCJSqlParserUtil;
import net.sf.jsqlparser.statement.Statement;
import net.sf.jsqlparser.statement.select.Select;

public class MyInterceptor implements InnerInterceptor {
    @Override
    ```
    @Override
    public void beforeQuery(Executor executor, MappedStatement ms, Object parameter, RowBounds rowBounds, ResultHandler resultHandler, BoundSql boundSql) throws SQLException {
        if (InterceptorIgnoreHelper.willIgnoreTenantLine(ms.getId())) return;
        PluginUtils.MPBoundSql mpBs = PluginUtils.mpBoundSql(boundSql);
        mpBs.sql(parserSingle(mpBs.sql(), null));
    }

}

通過修改beforeQuery函數(shù)中的代碼，即可修改執(zhí)行的SQL。

修改sql常用的工具類

JsqlParserSupport該類用于解析與修改SQL，并且MybatisPlus的多租戶插件更是直接繼承了該類。 CCJSqlParserUtil是JSqlParser中用于解析SQL語句的工具類。它提供了一些靜態(tài)方法，可以將SQL語句解析為Statement對象、Select對象、Update對象、Insert對象、Delete對象等。

常用的類還有Expression, Statement等類。

例如，使用Expression實(shí)現(xiàn)一個IN語句：

import net.sf.jsqlparser.expression.Expression;
import net.sf.jsqlparser.expression.LongValue;
import net.sf.jsqlparser.expression.StringValue;
import net.sf.jsqlparser.expression.operators.relational.InExpression;
import net.sf.jsqlparser.expression.operators.relational.ItemsList;
import net.sf.jsqlparser.expression.operators.relational.MultiExpressionList;
import net.sf.jsqlparser.expression.operators.relational.NamedParameter;
import net.sf.jsqlparser.expression.operators.relational.SubSelect;
import net.sf.jsqlparser.parser.CCJSqlParserUtil;

public class JSqlParserExample {
    public static void main(String[] args) throws Exception {
        // IN語句
        InExpression inExpr = new InExpression();
        inExpr.setLeftExpression(CCJSqlParserUtil.parseExpression("age"));
        ItemsList valueList = new MultiExpressionList();
        ((MultiExpressionList) valueList).addExpressions(new LongValue(18), new LongValue(21), new LongValue(30));
        inExpr.setItemsList(valueList);
        System.out.println(inExpr.toString());

        // 子查詢
        InExpression subqueryInExpr = new InExpression();
        subqueryInExpr.setLeftExpression(CCJSqlParserUtil.parseExpression("age"));
        SubSelect subquery = new SubSelect();
        subquery.setSelectBody(CCJSqlParserUtil.parse("SELECT age FROM users WHERE country = 'US'"));
        subqueryInExpr.setRightExpression(subquery);
        System.out.println(subqueryInExpr.toString());
    }
}

到此這篇關(guān)于利用Mybatis Plus實(shí)現(xiàn)一個SQL攔截器的文章就介紹到這了,更多相關(guān)Mybatis Plus SQL攔截器內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: