簡(jiǎn)單了解mybatis攔截器實(shí)現(xiàn)原理及實(shí)例

/**
 * mybatis 自定義攔截器
 * 三步驟：
 * 1 實(shí)現(xiàn) {@link Interceptor} 接口
 * 2 添加攔截注解 {@link Intercepts}
 * 3 配置文件中添加攔截器
 *
 * 1 實(shí)現(xiàn) {@link Interceptor} 接口
 *   具體作用可以看下面代碼每個(gè)方法的注釋
 * 2 添加攔截注解 {@link Intercepts}
 *   mybatis 攔截器默認(rèn)可攔截的類型只有四種，即四種接口類型 Executor、StatementHandler、ParameterHandler 和 ResultSetHandler
 *   對(duì)于我們的自定義攔截器必須使用 mybatis 提供的注解來指明我們要攔截的是四類中的哪一個(gè)類接口
 *   具體規(guī)則如下：
 *     a：Intercepts 攔截器: 標(biāo)識(shí)我的類是一個(gè)攔截器
 *     b：Signature 署名: 則是指明我們的攔截器需要攔截哪一個(gè)接口的哪一個(gè)方法
 *       type  對(duì)應(yīng)四類接口中的某一個(gè)，比如是 Executor
 *       method 對(duì)應(yīng)接口中的哪類方法，比如 Executor 的 update 方法
 *       args  對(duì)應(yīng)接口中的哪一個(gè)方法，比如 Executor 中 query 因?yàn)橹剌d原因，方法有多個(gè)，args 就是指明參數(shù)類型，從而確定是哪一個(gè)方法
 * 3 配置文件中添加攔截器
 *   攔截器其實(shí)就是一個(gè) plugin，在 mybatis 核心配置文件中我們需要配置我們的 plugin ：
 *     <plugin interceptor="liu.york.mybatis.study.plugin.MyInterceptor">
 *       <property name="username" value="LiuYork"/>
 *       <property name="password" value="123456"/>
 *     </plugin>
 *
 * 攔截器順序
 * 1 不同攔截器順序：
 *   Executor -> ParameterHandler -> StatementHandler -> ResultSetHandler
 *
 * 2 對(duì)于同一個(gè)類型的攔截器的不同對(duì)象攔截順序：
 *   在 mybatis 核心配置文件根據(jù)配置的位置，攔截順序是 從上往下
 */
@Intercepts({
    @Signature(method = "update", type = Executor.class, args = {MappedStatement.class, Object.class}),
    @Signature(method = "query", type = StatementHandler.class, args = {Statement.class, ResultHandler.class})
})
public class MyInterceptor implements Interceptor {

