Mybatis的分页功能很弱,它是基于内存的分页(查出所有记录再按偏移量和limit取结果),在大数据量的情况下这样的分页基本上是没有用的。本文基于插件,通过拦截StatementHandler重写sql语句,实现数据库的物理分页。本文适配的mybatis版本是3.2.2。

准备

为什么在StatementHandler拦截

深入浅出MyBatis-Sqlsession章节介绍了一次sqlsession的完整执行过程,从中可以知道sql的解析是在StatementHandler里完成的,所以为了重写sql需要拦截StatementHandler。

MetaObject简介

在我的实现里大量使用了MetaObject这个对象,因此有必要先介绍下它。MetaObject是Mybatis提供的一个的工具类,通过它包装一个对象后可以获取或设置该对象的原本不可访问的属性(比如那些私有属性)。它有个三个重要方法经常用到:

1)       MetaObject forObject(Object object,ObjectFactory objectFactory, ObjectWrapperFactory objectWrapperFactory)

2)       Object getValue(String name)

3)       void setValue(String name, Object value)

方法1)用于包装对象;方法2)用于获取属性的值(支持OGNL的方法);方法3)用于设置属性的值(支持OGNL的方法);

插件的原理

参见深入浅出Mybatis-插件原理。

有了上面这些基础知识的准备后,就可以我们的主题了。

拦截器签名

[java]  view plain  copy
  1. @Intercepts({@Signature(type =StatementHandler.class, method = "prepare", args ={Connection.class})})
  2. publicclass PageInterceptor implementsInterceptor {
  3. ...
  4. }

从签名里可以看出,要拦截的目标类型是StatementHandler(注意:type只能配置成接口类型),拦截的方法是名称为prepare参数为Connection类型的方法。

intercept的实现

[java]  view plain  copy
  1. public Object intercept(Invocation invocation) throws Throwable {
  2. StatementHandler statementHandler = (StatementHandler) invocation.getTarget();
  3. MetaObject metaStatementHandler = MetaObject.forObject(statementHandler,
  4. DEFAULT_OBJECT_FACTORY, DEFAULT_OBJECT_WRAPPER_FACTORY);
  5. // 分离代理对象链(由于目标类可能被多个拦截器拦截,从而形成多次代理,通过下面的两次循环
  6. // 可以分离出最原始的的目标类)
  7. while (metaStatementHandler.hasGetter("h")) {
  8. Object object = metaStatementHandler.getValue("h");
  9. metaStatementHandler = MetaObject.forObject(object, DEFAULT_OBJECT_FACTORY,
  10. DEFAULT_OBJECT_WRAPPER_FACTORY);
  11. }
  12. // 分离最后一个代理对象的目标类
  13. while (metaStatementHandler.hasGetter("target")) {
  14. Object object = metaStatementHandler.getValue("target");
  15. metaStatementHandler = MetaObject.forObject(object, DEFAULT_OBJECT_FACTORY,
  16. DEFAULT_OBJECT_WRAPPER_FACTORY);
  17. }
  18. Configuration configuration = (Configuration) metaStatementHandler.
  19. getValue("delegate.configuration");
  20. dialect = configuration.getVariables().getProperty("dialect");
  21. if (null == dialect || "".equals(dialect)) {
  22. logger.warn("Property dialect is not setted,use default 'mysql' ");
  23. dialect = defaultDialect;
  24. }
  25. pageSqlId = configuration.getVariables().getProperty("pageSqlId");
  26. if (null == pageSqlId || "".equals(pageSqlId)) {
  27. logger.warn("Property pageSqlId is not setted,use default '.*Page$' ");
  28. pageSqlId = defaultPageSqlId;
  29. }
  30. MappedStatement mappedStatement = (MappedStatement)
  31. metaStatementHandler.getValue("delegate.mappedStatement");
  32. // 只重写需要分页的sql语句。通过MappedStatement的ID匹配,默认重写以Page结尾的
  33. //  MappedStatement的sql
  34. if (mappedStatement.getId().matches(pageSqlId)) {
  35. BoundSql boundSql = (BoundSql) metaStatementHandler.getValue("delegate.boundSql");
  36. Object parameterObject = boundSql.getParameterObject();
  37. if (parameterObject == null) {
  38. throw new NullPointerException("parameterObject is null!");
  39. } else {
  40. // 分页参数作为参数对象parameterObject的一个属性
  41. PageParameter page = (PageParameter) metaStatementHandler
  42. .getValue("delegate.boundSql.parameterObject.page");
  43. String sql = boundSql.getSql();
  44. // 重写sql
  45. String pageSql = buildPageSql(sql, page);
  46. metaStatementHandler.setValue("delegate.boundSql.sql", pageSql);
  47. // 采用物理分页后,就不需要mybatis的内存分页了,所以重置下面的两个参数
  48. metaStatementHandler.setValue("delegate.rowBounds.offset",
  49. RowBounds.NO_ROW_OFFSET);
  50. metaStatementHandler.setValue("delegate.rowBounds.limit", RowBounds.NO_ROW_LIMIT);
  51. Connection connection = (Connection) invocation.getArgs()[0];
  52. // 重设分页参数里的总页数等
  53. setPageParameter(sql, connection, mappedStatement, boundSql, page);
  54. }
  55. }
  56. // 将执行权交给下一个拦截器
  57. return invocation.proceed();
  58. }

