Mybatis之plugin插件设计原理

　　大多数框架，都支持插件，用户可通过编写插件来自行扩展功能，Mybatis也不例外。

　　我们从插件配置、插件编写、插件运行原理、插件注册与执行拦截的时机、初始化插件、分页插件的原理等六个方面展开阐述。

　一、插件配置

　　Mybatis的插件配置在configuration内部，初始化时，会读取这些插件，保存于Configuration对象的InterceptorChain中。

<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE configuration PUBLIC "-//mybatis.org//DTD Config 3.0//EN" "http://mybatis.org/dtd/mybatis-3-config.dtd">
<configuration>
    <plugins>
        <plugin interceptor="com.mybatis3.interceptor.MyBatisInterceptor">
            <property name="value" value="100" />
        </plugin>
    </plugins>
</configuration>

public class Configuration 
{
    protected final InterceptorChain interceptorChain = new     
    InterceptorChain();
}

org.apache.ibatis.plugin.InterceptorChain.java源码。

public class InterceptorChain {

  private final List<Interceptor> interceptors = new ArrayList<Interceptor>();

  public Object pluginAll(Object target) {
    for (Interceptor interceptor : interceptors) {
      target = interceptor.plugin(target);
    }
    return target;
  }

  public void addInterceptor(Interceptor interceptor) {
    interceptors.add(interceptor);
  }
  
  public List<Interceptor> getInterceptors() {
    return Collections.unmodifiableList(interceptors);
  }

}

　　上面的for循环代表了只要是插件，都会以责任链的方式逐一执行，所谓插件，其实就类似于拦截器。

二、插件的编写

　　插件必须实现org.apache.ibatis.plugin.Interceptor接口。

public interface Interceptor {
  
  Object intercept(Invocation invocation) throws Throwable;

  Object plugin(Object target);

  void setProperties(Properties properties);

}

-intercept()方法：执行拦截内容的地方，拦截目标对象的目标方法的执行

-plugin()方法：决定是否触发intercept()方法。  作用：包装目标对象，包装就是为目标对象创建一个代理对象

-setProperties()方法：给自定义的拦截器传递xml配置的属性参数。将插件注册时的property属性设置进来

下面自定义一个拦截器：

package com.atguigu.mybatis.dao;

import java.util.Properties;

import org.apache.ibatis.executor.statement.StatementHandler;
import org.apache.ibatis.plugin.Interceptor;
import org.apache.ibatis.plugin.Intercepts;
import org.apache.ibatis.plugin.Invocation;
import org.apache.ibatis.plugin.Plugin;
import org.apache.ibatis.plugin.Signature;
import org.apache.ibatis.reflection.MetaObject;
import org.apache.ibatis.reflection.SystemMetaObject;


/*
 * 完成插件签名：
 *     告诉Mybatis当前插件用来拦截哪个对象的哪个方法
 * 
 * */
@Intercepts(
        {
            @Signature(type=StatementHandler.class,method ="parameterize",args = java.sql.Statement.class )
        }
        )
public class MyFirstPlugin implements Interceptor
{
    /*
     * 拦截目标对象的目标方法的执行
     * */
    @Override
    public Object intercept(Invocation invocation) throws Throwable 
    {
        System.out.println("MyFirstPlugin的intercept方法执行，拦截方法："+invocation.getMethod());
        
        //动态改变一下sql的参数，查3号员工，这里偷梁换柱一下查询4号员工
        Object target = invocation.getTarget();
        System.out.println("当前拦截的对象："+target);
        //拿到：StatementHandler==>ParameterHandler==>parameterObject
        //拿到target（拦截对象）的元数据
        MetaObject metaObject = SystemMetaObject.forObject(target);
        Object value = metaObject.getValue("parameterHandler.parameterObject");
        System.out.println("sql语句用的参数是："+value);
        metaObject.setValue("parameterHandler.parameterObject", 4);
        
        //执行目标方法
        Object proceed = invocation.proceed();
        
        
        
        //返回执行后的返回值
        return proceed;
    }

    
    /*
     * 插件：
     * 作用：包装目标对象，包装就是为目标对象创建一个代理对象
     * */
    @Override
    public Object plugin(Object target) 
    {
        System.out.println("MyFirstPlugin的plugin方法执行，包装的对象（目标对象）："+target);
        
        
        //我们可以借助Plugin的wrap方法来使用当前的拦截器包装我们的目标对象
        Object wrap = Plugin.wrap(target, this);
        
        //返回的就是为当前target创建的动态代理
        return wrap;
    }

    /*
     * setProperties：
     *     将插件注册时的property属性设置进来
     * 
     * */
    @Override
    public void setProperties(Properties properties) 
    {
        System.out.println("插件配置的信息："+properties);
    }

}

2.1、为什么要写Annotation注解？注解都是什么含义？

　　Mybatis规定插件必须编写Annotation注解，是必须，而不是可选。@Intercepts注解：装载一个@Signature列表，一个@Signature其实就是一个需要拦截的方法封装。那么，一个拦截器要拦截多个方法，自然就是一个@Signature列表。

　　type = Executor.class, method = "query", args = { MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class }  

　　解释：要拦截Executor接口内的query()方法，参数类型为args列表。

2.2、Plugin.wrap(target, this)是干什么的？

　　使用JDK的动态代理，给target对象创建一个delegate代理对象，以此来实现方法拦截和增强功能，它会回调intercept()方法。

2.3、Mybatis可以拦截哪些接口对象？

　　Mybatis只能拦截ParameterHandler、ResultSetHandler、StatementHandler、Executor共4个接口对象内的方法。

　　重新审视interceptorChain.pluginAll()方法：该方法在创建上述4个接口对象时调用，其含义为给这些接口对象注册拦截器功能，注意是注册，而不是执行拦截。

　　拦截器执行时机：plugin()方法注册拦截器后，那么，在执行上述4个接口对象内的具体方法时，就会自动触发拦截器的执行，也就是插件的执行。

2.4、Invocation

public class Invocation {
  private Object target;
  private Method method;
  private Object[] args;
}

　　可以通过invocation来获取拦截的目标方法，以及执行目标方法。

2.5、分页插件原理

　　由于Mybatis采用的是逻辑分页，而非物理分页，那么，市场上就出现了可以实现物理分页的Mybatis的分页插件。  要实现物理分页，就需要对String sql进行拦截并增强，Mybatis通过BoundSql对象存储String sql，而BoundSql则由StatementHandler对象获取。

public interface StatementHandler {
    <E> List<E> query(Statement statement, ResultHandler resultHandler) throws SQLException;

    BoundSql getBoundSql();
}

public class BoundSql {
   public String getSql() {
    return sql;
  }
}

　　因此，就需要编写一个针对StatementHandler的query方法拦截器，然后获取到sql，对sql进行重写增强。