  /**
   * 這個(gè)方法很好理解
   * 作用只有一個(gè)：我們不是攔截方法嗎，攔截之后我們要做什么事情呢？
   *   這個(gè)方法里面就是我們要做的事情
   *
   * 解釋這個(gè)方法前，我們一定要理解方法參數(shù) {@link Invocation} 是個(gè)什么鬼？
   * 1 我們知道，mybatis攔截器默認(rèn)只能攔截四種類型 Executor、StatementHandler、ParameterHandler 和 ResultSetHandler
   * 2 不管是哪種代理，代理的目標(biāo)對(duì)象就是我們要攔截對(duì)象，舉例說明：
   *   比如我們要攔截 {@link Executor#update(MappedStatement ms, Object parameter)} 方法，
   *   那么 Invocation 就是這個(gè)對(duì)象，Invocation 里面有三個(gè)參數(shù) target method args
   *     target 就是 Executor
   *     method 就是 update
   *     args  就是 MappedStatement ms, Object parameter
   *
   *  如果還是不能理解，我再舉一個(gè)需求案例：看下面方法代碼里面的需求
   *
   * 該方法在運(yùn)行時(shí)調(diào)用
   */
  @Override
  public Object intercept(Invocation invocation) throws Throwable {

    /*
     * 需求：我們需要對(duì)所有更新操作前打印查詢語(yǔ)句的 sql 日志
     * 那我就可以讓我們的自定義攔截器 MyInterceptor 攔截 Executor 的 update 方法，在 update 執(zhí)行前打印sql日志
     * 比如我們攔截點(diǎn)是 Executor 的 update 方法 ： int update(MappedStatement ms, Object parameter)
     *
     * 那當(dāng)我們?nèi)罩敬蛴〕晒χ?，我們是不是還需要調(diào)用這個(gè)query方法呢，如何如調(diào)用呢？
     * 所以就出現(xiàn)了 Invocation 對(duì)象，它這個(gè)時(shí)候其實(shí)就是一個(gè) Executor，而且 method 對(duì)應(yīng)的就是 query 方法，我們
     * 想要調(diào)用這個(gè)方法，只需要執(zhí)行 invocation.proceed()
     */

    /* 因?yàn)槲覕r截的就是Executor，所以我可以強(qiáng)轉(zhuǎn)為 Executor，默認(rèn)情況下，這個(gè)Executor 是個(gè) SimpleExecutor */
    Executor executor = (Executor)invocation.getTarget();

    /*
     * Executor 的 update 方法里面有一個(gè)參數(shù) MappedStatement，它是包含了 sql 語(yǔ)句的，所以我獲取這個(gè)對(duì)象
     * 以下是偽代碼，思路：
     * 1 通過反射從 Executor 對(duì)象中獲取 MappedStatement 對(duì)象
     * 2 從 MappedStatement 對(duì)象中獲取 SqlSource 對(duì)象
     * 3 然后從 SqlSource 對(duì)象中獲取獲取 BoundSql 對(duì)象
     * 4 最后通過 BoundSql#getSql 方法獲取 sql
     */
    MappedStatement mappedStatement = ReflectUtil.getMethodField(executor, MappedStatement.class);
    SqlSource sqlSource = ReflectUtil.getField(mappedStatement, SqlSource.class);
    BoundSql boundSql = sqlSource.getBoundSql(args);
    String sql = boundSql.getSql();
    logger.info(sql);

    /*
     * 現(xiàn)在日志已經(jīng)打印，需要調(diào)用目標(biāo)對(duì)象的方法完成 update 操作
     * 我們直接調(diào)用 invocation.proceed() 方法
     * 進(jìn)入源碼其實(shí)就是一個(gè)常見的反射調(diào)用 method.invoke(target, args)
     * target 對(duì)應(yīng) Executor對(duì)象
     * method 對(duì)應(yīng) Executor的update方法
     * args  對(duì)應(yīng) Executor的update方法的參數(shù)
     */

    return invocation.proceed();
  }

  /**
   * 這個(gè)方法也很好理解
   * 作用就只有一個(gè)：那就是Mybatis在創(chuàng)建攔截器代理時(shí)候會(huì)判斷一次，當(dāng)前這個(gè)類 MyInterceptor 到底需不需要生成一個(gè)代理進(jìn)行攔截，
   * 如果需要攔截，就生成一個(gè)代理對(duì)象，這個(gè)代理就是一個(gè) {@link Plugin}，它實(shí)現(xiàn)了jdk的動(dòng)態(tài)代理接口 {@link InvocationHandler}，
   * 如果不需要代理，則直接返回目標(biāo)對(duì)象本身
   *
   * Mybatis為什么會(huì)判斷一次是否需要代理呢？
   * 默認(rèn)情況下，Mybatis只能攔截四種類型的接口：Executor、StatementHandler、ParameterHandler 和 ResultSetHandler
   * 通過 {@link Intercepts} 和 {@link Signature} 兩個(gè)注解共同完成
   * 試想一下，如果我們開發(fā)人員在自定義攔截器上沒有指明類型，或者隨便寫一個(gè)攔截點(diǎn)，比如Object，那Mybatis瘋了，難道所有對(duì)象都去攔截
   * 所以Mybatis會(huì)做一次判斷，攔截點(diǎn)看看是不是這四個(gè)接口里面的方法，不是則不攔截，直接返回目標(biāo)對(duì)象，如果是則需要生成一個(gè)代理
   *
   * 該方法在 mybatis 加載核心配置文件時(shí)被調(diào)用
   */
  @Override
  public Object plugin(Object target) {
    /*
     * 看了這個(gè)方法注釋，就應(yīng)該理解，這里的邏輯只有一個(gè)，就是讓mybatis判斷，要不要進(jìn)行攔截，然后做出決定是否生成一個(gè)代理
     *
     * 下面代碼什么鬼，就這一句就搞定了？
     * Mybatis判斷依據(jù)是利用反射，獲取這個(gè)攔截器 MyInterceptor 的注解 Intercepts和Signature，然后解析里面的值，
     * 1 先是判斷要攔截的對(duì)象是四個(gè)類型中 Executor、StatementHandler、ParameterHandler、 ResultSetHandler 的哪一個(gè)
     * 2 然后根據(jù)方法名稱和參數(shù)(因?yàn)橛兄剌d)判斷對(duì)哪一個(gè)方法進(jìn)行攔截 Note：mybatis可以攔截這四個(gè)接口里面的任一一個(gè)方法
     * 3 做出決定，是返回一個(gè)對(duì)象呢還是返回目標(biāo)對(duì)象本身(目標(biāo)對(duì)象本身就是四個(gè)接口的實(shí)現(xiàn)類，我們攔截的就是這四個(gè)類型)
     *
     * 好了，理解邏輯我們寫代碼吧~~~ What !!! 要使用反射，然后解析注解，然后根據(jù)參數(shù)類型，最后還要生成一個(gè)代理對(duì)象
     * 我一個(gè)小白我怎么會(huì)這么高大上的代碼嘛，怎么辦？
     *
     * 那就是使用下面這句代碼吧 哈哈
     * mybatis 早就考慮了這里的復(fù)雜度，所以提供這個(gè)靜態(tài)方法來實(shí)現(xiàn)上面的邏輯
     */
    return Plugin.wrap(target, this);
  }

