mybatis拦截器使用实践

Posted 2023-04-16

参考技术A mybatis拦截器使用实践

一般数据表中都有创建/修改时间字段，新增/修改数据时，更新时间戳。可以通过拦截器来实现这一需求

Interceptor接口中，有3个方法

intercept：拦截器的具体业务逻辑

plugin：生成代理对象

setProperties：给拦截器设置一些属性

@Intercepts：声明这是一个拦截器

@Signature：要拦截的具体方法，type-拦截接口，method-拦截的方法，args-重载时根据参数列表确定要拦截的方法

此处我们拦截的是 Executor 接口中的update方法，mybatis拦截器可以拦截4大接口，其他的这里不做陈述

1）首先拦截的目标对象是否正确，不是则不做处理；

2）因为只处理insert/update的sql，所以做个判断；

3）通过反射获取到DO中声明的字段，一般创建/修改时间这类公共字段都会写在基础父类中，所以从父类中获取；

4）判断时insert还是update操作，insert则同时更新创建及修改时间，update只更新修改时间；

将MybatisGmtInterceptor拦截器注册到sqlSessionFactory中

只要是这个sqlSessionFactory生产的sqlSession，执行insert和update语句时，就会自动更新创建及修改时间

Mybatis——拦截器Interceptor

本篇介绍Mybatis拦截器/插件相关知识，包括相关的类及作用、拦截器有哪几种、拦截器可以在Mybatis的执行过程中的哪些节点起作用、拦截器如何使用、相关的API，最后附上代码示例。

1. Mybatis拦截器是什么？一般用途？

MyBatis允许使用者在映射语句执行过程中的某一些指定的节点进行拦截调用，通过织入拦截器，在不同节点修改一些执行过程中的关键属性，从而影响SQL的生成、执行和返回结果，如：来影响Mapper.xml到SQL语句的生成、执行SQL前对预编译的SQL执行参数的修改、SQL执行后返回结果到Mapper接口方法返参POJO对象的类型转换和封装等。

根据上面的对Mybatis拦截器作用的描述，可以分析其可能的用途；最常见的就是Mybatis自带的分页插件PageHelper或Rowbound参数，通过打印实际执行的SQL语句，发现我们的分页查询之前，先执行了COUNT(*)语句查询数量，然后再执行查询时修改了SQL语句即在我们写的SQL语句后拼接上了分页语句LIMIT(offset, pageSize)；

此外，实际工作中，可以使用Mybatis拦截器来做一些数据过滤、数据加密脱敏、SQL执行时间性能监控和告警等；既然要准备使用它，下面先来了解下其原理；

2. Mybatis执行过程

2.1 核心对象

从Mybatis关键代码的来看，MyBatis的主要的核心部件有以下几个：

• Configuration：初始化基础配置，比如MyBatis的别名等，一些重要的类型对象，如插件，映射器，ObjectFactory和typeHandler对象，MyBatis所有的配置信息都维持在Configuration对象之中。
• SqlSessionFactory：SqlSession工厂。
• SqlSession：作为MyBatis工作的主要顶层API，表示和数据库交互的会话，完成必要的数据库增删改查功能。
• Executor：MyBatis的内部执行器，它负责调用StatementHandler操作数据库，并把结果集通过ResultSetHandler进行自动映射，另外，它还处理二级缓存的操作。
• StatementHandler：MyBatis直接在数据库执行SQL脚本的对象。另外它也实现了MyBatis的一级缓存。
• ParameterHandler：负责将用户传递的参数转换成JDBC Statement所需要的参数。是MyBatis实现SQL入参设置的对象。
• ResultSetHandler：负责将JDBC返回的ResultSet结果集对象转换成List类型的集合。是MyBatis把ResultSet集合映射成POJO的接口对象。
• TypeHandler：负责Java数据类型和JDBC数据类型之间的映射和转换。
• MappedStatement：MappedStatement维护了一条<select|update|delete|insert>节点的封装。
• SqlSource ：负责根据用户传递的parameterObject，动态地生成SQL语句，将信息封装到BoundSql对象中，并返回。
• BoundSql：表示动态生成的SQL语句以及相应的参数信息。

2.2 执行过程分析

详细的执行过程可以看数据加密（二）Mybatis拦截器中的3.2 Mybatis整体执行流程，或Mybatis执行流程及关键代码走读；先看下图：

问题：通过mybatis的<if>标签动态修改SQL语句，为什么不能通过ParameterHandler？

通过上面的图可以回答这个问题，在执行ParameterHandler之前，BoundSQL对象已经生成了，即预编译的SQL语句已经生成了；可见，BoundSql生成SQL语句的时机在于 statementHandler 的创建，由MappedStatementHandler.getBoundSql(Object parameterObject)函数解析动态标签，生成静态SQL语句，早于parameterHandler和ResultSetHandler的时机；因此，只能通过影响parameterObject（暂时先将其看成一个Map，键值分别为参数名和参数值）的值，来干预对Mapper.xml标签的解析过程，从而改变生成的BoundSql对象；

3. 如何使用Mybatis拦截器？（使用步骤）

自定义一个mybatis的拦截器步骤包含：

• 定义一个实现org.apache.ibatis.plugin.Interceptor接口的拦截器类，并实现其中的方法。
• 添加@Intercepts注解，写上需要拦截的对象和方法，以及方法参数。
• 在mybatis的全局配置xml中配置插件plugin；对于去xml的Spring工程，显示的注册这个拦截器Bean即可；

3.1 添加注解

MyBatis拦截器默认可以拦截的类型只有四种，即四种接口类型Executor、StatementHandler、ParameterHandler和ResultSetHandler。对于我们的自定义拦截器必须使用MyBatis提供的@Intercepts注解来指明我们要拦截的是四种类型中的哪一种接口。

