MyBatis 怎样通过拦截器修改 SQL

如果我们想实现多租户，或者在某些 SQL 背面自动拼接查询条件。在开发过程中大部分场景可能都是一个查询写一个 SQL 行止理，我们如果想修改最终 SQL 可以通过修改各个 mapper.xml 中的 SQL 来处理。
但实际过程中我们可能穿插着 ORM 和 SQL 的混合使用，隐藏在代码中不轻易被发现，还有如果项目中有许多许多的 SQL 我们不可能一个一个的去修改办理。
这个时候我们就需要通过 mybatis 拦截 SQL 并且最终修改 SQL。

详细利用如下：
1. 实现Interceptor接口

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

1. package cn.source.framework.interceptor.impl;
3. import cn.source.common.core.domain.BaseEntity;
4. import cn.source.common.core.domain.entity.SysUser;
5. import cn.source.common.utils.DateUtils;
6. import cn.source.common.utils.SecurityUtils;
7. import org.apache.ibatis.executor.Executor;
8. import org.apache.ibatis.mapping.MappedStatement;
9. import org.apache.ibatis.mapping.SqlCommandType;
10. import org.apache.ibatis.plugin.*;
12. import java.util.Properties;
14. /**
15. * @Description: 自定义拦截器，注入创建人，创建日期，修改人，修改日期，企业id
16. */
17. @Intercepts({
18.    @Signature(
19.        type = Executor.class,
20.        method = "update",
21.        args = {MappedStatement.class, Object.class}),
22. })
23. public class HandleBaseInfoInterceptor implements Interceptor {
25.      @Override
26.      public Object intercept(Invocation invocation) throws Throwable {
27.        Object[] args = invocation.getArgs();
28.        MappedStatement mappedStatement = (MappedStatement) args[0];
29.        SqlCommandType sqlCommandType = mappedStatement.getSqlCommandType();
30.        // 用户类不处理，会导致获取不到用户信息出错
31.        if (args[1] instanceof SysUser) {
32.          return invocation.proceed();
33.        }
34.        if (args[1] instanceof BaseEntity) {
35.          BaseEntity baseEntity = (BaseEntity) args[1];
36.          if (sqlCommandType == SqlCommandType.UPDATE) {
37.            baseEntity.setUpdateTime(DateUtils.getNowDate());
38.            baseEntity.setUpdateBy(SecurityUtils.getUsername());
39.          } else if (sqlCommandType == SqlCommandType.INSERT) {
40.            baseEntity.setCreateTime(DateUtils.getNowDate());
41.            baseEntity.setCreateBy(SecurityUtils.getUsername());
42.            // System.out.println("赋值企业ID:" + baseEntity.getCompanyId());
43.            baseEntity.setCompanyId(SecurityUtils.getLoginUser().getUser().getCompanyId());
44.          }
45.        }
46.        return invocation.proceed();
47.      }
49.      @Override
50.      public Object plugin(Object target) {
51.        return Plugin.wrap(target, this);
52.      }
53.    
54.      @Override
55.      public void setProperties(Properties properties) {
56.      }
57.         
58. }

复制代码

2. 注册配置文件

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

1. <plugins>
2.     <!-- 自定义拦截器，注入企业id-->
3.         <plugin interceptor="cn.source.framework.interceptor.impl.HandleSelectInterceptor">   </plugin>
4.         <!-- 自定义拦截器，注入创建人，创建日期，修改人，修改日期，企业id-->
5.         <plugin interceptor="cn.source.framework.interceptor.impl.HandleBaseInfoInterceptor">  </plugin>
6. </plugins>

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。