MyBatis 如何通过拦截器修改 SQL

目录

    • [1. 实现Interceptor接口](#1. 实现Interceptor接口)
    • [2. 注册配置文件](#2. 注册配置文件)

假如我们想实现多租户,或者在某些 SQL 后面自动拼接查询条件。在开发过程中大部分场景可能都是一个查询写一个 SQL 去处理,我们如果想修改最终 SQL 可以通过修改各个 mapper.xml 中的 SQL 来处理。

但实际过程中我们可能穿插着 ORM 和 SQL 的混合使用,隐藏在代码中不容易被发现,还有假如项目中有很多很多的 SQL 我们不可能一个一个的去修改解决。

这个时候我们就需要通过 mybatis 拦截 SQL 并且最终修改 SQL。

具体操作如下:

1. 实现Interceptor接口

实现Interceptor接口,并写相关逻辑

java 复制代码
package cn.source.framework.interceptor.impl;

import cn.source.common.core.domain.BaseEntity;
import cn.source.common.core.domain.entity.SysUser;
import cn.source.common.utils.DateUtils;
import cn.source.common.utils.SecurityUtils;
import org.apache.ibatis.executor.Executor;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.SqlCommandType;
import org.apache.ibatis.plugin.*;

import java.util.Properties;

/**
 * @Description: 自定义拦截器,注入创建人,创建日期,修改人,修改日期,企业id
 */
@Intercepts({
   @Signature(
       type = Executor.class,
       method = "update",
       args = {MappedStatement.class, Object.class}),
})
public class HandleBaseInfoInterceptor implements Interceptor {

     @Override
     public Object intercept(Invocation invocation) throws Throwable {
       Object[] args = invocation.getArgs();
       MappedStatement mappedStatement = (MappedStatement) args[0];
       SqlCommandType sqlCommandType = mappedStatement.getSqlCommandType();
       // 用户类不处理,会导致获取不到用户信息出错
       if (args[1] instanceof SysUser) {
         return invocation.proceed();
       }
       if (args[1] instanceof BaseEntity) {
         BaseEntity baseEntity = (BaseEntity) args[1];
         if (sqlCommandType == SqlCommandType.UPDATE) {
           baseEntity.setUpdateTime(DateUtils.getNowDate());
           baseEntity.setUpdateBy(SecurityUtils.getUsername());
         } else if (sqlCommandType == SqlCommandType.INSERT) {
           baseEntity.setCreateTime(DateUtils.getNowDate());
           baseEntity.setCreateBy(SecurityUtils.getUsername());
           // System.out.println("赋值企业ID:" + baseEntity.getCompanyId());
           baseEntity.setCompanyId(SecurityUtils.getLoginUser().getUser().getCompanyId());
         }
       }
       return invocation.proceed();
     }

     @Override
     public Object plugin(Object target) {
       return Plugin.wrap(target, this);
     }
    
     @Override
     public void setProperties(Properties properties) {
     }
        
}

2. 注册配置文件

将插件注册到 mybatis 的配置文件 mybatis-config.xml

xml 复制代码
<plugins>
    <!-- 自定义拦截器,注入企业id-->
	<plugin interceptor="cn.source.framework.interceptor.impl.HandleSelectInterceptor">   </plugin>
	<!-- 自定义拦截器,注入创建人,创建日期,修改人,修改日期,企业id-->
	<plugin interceptor="cn.source.framework.interceptor.impl.HandleBaseInfoInterceptor">  </plugin>
</plugins>
相关推荐
程序员的世界你不懂1 小时前
【Flask】测试平台开发,新增说明书编写和展示功能 第二十三篇
java·前端·数据库
自学也学好编程1 小时前
【数据库】Redis详解:内存数据库与缓存之王
数据库·redis
JAVA不会写2 小时前
在Mybatis plus中如何使用自定义Sql
数据库·sql
IT 小阿姨(数据库)2 小时前
PgSQL监控死元组和自动清理状态的SQL语句执行报错ERROR: division by zero原因分析和解决方法
linux·运维·数据库·sql·postgresql·centos
ChinaRainbowSea2 小时前
7. LangChain4j + 记忆缓存详细说明
java·数据库·redis·后端·缓存·langchain·ai编程
小马学嵌入式~3 小时前
嵌入式 SQLite 数据库开发笔记
linux·c语言·数据库·笔记·sql·学习·sqlite
Java小白程序员4 小时前
MyBatis基础到高级实践:全方位指南(中)
数据库·mybatis
Monly214 小时前
人大金仓:merge sql error, dbType null, druid-1.2.20
数据库·sql
不宕机的小马达4 小时前
【Mysql|第一篇】Mysql的安装与卸载、Navicat工具的使用
数据库·mysql
float_六七4 小时前
数据库连接池:性能优化的秘密武器
数据库·oracle·性能优化