3.1.1 @Signature注解

@Intercepts	// 描述：标志该类是一个拦截器
@Signature 	// 描述：指明该拦截器需要拦截哪一个接口的哪一个方法

3.1.2 @Signature注解中的type属性

type; // 四种类型接口中的某一个接口，如Executor.class；
method; // 对应接口中的某一个方法名，比如Executor的query方法；
args; // 对应接口中的某一个方法的参数，比如Executor中query方法因为重载原因，有多个，args就是指明参数类型，从而确定是具体哪一个方法；

MyBatis拦截器默认会按顺序拦截以下的四个接口中的所有方法：

org.apache.ibatis.executor.Executor  //拦截执行器方法
org.apache.ibatis.executor.statement.StatementHandler  //拦截SQL语法构建处理
org.apache.ibatis.executor.parameter.ParameterHandler  //拦截参数处理
org.apache.ibatis.executor.resultset.ResultSetHandler  //拦截结果集处理

实际上，具体是拦截这四个接口对应的实现类：

org.apache.ibatis.executor.CachingExecutor
org.apache.ibatis.executor.statement.RoutingStatementHandler
org.apache.ibatis.scripting.defaults.DefaultParameterHandler
org.apache.ibatis.executor.resultset.DefaultResultSetHandler

这几个接口之间的关系大概是这样的：

3.2 实现org.apache.ibatis.plugin.Interceptor接口

实现Interceptor接口，主要是实现下面几个方法：intercept(Invocation invocation)、plugin(Object target) 、setProperties(Properties properties)；

3.2.1 intercept方法

进行拦截的时候要执行的方法。该方法参数Invocation类中有三个字段：

  private final Object target;
  private final Method method;
  private final Object[] args;

可通过这三个字段分别获取下面的信息：

Object target = invocation.getTarget(); //被代理对象
Method method = invocation.getMethod(); //代理方法
Object[] args = invocation.getArgs(); //方法参数

方法实现的结尾处，一般要真正调用被代理类的方法，即"return invocation.proceed();"

3.2.2 plugin方法

插件用于封装目标对象的，通过该方法我们可以返回目标对象本身，也可以返回一个它的代理，可以决定是否要进行拦截进而决定要返回一个什么样的目标对象，官方提供了示例："return Plugin.wrap(target, this);"；

Tips：可以在这个方法中提前进行拦截对象类型判断从而提高性能，如我们仅对Executor拦截，可以这么写：

    @Override
    public Object plugin(Object target) 
        // 只对要拦截制定类型的对象生成代理
        if(target instanceof Executor)
            // 调用插件
            return Plugin.wrap(target, this);
        
        return target;
    

MyBatis拦截器用到责任链模式+动态代理+反射机制；所有可能被拦截的处理类都会生成一个代理类，如果有N个拦截器，就会有N个代理，层层生成动态代理是比较耗性能的。而且虽然能指定插件拦截的位置，但这个是在执行方法时利用反射动态判断的，初始化的时候就是简单的把拦截器插入到了所有可以拦截的地方。所以尽量不要编写不必要的拦截器。另外我们可以在调用插件的地方添加判断，只要是当前拦截器拦截的对象才进行调用，否则直接返回目标对象本身，这样可以减少反射判断的次数，提高性能。

3.2.3 setProperties方法

如果我们拦截器需要用到一些变量参数，而且这个参数是支持可配置的，类似Spring中的@Value("$")从application.properties文件获取自定义变量属性，这个时候我们就可以使用这个方法。

参考下方代码，在Mybatis配置文件中配置插件时可以设置参数，在setProperties函数中调用 Properties.getProperty("param1") 方法可以得到配置的值：

<plugins>
    <plugin interceptor="com.xx.xx.xxxInterceptor">
        <property name="param1" value="value1"/>
    </plugin>
</plugins>

实际上是解析XML，将XML中注册的拦截器的配置带过来；

3.3 代码示例

 这里给一个拦截器的代码示例，如拦截Executor的查询和更新方法：

@Intercepts(@Signature(type = Executor.class, method = "update", args = MappedStatement.class, Object.class),
        @Signature(type = Executor.class, method = "query", args = MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class))
public class ExecutorEncryptInterceptor implements Interceptor 
 
    /* ... */
	
	Properties properties = null;
	
	/**
     * 拦截方法逻辑
     * 这里主要是通过反射去获取要执行的SQL相关信息，然后进行操作
     */
    @Override
    public Object intercept(Invocation invocation) throws Throwable 
 
		Object target = invocation.getTarget(); //被代理对象
        Method method = invocation.getMethod(); //代理方法
        Object[] args = invocation.getArgs(); //方法参数
		
		MappedStatement mappedStatement = (MappedStatement) args[0];
        Object parameterObject = args[1];
		
        // do something ...方法拦截前执行代码块
		
        Object result = invocation.proceed();
		
        // do something ...方法拦截后执行代码块
		
        return result;

    
 
    /**
     * 生成MyBatis拦截器代理对象
     */
    @Override
    public Object plugin(Object target) 
        if(target instanceof Executor)
            // 调用插件
            return Plugin.wrap(target, this);
        
        return target;
    
 
    /**
     * 设置插件属性（直接通过Spring的方式获取属性，所以这个方法一般也用不到）
     * 项目启动的时候数据就会被加载
     */
    @Override
    public void setProperties(Properties properties) 
        // 赋值成员变量，在其他方法使用
        this.properties = properties;
    
 
   /* ... */
   

参考：

MyBatis拦截器原理介绍 - 简书

MyBatis拦截器使用方法总结