  /**
   * 這個(gè)方法最好理解，如果我們攔截器需要用到一些變量參數(shù)，而且這個(gè)參數(shù)是支持可配置的，
   * 類似Spring中的@Value("${}")從application.properties文件獲取
   * 這個(gè)時(shí)候我們就可以使用這個(gè)方法
   *
   * 如何使用？
   * 只需要在 mybatis 配置文件中加入類似如下配置，然后 {@link Interceptor#setProperties(Properties)} 就可以獲取參數(shù)
   *   <plugin interceptor="liu.york.mybatis.study.plugin.MyInterceptor">
   *      <property name="username" value="LiuYork"/>
   *      <property name="password" value="123456"/>
   *   </plugin>
   *   方法中獲取參數(shù)：properties.getProperty("username");
   *
   * 問題：為什么要存在這個(gè)方法呢，比如直接使用 @Value("${}") 獲取不就得了？
   * 原因是 mybatis 框架本身就是一個(gè)可以獨(dú)立使用的框架，沒有像 Spring 這種做了很多依賴注入的功能
   *
   * 該方法在 mybatis 加載核心配置文件時(shí)被調(diào)用 
   */
  @Override
  public void setProperties(Properties properties) {
    String username = properties.getProperty("username");
    String password = properties.getProperty("password");
    // TODO: 2019/2/28 業(yè)務(wù)邏輯處理...
  }
}

package org.apache.ibatis.plugin;

/**
 * Plugin 類其實(shí)就是一個(gè)代理類，因?yàn)樗鼘?shí)現(xiàn)了jdk動(dòng)態(tài)代理接口 InvocationHandler
 * 我們核心只需要關(guān)注兩個(gè)方法
 * wrap：
 *   如果看懂了代碼案例1的例子，那么這個(gè)方法很理解，這個(gè)方法就是 mybatis 提供給開發(fā)人員使用的一個(gè)工具類方法，
 *   目的就是幫助開發(fā)人員省略掉 反射解析注解 Intercepts 和 Signature，有興趣的可以去看看源碼 Plugin#getSignatureMap 方法
 *
 * invoke：
 *   這個(gè)方法就是根據(jù) wrap 方法的解析結(jié)果，判斷當(dāng)前攔截器是否需要進(jìn)行攔截，
 *   如果需要攔截：將 目標(biāo)對(duì)象+目標(biāo)方法+目標(biāo)參數(shù) 封裝成一個(gè) Invocation 對(duì)象，給我們自定義的攔截器 MyInterceptor 的 intercept 方法
 *          這個(gè)時(shí)候就剛好對(duì)應(yīng)上了上面案例1中對(duì) intercept 方法的解釋了，它就是我們要處理自己邏輯的方法，
 *          處理好了之后是否需要調(diào)用目標(biāo)對(duì)象的方法，比如上面說的 打印了sql語(yǔ)句，是否還要查詢數(shù)據(jù)庫(kù)呢？答案是肯定的
 *   如果不需要攔截：則直接調(diào)用目標(biāo)對(duì)象的方法
 *          比如直接調(diào)用 Executor 的 update 方法進(jìn)行更新數(shù)據(jù)庫(kù)
 *
 */
class Plugin implements InvocationHandler {

  public static Object wrap(Object target, Interceptor interceptor) {
    // 省略
  }

  @Override
  public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
    // 省略
  }
}