StatementHandler的默认实现类是RoutingStatementHandler,因此拦截的实际对象是它。RoutingStatementHandler的主要功能是分发,它根据配置Statement类型创建真正执行数据库操作的StatementHandler,并将其保存到delegate属性里。由于delegate是一个私有属性并且没有提供访问它的方法,因此需要借助MetaObject的帮忙。通过MetaObject的封装后我们可以轻易的获得想要的属性。

在上面的方法里有个两个循环,通过他们可以分离出原始的RoutingStatementHandler(而不是代理对象)。

前面提到,签名里配置的要拦截的目标类型是StatementHandler拦截的方法是名称为prepare参数为Connection类型的方法,而这个方法是每次数据库访问都要执行的。因为我是通过重写sql的方式实现分页,为了不影响其他sql(update或不需要分页的query),我采用了通过ID匹配的方式过滤。默认的过滤方式只对id以Page结尾的进行拦截(注意区分大小写),如下:

[html]  view plain  copy
  1. <select id="queryUserByPage" parameterType="UserDto" resultType="UserDto">
  2. <![CDATA[
  3. select * from t_user t where t.username = #{username}
  4. ]]>
  5. </select>

当然,也可以自定义拦截模式,在mybatis的配置文件里加入以下配置项:

[html]  view plain  copy
  1. <properties>
  2. <property name="dialect" value="mysql" />
  3. <property name="pageSqlId" value=".*Page$" />
  4. </properties>

其中,属性dialect指示数据库类型,目前只支持mysql和oracle两种数据库。其中,属性pageSqlId指示拦截的规则,以正则方式匹配。

sql重写

sql重写其实在原始的sql语句上加入分页的参数,目前支持mysql和oracle两种数据库的分页。

[java]  view plain  copy
  1. private String buildPageSql(String sql, PageParameter page) {
  2. if (page != null) {
  3. StringBuilder pageSql = new StringBuilder();
  4. if ("mysql".equals(dialect)) {
  5. pageSql = buildPageSqlForMysql(sql, page);
  6. } else if ("oracle".equals(dialect)) {
  7. pageSql = buildPageSqlForOracle(sql, page);
  8. } else {
  9. return sql;
  10. }
  11. return pageSql.toString();
  12. } else {
  13. return sql;
  14. }
  15. }

mysql的分页实现:

[java]  view plain  copy
  1. public StringBuilder buildPageSqlForMysql(String sql, PageParameter page) {
  2. StringBuilder pageSql = new StringBuilder(100);
  3. String beginrow = String.valueOf((page.getCurrentPage() - 1) * page.getPageSize());
  4. pageSql.append(sql);
  5. pageSql.append(" limit " + beginrow + "," + page.getPageSize());
  6. return pageSql;
  7. }

oracle的分页实现:

[java]  view plain  copy
  1. public StringBuilder buildPageSqlForOracle(String sql, PageParameter page) {
  2. StringBuilder pageSql = new StringBuilder(100);
  3. String beginrow = String.valueOf((page.getCurrentPage() - 1) * page.getPageSize());
  4. String endrow = String.valueOf(page.getCurrentPage() * page.getPageSize());
  5. pageSql.append("select * from ( select temp.*, rownum row_id from ( ");
  6. pageSql.append(sql);
  7. pageSql.append(" ) temp where rownum <= ").append(endrow);
  8. pageSql.append(") where row_id > ").append(beginrow);
  9. return pageSql;
  10. }

分页参数重写

有时候会有这种需求,就是不但要查出指定页的结果,还需要知道总的记录数和页数。我通过重写分页参数的方式提供了一种解决方案:

[java]  view plain  copy
  1. /**
  2. * 从数据库里查询总的记录数并计算总页数,回写进分页参数<code>PageParameter</code>,这样调用
  3. * 者就可用通过 分页参数<code>PageParameter</code>获得相关信息。
  4. *
  5. * @param sql
  6. * @param connection
  7. * @param mappedStatement
  8. * @param boundSql
  9. * @param page
  10. */
  11. private void setPageParameter(String sql, Connection connection, MappedStatement mappedStatement,
  12. BoundSql boundSql, PageParameter page) {
  13. // 记录总记录数
  14. String countSql = "select count(0) from (" + sql + ") as total";
  15. PreparedStatement countStmt = null;
  16. ResultSet rs = null;
  17. try {
  18. countStmt = connection.prepareStatement(countSql);
  19. BoundSql countBS = new BoundSql(mappedStatement.getConfiguration(), countSql,
  20. boundSql.getParameterMappings(), boundSql.getParameterObject());
  21. setParameters(countStmt, mappedStatement, countBS, boundSql.getParameterObject());
  22. rs = countStmt.executeQuery();
  23. int totalCount = 0;
  24. if (rs.next()) {
  25. totalCount = rs.getInt(1);
  26. }
  27. page.setTotalCount(totalCount);
  28. int totalPage = totalCount / page.getPageSize() + ((totalCount % page.getPageSize() == 0) ? 0 : 1);
  29. page.setTotalPage(totalPage);
  30. } catch (SQLException e) {
  31. logger.error("Ignore this exception", e);
  32. } finally {
  33. try {
  34. rs.close();
  35. } catch (SQLException e) {
  36. logger.error("Ignore this exception", e);
  37. }
  38. try {
  39. countStmt.close();
  40. } catch (SQLException e) {
  41. logger.error("Ignore this exception", e);
  42. }
  43. }
  44. }
  45. /**
  46. * 对SQL参数(?)设值
  47. *
  48. * @param ps
  49. * @param mappedStatement
  50. * @param boundSql
  51. * @param parameterObject
  52. * @throws SQLException
  53. */
  54. private void setParameters(PreparedStatement ps, MappedStatement mappedStatement, BoundSql boundSql,
  55. Object parameterObject) throws SQLException {
  56. ParameterHandler parameterHandler = new DefaultParameterHandler(mappedStatement, parameterObject, boundSql);
  57. parameterHandler.setParameters(ps);
  58. }

plugin的实现

[java]  view plain  copy
  1. public Object plugin(Object target) {
  2. // 当目标类是StatementHandler类型时,才包装目标类,否者直接返回目标本身,减少目标被代理的
  3. // 次数
  4. if (target instanceof StatementHandler) {
  5. return Plugin.wrap(target, this);
  6. } else {
  7. return target;
  8. }
  9. }

源码

下载地址:点击打开链接

