为了账号安全,请及时绑定邮箱和手机立即绑定

spring-boot-2.0.3不一样系列之源码篇 - pageHelper分页,绝对有值得你看的地方

标签:
SpringBoot

前言

  开心一刻

    说实话,作为一个宅男,每次被淘宝上的雄性店主追着喊亲,亲,亲,这感觉真是恶心透顶,好像被强吻一样。。。。。。。。。更烦的是我每次为了省钱,还得用个女号,跟那些店主说:“哥哥包邮嘛么叽。”,“哥哥再便宜点呗,我钱不够了嘛,5555555,”。

知道后的店主

  路漫漫其修远兮,吾将上下而求索!

  github:https://github.com/youzhibing

  码云(gitee):https://gitee.com/youzhibing

问题背景

  用过pageHelper的都知道(没用过的感觉去google下),实现分页非常简单,service实现层调用dao(mapper)层之前进行page设置,mapper.xml中不处理分页,这样就够了,就能实现分页了,具体如下

    UserServiceImpl.java

复制代码

@Overridepublic PageInfo listUser(int pageNum, int pageSize) {
    PageHelper.startPage(pageNum, pageSize);
    List<User> users = userMapper.listUser();
    PageInfo pageInfo = new PageInfo(users);    return pageInfo;
}

复制代码

    UserMapper.xml

复制代码

<select id="listUser" resultType="User">
    SELECT
        id,username,password,salt,state,description
    FROM
        tbl_user</select>

复制代码

  哎我去,这样就实现分页了? 老牛皮了,这是为什么,这是怎么做到的? 凡事有果必有因,我们一起来看看这个因到底是什么

JDK的动态代理

  在进入正题之前了,我们先来做下准备,如果对动态代理很熟悉的直接略过往下看,建议还是看看,权且当做热身

  我们来看看JDK下的动态代理的具体实现:proxyDemo,运行ProxyTest的main方法,结果如下    

  可以看到我们对 张三 进行了增强处理,追加了后缀:_proxy

Mybatis sql执行流程

  当我们对JDK的动态代理有了一个基本认识之后了,我们再完成个一公里的慢跑:熟悉Mybatis的sql执行流程。流程图懒得画了,有人处理的很优秀了,我引用下

图片摘至《深入理解mybatis原理》 MyBatis的架构设计以及实例分析

分页源码解析

  业务代码中的PageHelper

    我们先来跟一跟业务代码中的PageHelper的代码

PageHelper.startPage(pageNum, pageSize);

    看它到底做了什么,如下图

    我们发现,进行了Page的相关设置后,将Page放到了当前线程中,没做其他的什么,那么分页肯定不是在这做的。

  PageHelper自动配置

    关于怎么找自动配置类,可参考:spring-boot-2.0.3不一样系列之源码篇 - springboot源码一,绝对有值得你看的地方,此时我们找到了PageHelperAutoConfiguration,源代码如下

/**
 * 自定注入分页插件
 *
 * @author liuzh */@Configuration
@ConditionalOnBean(SqlSessionFactory.class)
@EnableConfigurationProperties(PageHelperProperties.class)
@AutoConfigureAfter(MybatisAutoConfiguration.class)public class PageHelperAutoConfiguration {

    @Autowired    private List<SqlSessionFactory> sqlSessionFactoryList;

    @Autowired    private PageHelperProperties properties;    /**
     * 接受分页插件额外的属性
     *
     * @return
     */
    @Bean
    @ConfigurationProperties(prefix = PageHelperProperties.PAGEHELPER_PREFIX)    public Properties pageHelperProperties() {        return new Properties();
    }

    @PostConstruct    public void addPageInterceptor() {
        PageInterceptor interceptor = new PageInterceptor();
        Properties properties = new Properties();        //先把一般方式配置的属性放进去        properties.putAll(pageHelperProperties());        //在把特殊配置放进去,由于close-conn 利用上面方式时,属性名就是 close-conn 而不是 closeConn,所以需要额外的一步
        properties.putAll(this.properties.getProperties());
        interceptor.setProperties(properties);        for (SqlSessionFactory sqlSessionFactory : sqlSessionFactoryList) {            // 将PageInterceptor实例添加到了Configuration实例的interceptor链中            sqlSessionFactory.getConfiguration().addInterceptor(interceptor);
        }
    }

}

