MyBatis 如何通过拦截器修改 SQL
作者:源字节1号
- 2024-05-15 浙江
本文字数:1490 字
阅读完需:约 5 分钟

假如我们想实现多租户,或者在某些 SQL 后面自动拼接查询条件。在开发过程中大部分场景可能都是一个查询写一个 SQL 去处理,我们如果想修改最终 SQL 可以通过修改各个 mapper.xml 中的 SQL 来处理。
但实际过程中我们可能穿插着 ORM 和 SQL 的混合使用,隐藏在代码中不容易被发现,还有假如项目中有很多很多的 SQL 我们不可能一一的去修改解决。
这个时候我们就需要通过 mybatis 拦截 SQL 并且最终修改 SQL。具体操作如下:
一、实现Interceptor接口,并写相关逻辑
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 * @author: sourcebyte.vip * @date: 2023年10月18日 14:33 */@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) { }}
复制代码
二、将插件注册到 mybatis 的配置文件 mybatis-config.xml
<plugins> <!-- 自定义拦截器,注入企业id--> <plugin interceptor="cn.source.framework.interceptor.impl.HandleSelectInterceptor"> </plugin> <!-- 自定义拦截器,注入创建人,创建日期,修改人,修改日期,企业id--> <plugin interceptor="cn.source.framework.interceptor.impl.HandleBaseInfoInterceptor"> </plugin></plugins>复制代码
如若转载,请注明出处:开源字节 https://sourcebyte.vip/article/347.html
划线
评论
复制
发布于: 刚刚阅读数: 4
源字节1号
关注
一个着迷于技术又喜欢不断折腾的技术活跃者 2022-03-09 加入
一个着迷于技术又喜欢不断折腾的技术活跃者。喜欢并热爱编程,执着于努力之后所带来的美好生活!







评论