深入浅出Mybatis-分页相关推荐

  1. 深入浅出mybatis分页

    MyBatis 分页插件 PageHelper PageHelper 如何使用? 如何使用分页插件 1:添加依赖 <dependency><groupId>com.github ...

  2. 【深入浅出MyBatis系列八】SQL自动生成插件

    为什么80%的码农都做不了架构师?>>>    #0 系列目录# 深入浅出MyBatis系列 [深入浅出MyBatis系列一]MyBatis入门 [深入浅出MyBatis系列二]配置 ...

  3. 【深入浅出MyBatis系列十一】缓存源码分析

    为什么80%的码农都做不了架构师?>>>    #0 系列目录# 深入浅出MyBatis系列 [深入浅出MyBatis系列一]MyBatis入门 [深入浅出MyBatis系列二]配置 ...

  4. 深入浅出MyBatis:MyBatis与Spring集成及实用场景

    为什么80%的码农都做不了架构师?>>>    本系列是「深入浅出MyBatis:技术原理与实践」书籍的总结笔记. 本篇是「深入浅出MyBatis」系列的最后一篇,主要介绍与Spri ...

  5. mybatis分页练手

    最近碰到个需求,要做个透明的mybatis分页功能,描述如下: 目标:搜索列表的Controller action要和原先保持一样,并且返回的json需要有分页信息,如: @ResponseBody ...

  6. 深入浅出 MyBatis 的一级、二级缓存机制

    一.MyBatis 缓存 缓存就是内存中的数据,常常来自对数据库查询结果的保存.使用缓存,我们可以避免频繁与数据库进行交互,从而提高响应速度. MyBatis 也提供了对缓存的支持,分为一级缓存和二级 ...

  7. springboot整合mybatis分页插件

    1.springboot版本为2.0.1,数据库为mysql,引入pagehelper的pom依赖 <!--mybatis分页插件--> <dependency><gro ...

  8. 解决使用mybatis分页插件PageHelper的一个报错问题

    解决使用mybatis分页插件PageHelper的一个报错问题 参考文章: (1)解决使用mybatis分页插件PageHelper的一个报错问题 (2)https://www.cnblogs.co ...

  9. 深入浅出Mybatis系列(四)---配置详解之typeAliases别名(mybatis源码篇)

    上篇文章<深入浅出Mybatis系列(三)---配置详解之properties与environments(mybatis源码篇)> 介绍了properties与environments, ...

  10. MyBatis分页插件PageHelper使用练习

    转载自:http://git.oschina.net/free/Mybatis_PageHelper/blob/master/wikis/HowToUse.markdown 1.环境准备: 分页插件p ...

最新文章

  1. Linux网络编程--TCP中的三次握手和四次挥手
  2. 位运算+取某一位+java_Java位运算小节
  3. 如何搭建高可用redis架构?
  4. 机会的度量:概率和分布
  5. Python :给类或者类的对象添加打印内容 def __repr__(self)
  6. 怎么实现登录之后跳转到登录之前的页面?SpringMVC+Freemarker
  7. all方法 手写promise_实现Promise.allSettled
  8. 为什么使用HashMap需要重写hashcode和equals方法_不同时重写equals和hashCode又会怎样?听听过来人的经验...
  9. python数据挖掘笔记】二十.KNN最近邻分类算法分析详解及平衡秤TXT数据集读取
  10. 爱创课堂每日一题第四十八天- html5有哪些新特性、移除了那些元素?
  11. java整人代码大全_整人代码vbs大全分享
  12. emwin emf格式视频生成环境搭建
  13. Unity 3D 中基础的三种传值技术(脚本组件,类静态字段,SendMessage)
  14. 淡水鱼生鱼片含有大量肝吸虫 可寄生于肝脏内30年
  15. Linux线程性能分析和CPU亲和力
  16. 转:张五常:比知识更重要的,是思维方式
  17. 云流量成为数据中心的王者 五年内全球超大规模IDC将不断出现
  18. MoveIt编程实现关节空间机械臂运动(逆运动学)
  19. 基于android平台的出题软件---- 每日30题
  20. 什么是冲突域,什么是广播域?区别又是什么

热门文章

  1. Unity中关于IphoneX的屏幕适配
  2. 或再被“转手”,家乐福中国还能“攀”上哪座靠山?
  3. JavaScript检测邮箱 e-mail
  4. java使用jacob将word,excel,ppt转成html
  5. tar linux 跳过解压,【linux命令】linux解压压缩命令tar详解以及压缩的时候如何跳过某一个压缩目录或文件...
  6. 看什么看!你的学习与就业我包了!!!
  7. 吸血鬼数字java_Java求吸血鬼数算法(通用)
  8. WIFi 开关控制实现-ESP8266 物联网 android studio arduino QT多线程服务器
  9. python找不到了_python包找不到
  10. sql 用户定义函数自动生成自增长ID