View Code

    在PageHelperAutoConfiguration的构造方法执行完之后,会执行addPageInterceptor方法,完成配置属性的注入,并将PageInterceptor实例添加到了Configuration实例的interceptorChain中。

    从Mybatis的SQL执行流程图中可以Mybatis的四大对象Executor、ParameterHandler、ResultSetHandler、StatementHandler,由他们一起合作完成SQL的执行,那么这四大对象是由谁创建的呢?没错,就是Mybatis的配置中心:Configuration,创建源代码如下:

  public ParameterHandler newParameterHandler(MappedStatement mappedStatement, Object parameterObject, BoundSql boundSql) {
    ParameterHandler parameterHandler = mappedStatement.getLang().createParameterHandler(mappedStatement, parameterObject, boundSql);
    parameterHandler = (ParameterHandler) interceptorChain.pluginAll(parameterHandler);    return parameterHandler;
  }  public ResultSetHandler newResultSetHandler(Executor executor, MappedStatement mappedStatement, RowBounds rowBounds, ParameterHandler parameterHandler,
      ResultHandler resultHandler, BoundSql boundSql) {
    ResultSetHandler resultSetHandler = new DefaultResultSetHandler(executor, mappedStatement, parameterHandler, resultHandler, boundSql, rowBounds);
    resultSetHandler = (ResultSetHandler) interceptorChain.pluginAll(resultSetHandler);    return resultSetHandler;
  }  public StatementHandler newStatementHandler(Executor executor, MappedStatement mappedStatement, Object parameterObject, RowBounds rowBounds, ResultHandler resultHandler, BoundSql boundSql) {
    StatementHandler statementHandler = new RoutingStatementHandler(executor, mappedStatement, parameterObject, rowBounds, resultHandler, boundSql);
    statementHandler = (StatementHandler) interceptorChain.pluginAll(statementHandler);    return statementHandler;
  }  public Executor newExecutor(Transaction transaction) {    return newExecutor(transaction, defaultExecutorType);
  }  public Executor newExecutor(Transaction transaction, ExecutorType executorType) {
    executorType = executorType == null ? defaultExecutorType : executorType;
    executorType = executorType == null ? ExecutorType.SIMPLE : executorType;
    Executor executor;    if (ExecutorType.BATCH == executorType) {
      executor = new BatchExecutor(this, transaction);
    } else if (ExecutorType.REUSE == executorType) {
      executor = new ReuseExecutor(this, transaction);
    } else {
      executor = new SimpleExecutor(this, transaction);
    }    if (cacheEnabled) {
      executor = new CachingExecutor(executor);
    }
    executor = (Executor) interceptorChain.pluginAll(executor);    return executor;
  }

View Code

    可以看到四大对象创建的最后,都会调用interceptorChain.pluginAll,我们来看看pluginAll方法做了什么

    其中Plugin的wrap方法要注意下

  public static Object wrap(Object target, Interceptor interceptor) {    // 获取PageInterceptor的Intercepts注解中@Signature的method,存放到Plugin的signatureMap中
    Map<Class<?>, Set<Method>> signatureMap = getSignatureMap(interceptor);
    Class<?> type = target.getClass();    // 获取目标对象实现的全部接口;四大对象是接口,都有默认的子类实现    // JDK的动态代理只支持接口
    Class<?>[] interfaces = getAllInterfaces(type, signatureMap);    if (interfaces.length > 0) {      return Proxy.newProxyInstance(
          type.getClassLoader(),
          interfaces,          new Plugin(target, interceptor, signatureMap));
    }    return target;
  }

