mybatis的拦截器interceptor源代码解析

mybatis支持拦截器，实现的原理就是利用JDK的动态代理。先前利用拦截器实现了分页功能，如今又利用拦截器实现日志记录的功能，感觉拦截器还是不错的，只是相对于spring3 mvc的拦截器感觉有些逊色。接下来，我着重分析一些拦截的使用。

【基本思路】拦截器在哪里拦截？什么情况下才会拦截代理？怎么代理呢？只要搞清楚这些，基本的拦截器功能也就了如指掌啦。

拦截器在哪里呢？mybatis到底提供几处可以拦截呢？请看下图，通过分析源码可知基本查询流程如下:

【简洁的流程】首先mybatis初始化加载的时候，利用MapperProxy代理自己的Mapper接口类，生成一个代理处理类，代理处理的逻辑都在invoke方法里，首先获取到目标类的接口declaringInterface，然后生成MapperMethod，（其中sqlSession是SqlSessionTemplate，是spring连接mybatis的模板类，是由spring负责生成的），生成了MapperMethod,直接调用MapperMethod可执行的方法execute就能获取执行结果，整个流程非常简洁，非常和spring3 mvc的servlet实现类似，也就是说，MapperMethod就是一个命令，执行命令获取执行结果，也就是标准的命令模式。整个流程代码如下：

public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
    if (method.getDeclaringClass() == Object.class) {
      return method.invoke(this, args);
    }
    final Class<?> declaringInterface = findDeclaringInterface(proxy, method);
    final MapperMethod mapperMethod = new MapperMethod(declaringInterface, method, sqlSession);
    final Object result = mapperMethod.execute(args);
    if (result == null && method.getReturnType().isPrimitive() 
       && !method.getReturnType().equals(Void.TYPE)) {
      throw new BindingException(
      "Mapper method '" + method.getName() + "' (" + method.getDeclaringClass() 
      + ") attempted to return null from a method with a primitive return type (" 
      + method.getReturnType() + ").");
    }
    return result;
  }

【分步分析】

MapperMethod

查看excute方法可知，就是根据不同的sql类型，交给sqlSession，也就是SqlSessionTemplate类的对象，主要是完成获取执行参数param和commandName，执行具体的操作，例如，result = sqlSession.selectList(commandName, param);。其中sql类型的获取可参考初始化的时候加载的setupCommandType()，初始化的时候还包括获取map的sqlID等

SqlSessionTemplate和SqlSessionInterceptor

利用eclipse的f4一下，可知SqlSessionTemplate的实际执行是交给它的代理类完成的。查看SqlSessionTemplate构造函数可知，是由SqlSessionInterceptor动态代理的，那么所有的处理逻辑都是在invoke方法里，从中可知，就是获取SqlSession,执行操作。其中SqlSessionUtils.getSqlSession静态方法调用外部类SqlSessionTemplate的SqlSessionFactory，ExecutorType，这些都是在mybatis加载SqlSessionTemplate构造函数初始化的。

DefaultSqlSession

这个就是从DefaultSqlSessionFactory工厂生成获取的，具体的加载流程就是在sqlSession.selectList静态方法里。

接着上面的查询流程可知，调用的selectList方法，通过MapperMethod方法中获取的commandName，获取到MappedStatement对象，主要负责的就是sql里xml信息等，具体执行交给executor，executor是由SqlSessionTemplate的ExecutorType决定的。

executor和Plugin

executor的获取是DefaultSqlSessionFactory生成DefaultSqlSession，加载的方法是在Configuration里newExecutor从中可知cacheEnabled默认为ture，追加缓存实现，具体的实现还是SimpleExecutor，属于静态代理，cacheEnabled只是实现了缓存功能。executor = (Executor) interceptorChain.pluginAll(executor);

表明可追加拦截器。查看代码可知，

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

就是调用拦截Interceptor的方法plugin,我们在plugin方法里一般都需要调用Plugin的静态方法wrap生成Plugin动态代理目标类，由于又是动态代理，实现逻辑都是在invoke方法里，查看可知，调用Interceptor的intercept方法，由Invoction封装目标类，调用方法，调用参数。因此，我们实现自己的拦截器的时候，都可从intercept方法里获取到拦截的对象，方法，参数等。这里其中涉及到两个问题就是拦截器什么情况下才会代理的，怎样代理呢。第一个问题，查看wrap方法可知，就是查看Map里查找是否设定代理啦，以代理对象的class做为key，代理方法的set集合作为value，就能判定是否要代理的。Map里的数据是遍历拦截器的注解Intercepts，实现很简单。怎样打理的，就是调用Interceptor的intercept方法，具体的实现都是在自定义的拦截器方法里。

StatementHandler,ParameterHandler, ResultSetHandler和Plugin

由SimpleExecutor的doQuery方法是具体的实现，获取StatementHandler，然后通过StatementHandler获取Statement，其中StatementHandler的获取又是在Configuration里newStatementHandler，同executor实现的类似，也是有RoutingStatementHandler代理实际的StatementHandler实现，通过StatementType获取具体的StatementHandler类，从默认配置可知，PreparedStatementHandler是实际代理的对象。这里也有executor = (Executor) interceptorChain.pluginAll(executor)可知同样可添加拦截器。其中构造StatementHandler的时候，查看BaseStatementHandler构造函数可知，ParameterHandler和ResultSetHandler也是有Configuration生成的，同样也可追加拦截器。

Statement和日志代理

由SimpleExecutor的doQuery方法调用prepareStatement可知，Statement获取是通过StatementHandler的prepare方法生成的，由于PreparedStatementHandler是具体的实现类，查看可知，instantiateStatement方法里是具体的获取Statement方法，由ConnectionLogger代理类调用PreparedStatementLogger.newInstance生成PreparedStatementLogger。其中ConnectionLogger是DefaultSqlSessionFactory里wrapConnection方法获取的。

【总结】

mybatis的拦截器都是在Configuration类的InterceptorChain负责调用位置，Plugin类负责生成代理，调用具体的拦截器实现类

mybatis的拦截器interceptor源代码解析

相关推荐