Mybatis拦截器

MyBatis 允许你在已映射语句执行过程中的某一点进行拦截调用。默认情况下，MyBatis 允许使用插件来拦截的方法调用包括：

1. Executor (update, query, flushStatements, commit, rollback, getTransaction, close, isClosed)
2. ParameterHandler (getParameterObject, setParameters)
3. ResultSetHandler (handleResultSets, handleOutputParameters)
4. StatementHandler (prepare, parameterize, batch, update, query)

使用方法

1. 实现Interceptor接口，并指定想要拦截的方法签名args即可

@Intercepts({@Signature(type = Executor.class,method = "query",
        args = {MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class})
})
public class MyPageInterceptor implements Interceptor {
    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        System.out.println("将逻辑分页改为物理分页");
        Object[] args = invocation.getArgs();
        MappedStatement ms = (MappedStatement) args[0]; // MappedStatement
        BoundSql boundSql = ms.getBoundSql(args[1]); // Object parameter
        RowBounds rb = (RowBounds) args[2]; // RowBounds
        // RowBounds为空，无需分页
        if (rb == RowBounds.DEFAULT) {
            return invocation.proceed();
        }
    }
    @Override
    public Object plugin(Object target) {
        return Plugin.wrap(target, this);
    }
    @Override
    public void setProperties(Properties properties) {
    }
}

2. 全局xml相关配置

<!-- 控制全局缓存（二级缓存）-->
<setting name="cacheEnabled" value="true"/>
<plugins>
    <plugin interceptor="com.eason.MyPageInterceptor"></plugin>
</plugins>

这个拦截器是拦截Executor实例中的query方法，这里的Executor是负责执行底层映射语句的内部对象。
接下来就开始看源码啦。

源码解析

1. 编程式代码

String resource = "mybatis-config.xml";
InputStream inputStream = Resources.getResourceAsStream(resource); // 1
SqlSessionFactory sqlSessionFactory = new SqlSessionFactoryBuilder().build(inputStream);// 2
SqlSession session = sqlSessionFactory.openSession(); // ExecutorType.BATCH
try {
    BlogMapper mapper = session.getMapper(BlogMapper.class);
    Blog blog = mapper.selectBlogById(1);
    System.out.println(blog);
} finally {
    session.close();
}

首先看

InputStream inputStream = Resources.getResourceAsStream(resource); // 1
SqlSessionFactory sqlSessionFactory = new SqlSessionFactoryBuilder().build(inputStream);// 2

先获取配置文件输入流，传入build方法，build方法会根据配置信息生成DefaultSqlSessionFactory,

1.png

再看parser.parse()，解析过就不会再解析了，解析主要是在parseConfiguration(parser.evalNode("/configuration"));中，

public Configuration parse() {
  if (parsed) {
    throw new BuilderException("Each XMLConfigBuilder can only be used once.");
  }
  parsed = true;
  parseConfiguration(parser.evalNode("/configuration"));
  return configuration;
}

解析各个节点信息并且保存下来

2.png

再看

SqlSession session = sqlSessionFactory.openSession(); // ExecutorType.BATCH

默认会创建new DefaultSqlSession(configuration, executor, autoCommit);

3.png

这里的executor是SIMPLE类型，并且在创建Executor的时候会在所有自定义的拦截器链中再包装一次（如果自定了对应的Executor的拦截器的话）

4.png

5.png

会将当前executor包装一次，

6.png

7.png

BlogMapper mapper = session.getMapper(BlogMapper.class);

会从缓存的Map knownMappers中去拿到MapperProxyFactory对象，

8.png

这个knownMappers里面在前面解析配置文件的时候会以配置文件的namespace生成的class为key，MapperProxyFactory为value缓存起来。

9.png

10.png

MapperProxy实现了InvocationHandler接口，所以执行mapper.selectBlogById方法是在MapperProxy对象的invoke方法中被代理执行，最后执行mapperMethod.execute(sqlSession, args);

11.png

12.png

13.png

14.png

15.png

因为我开启了二级缓存，所以会先去二级缓存中去查找数据，先生存二级缓存的key

17.png

18.png

19.png

20.png

21.png

重要的来了

查询前先new StatementHandler对象，生成StatementHandler对象的过程中也会new ParameterHandler和ResultSetHandler对象，也会加入拦截器链中去，

22.png

23.png

24.png

25.png

26.png

27.png

28.png

29.png

30.png

31.png

32.png