写点什么

MyBatis 如何通过拦截器修改 SQL

作者:源字节1号
  • 2024-05-15
    浙江
  • 本文字数:1490 字

    阅读完需:约 5 分钟

MyBatis如何通过拦截器修改SQL

假如我们想实现多租户,或者在某些 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

用户头像

源字节1号

关注

一个着迷于技术又喜欢不断折腾的技术活跃者 2022-03-09 加入

一个着迷于技术又喜欢不断折腾的技术活跃者。喜欢并热爱编程,执着于努力之后所带来的美好生活!

评论

发布
暂无评论
MyBatis如何通过拦截器修改SQL_开源_源字节1号_InfoQ写作社区