View Code

    pluginAll其实就是给四大对象创建代理,一个Interceptor就会创建一层代理,而我们的PageInterceptor只是其中一层代理;我们接着往下看,Plugin继承了InvocationHandler,相当于上述:JDK的动态代理示例中的MyInvocationHandler,那么它的invoke方法肯定会被调用

  @Override  public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {    try {      // 获取PageInterceptor的Intercepts注解中@Signature的method
      Set<Method> methods = signatureMap.get(method.getDeclaringClass());      // 当methods包含目标方法时,调用PageInterceptor的intercept方法完成SQL的分页处理
      if (methods != null && methods.contains(method)) {        return interceptor.intercept(new Invocation(target, method, args));
      }      return method.invoke(target, args);
    } catch (Exception e) {      throw ExceptionUtil.unwrapThrowable(e);
    }
  }

View Code

  拦截器:PageInterceptor

    上述我们讲到了,当匹配时会进入到PageInterceptor的intercept方法中,在解读intercept方法之前,我们先来看看PageInterceptor类上的注解

复制代码

@Intercepts(
    {        // 相当于对Executor的query方法做拦截处理
        @Signature(type = Executor.class, method = "query", args = {MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class}),
        @Signature(type = Executor.class, method = "query", args = {MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class, CacheKey.class, BoundSql.class}),
    }
)

复制代码

    这就标明了PageInterceptor拦截的是Executor的query方法;还记上述wrap方法的

Map<Class<?>, Set<Method>> signatureMap = getSignatureMap(interceptor);

    吗?读取的就是@Intercepts下@Signature中的内容。我们接着看intercept

@Overridepublic Object intercept(Invocation invocation) throws Throwable {    try {
        Object[] args = invocation.getArgs();
        MappedStatement ms = (MappedStatement) args[0];
        Object parameter = args[1];
        RowBounds rowBounds = (RowBounds) args[2];
        ResultHandler resultHandler = (ResultHandler) args[3];
        Executor executor = (Executor) invocation.getTarget();
        CacheKey cacheKey;
        BoundSql boundSql;        //由于逻辑关系,只会进入一次
        if(args.length == 4){            //4 个参数时
            boundSql = ms.getBoundSql(parameter);
            cacheKey = executor.createCacheKey(ms, parameter, rowBounds, boundSql);
        } else {            //6 个参数时
            cacheKey = (CacheKey) args[4];
            boundSql = (BoundSql) args[5];
        }
        List resultList;        //调用方法判断是否需要进行分页,如果不需要,直接返回结果        // 此处会从当前线程取Page信息,还记得什么时候在哪将Page信息放进当前线程的吗?
        if (!dialect.skip(ms, parameter, rowBounds)) {            //反射获取动态参数
            String msId = ms.getId();
            Configuration configuration = ms.getConfiguration();
            Map<String, Object> additionalParameters = (Map<String, Object>) additionalParametersField.get(boundSql);            //判断是否需要进行 count 查询
            if (dialect.beforeCount(ms, parameter, rowBounds)) {
                String countMsId = msId + countSuffix;
                Long count;                //先判断是否存在手写的 count 查询
                MappedStatement countMs = getExistedMappedStatement(configuration, countMsId);                if(countMs != null){
                    count = executeManualCount(executor, countMs, parameter, boundSql, resultHandler);
                } else {
                    countMs = msCountMap.get(countMsId);                    //自动创建
                    if (countMs == null) {                        //根据当前的 ms 创建一个返回值为 Long 类型的 ms
                        countMs = MSUtils.newCountMappedStatement(ms, countMsId);
                        msCountMap.put(countMsId, countMs);
                    }
                    count = executeAutoCount(executor, countMs, parameter, boundSql, rowBounds, resultHandler);
                }                //处理查询总数                //返回 true 时继续分页查询,false 时直接返回
                if (!dialect.afterCount(count, parameter, rowBounds)) {                    //当查询总数为 0 时,直接返回空的结果
                    return dialect.afterPage(new ArrayList(), parameter, rowBounds);
                }
            }            //判断是否需要进行分页查询
            if (dialect.beforePage(ms, parameter, rowBounds)) {                //生成分页的缓存 key
                CacheKey pageKey = cacheKey;                //处理参数对象
                parameter = dialect.processParameterObject(ms, parameter, boundSql, pageKey);                //调用方言获取分页 sql
                String pageSql = dialect.getPageSql(ms, boundSql, parameter, rowBounds, pageKey);
                BoundSql pageBoundSql = new BoundSql(configuration, pageSql, boundSql.getParameterMappings(), parameter);                //设置动态参数
                for (String key : additionalParameters.keySet()) {
                    pageBoundSql.setAdditionalParameter(key, additionalParameters.get(key));
                }                //执行分页查询
                resultList = executor.query(ms, parameter, RowBounds.DEFAULT, resultHandler, pageKey, pageBoundSql);
            } else {                //不执行分页的情况下,也不执行内存分页
                resultList = executor.query(ms, parameter, RowBounds.DEFAULT, resultHandler, cacheKey, boundSql);
            }
        } else {            //rowBounds用参数值,不使用分页插件处理时,仍然支持默认的内存分页
            resultList = executor.query(ms, parameter, rowBounds, resultHandler, cacheKey, boundSql);
        }        return dialect.afterPage(resultList, parameter, rowBounds);
    } finally {
        dialect.afterAll();
    }
}

