mybatis的插件机制

一、mybatis的插件介绍

关于mybatis的插件，我想大家也都用过，就比如最常用的逆向工程，根据表结构生成model，dao，xml文件，还有分页插件，那这些插件的工作原理是怎么样的呢，就比如分页插件，它为什么能改变我们在xml文件中写的sql语句，本文将带大家一起来了解mybatis的插件机制。（由于本人也是在不断学习，文中难免有错误或不足之处，还望指正，本文基于mybatis3.3.0版本），下面将围绕这几个方面

1、插件入口，即怎么把插件注入到mybatis代码里面

2、插件能拦截哪些类或哪些方法

3、举例简易分表插件原理

二、插件入口

在先了解前，我们来一段自定义mybatis插件的代码

import org.apache.ibatis.executor.statement.StatementHandler;
import org.apache.ibatis.plugin.*;

import java.sql.Connection;
import java.util.Properties;

@Intercepts(
        value = {
                @Signature(
                        type = StatementHandler.class,
                        method = "prepare",
                        args = {Connection.class} // 不同版本的prepare方法参数不一样，高版本的还有一个Integer参数
                )
        }
)
public class PluginDemo implements Interceptor {
    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        return invocation.proceed();
    }

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

    @Override
    public void setProperties(Properties properties) {

    }

要自定义mybatis插件，必须得实现Interceptor接口，这个接口有三个抽象方法

1、intercept，这个方法是mybatis的核心方法，要实现自定义逻辑，基本都是改造这个方法，其中invocation参数可以通过反射要获取原始方法和对应参数信息

2、plugin，它的作用是用来生成一个拦截对方，也就是代理对象，使得被代理的对象一定会经过intercept方法，通常都会使用mybatis提供的工具类Plugin来获取代理对象，如果有自己独特需求，可以自定义

3、setProperties，这个方法就是用来设置插件的一些属性

其中@intercepts注解就是用来标明拦截哪些类，哪些方法。

当我们脱离spring容器来使用mybatis的时候，我们通常是这样写代码的

 而在mybatisConfig.xml文件中，我们有配置数据库连接信息，插件，别名，mapper文件映射地址等信息，(温馨提示，这些配置有一定顺序，如果不按照顺序配置，则mybatis解析时会抛出异常，详细配置信息可以参考mbatis的dtd文件配置）既然在配置文件中配置了，那肯定也会被解析掉，如下图，在XmlConfigBuild中进行解析

解析后，插件信息会被存储到configuation中InterceptorChain集合里面，这个configuation在mybatis运行周期是一个单例的，它负责存储所有的配置信息

 

 最终，mybatis的插件信息完整的注入到了configuation里面。

二、mybaits插件能拦截哪些类或哪些方法

在正常开发流程中，我们基本都是通过先获取sqlSession，如果不采用自定义配置，在默认的sqlsession实现就是defaultSqlSession，在该类的一个方法里就隐含了融合插件，

其中会获取到executor，这个类是通过configuation获取的，

 最终融合插件的方法就是图中红框的代码。它会为原始的executor类生成代理类。从而你在执行executor类的一些方法时，比如query,update方法，会先生成对应的代理对象，myabtis采用的是jdk的动态代理，代理后，你执行executor类的query，update方法时会自动转接到你自定义的插件intercept方法里面，也可以理解为覆盖原来的方法。

通过这，我们大概也可以猜测的出，mybatis插件要拦截的类，很大原因在configuation类中有生成，也不带大家绕弯子了，来看截图

拦截statementHandler接口的实现类，它通过routingStatementHandler来实现路由，最终定向到prepareStatementHandler类

拦截parameterHandler。这个接口是用来封装参数用的，也就是最终将参数设置到sql里面

拦截ResultSetHandler接口，它主要用于处理sql运行返回的结果

 

 

 总结，mybatis可以拦截的类分四大类

一、executor，executor类可以说是执行sql的全过程，如组装参数，sql改造，结果处理，比较广泛，但实际用的不多

二、StatementHandler，这个是执行sql的过程，可以获取到待执行的sql，可用来改造sql，如分页，分表，最常拦截的类

三、paremeterHandler，这个用来拦截sql的参数，可以自定义参数组装规则

四、resultHandler，这个用来处理结果

三、简易版分表插件

import org.apache.ibatis.executor.statement.StatementHandler;
import org.apache.ibatis.mapping.BoundSql;
import org.apache.ibatis.plugin.*;
import org.apache.ibatis.reflection.MetaObject;
import org.apache.ibatis.reflection.SystemMetaObject;

import java.sql.Connection;
import java.util.Properties;

@Intercepts(
        value = {
                @Signature(
                        type = StatementHandler.class,
                        method = "prepare",
                        args = {Connection.class}
                )
        }
)
public class PluginDemo implements Interceptor {
    @Override
    public Object intercept(Invocation invocation) throws Throwable {

        /**
         * 获取被拦截的目前类，在这里是拦截了statementHandler，所有目前类也就是它
         * 通过这个类我们可以拿到待执行的sql语句，通常使用mataObject工具类来获取
         * 关于这个工具类，大家可自行了解，个人认为这个工具类很强大
          */
        StatementHandler statementHandler = (StatementHandler) invocation.getTarget();
        MetaObject metaObject = SystemMetaObject.forObject(statementHandler);
        /**
         * 先解释下为什么写成delegate.boundSql就可以拿到boundSql类
         * 从前面也可以得知，statementHandler的默认实现是routingStatementHandler。
         * 这个类有一个属性statementHandler，属性名就叫delegate，而这个属性的默认实现又是preparedStatementHandler
         * 后面这个类又有属性boundSql，所以，最终形成的写法就是delegate.boundSql。
         * 所以这也体现了MetaObject工具类的强大，可以通过实例传参，就可以根据属性名获取对应属性值
         */
        BoundSql boundSql = (BoundSql) metaObject.getValue("delegate.boundSql");

        // 待执行的sql，在这里也就是预编译后的sql，即参数位都是?号
        String sql = boundSql.getSql();
        /**
         * 既然拿到了预编译后的sql，那就可以按照你自己的想法为所欲为，如分页，按年分表等等
         * 分表的话，个人推荐druid的sql解析器，我认为还是不错的，大家可以自行了解
         * 最后改造完sql，别忘了把它设置回去
         * metaObject.setValue("delegate.boundSql.sql",sql);
         *  invocation.proceed，即原始方法的执行
         *  注意点就是，因为mybatis插件采用的是代理模式，所以如果存在多个插件，会形成多个代理
         *  你如果要拿到最原始的对象，还得进一步进行分解
         *  如：while(metaObject.getValue(""h) != null){
         *      Object obj = metaObject.getValue("h");
         *       ....
         *  }
         */
        return invocation.proceed();
    }

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

    @Override
    public void setProperties(Properties properties) {

    }
}

 -----------------------------------------------------------------------------------------------------------------------------分界线-------------------------------------------------------------------------------------

以上就是全部内容