View Code

    其中会读取当前线程中的Page信息,根据Page信息来断定是否需要分页;而Page信息就是从我们的业务代码中存放到当前线程的。PageHelper的作者已经将intercept方法中的注释写的非常清楚了,相信大家都能看懂。

    到了此刻,相信大家都清楚了,还不清楚的静下心来好好捋一捋。

总结

  1、PageHelper属于Mybatis插件拓展,也可称拦截器拓展,是基于Mybatis的Interceptor实现;

  2、Page信息是在我们的业务代码中放到当前线程的,作为后续是否需要分页的条件;

  3、Mybatis创建mapper代理的过程(详情请看:Mybatis源码解析 - mapper代理对象的生成)中,也会创建四大对象的代理(有必要的话),而PageInterceptor对应的四大对象的代理会拦截Executor的query方法,将分页参数添加到目标SQL中;

  4、不管我们是否需要分页,只要我们集成了PageHelper,那么四大对象的代理实现中肯定包含了一层PageHelper的代理(可能是多层代理,包括其他第三方的Mybatis插件,或者我们自定义的Mybatis插件),如果当前线程中设置了Page,那么就表示需要分页,PageHelper就会读取当前线程中的Page信息,将分页条件添加到目标SQL中(Mysql是后面添加LIMIT,而Oracle则不一样),那么此时发送到数据库的SQL是有分页条件的,也就完成了分页处理;

  5、@Interceptors、@Signature以及Plugin类,三者配合起来,完成了分页逻辑的植入,Mybatis这么做便于拓展,使用起来更灵活,包容性更强;我们自定义插件的话,可以基于此,也可以抛弃这3个类,直接在plugin方法内部根据target实例的类型做相应的操作;个人推荐基于这3个来实现;

  6、Mybatis的Interceptor是基于JDK的动态代理,只能针对接口进行处理;另外,当我们进行Mybatis插件开发的时候,需要注意顺序问题,可能会与其他的Mybatis插件有冲突。

参考

  MyBatis拦截器原理探究

  《深入理解mybatis原理》 MyBatis的架构设计以及实例分析

作者:         youzhibing2904

原文出处:https://www.cnblogs.com/youzhibing/p/9603149.html  

点击查看更多内容
TA 点赞

若觉得本文不错,就分享一下吧!

评论

作者其他优质文章

正在加载中
  • 推荐
  • 评论
  • 收藏
  • 共同学习,写下你的评论
感谢您的支持,我会继续努力的~
扫码打赏,你说多少就多少
赞赏金额会直接到老师账户
支付方式
打开微信扫一扫,即可进行扫码打赏哦
今天注册有机会得

100积分直接送

付费专栏免费学

大额优惠券免费领

立即参与 放弃机会
意见反馈 帮助中心 APP下载
官方微信

举报

0/150
提交
取消