mybatis高级应用系列一:分页功能

Mybatis3.0出来已有段时间了,其实自己挺喜欢这样的一个持久化框架的,因为它简单实用,学习成本低。Mybatis3.0在整体结构上和ibatis2.X差不多,改进特性如下:

1.         解析xml引进了Xpath,不像ibatis2.x那样业余

2.         动态sql用OGNL解析

3.         加入注解配置sql,感觉没什么特别大的用途,我更喜欢xml方式,代码和配置分离,这也是ibatis的初衷

4.         加强了缓存这块的功能。Mybatis3.0把缓存模块分得更细,分为“持久实现(prepetual)”和“资源回收策略实现(eviction)”,更好的对缓存功能进行自己组合和扩展

5.         终于加入的plugin功能,就像struts一样,这样就可以很好的扩展内部的Executor,,StatementHandler….等内部对象功能。

一下只能想到这些了,总之改动后的代码结构清晰多了,如果各位看下源码的话,也是学习设计模式很好的课件,里面的代码用到了很多经典的设计模式,这在之后的系列学习中会讲到。

这一篇文章讲下分页的功能。

正如和ibatis以前的版本一样,mybatis的分页还是基于内存分页(查找出所有记录再取出偏移量的记录,如果jdbc驱支持absolute定位或者rs.next()到指定偏移位置),其实这样的分页实现基本没用,特别是大量数据情况下。

要想改变mybatis内部的分页行为,理论上只要把最终要执行的sql转变成对应的分页语句就行了。首先,我们熟悉下mybatis内部执行查询的动态交互图:

mybatis高级应用系列一:分页功能

 

可以很清楚的看到,真正生成Statement并执行sql的语句是StatementHandler接口的某个实现,这样就可以写个插件对StatementHandler的行为进行拦截。  

 
package study.mybatis.interceptor;

 

 

import java.sql.Connection;

import java.util.Properties;

 

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.apache.ibatis.executor.statement.StatementHandler;

import org.apache.ibatis.mapping.BoundSql;

import org.apache.ibatis.plugin.Interceptor;

import org.apache.ibatis.plugin.Intercepts;

import org.apache.ibatis.plugin.Invocation;

import org.apache.ibatis.plugin.Plugin;

import org.apache.ibatis.plugin.Signature;

import org.apache.ibatis.reflection.MetaObject;

import org.apache.ibatis.session.Configuration;

import org.apache.ibatis.session.RowBounds;

 

import study.mybatis.dialect.Dialect;

import study.mybatis.dialect.MySql5Dialect;

 

@Intercepts({@Signature(type=StatementHandler.class,method="prepare",args={Connection.class})})

public class PaginationInterceptor implements Interceptor{

 

    private final static Log log = LogFactory.getLog(PaginationInterceptor.class);

   

    @Override

    public Object intercept(Invocation invocation) throws Throwable {

       StatementHandler statementHandler = (StatementHandler)invocation.getTarget();

       BoundSql boundSql = statementHandler.getBoundSql();

       MetaObject metaStatementHandler = MetaObject.forObject(statementHandler);

       RowBounds rowBounds = (RowBounds)metaStatementHandler.getValue("delegate.rowBounds");

       if(rowBounds == null || rowBounds == RowBounds.DEFAULT){

           return invocation.proceed();

       }

       Configuration configuration = (Configuration)metaStatementHandler.getValue("delegate.configuration");

       Dialect.Type databaseType  = null;

       try{

           databaseType = Dialect.Type.valueOf(configuration.getVariables().getProperty("dialect").toUpperCase());

       } catch(Exception e){

           //ignore

       }

       if(databaseType == null){

           throw new RuntimeException("the value of the dialect property in configuration.xml is not defined : " + configuration.getVariables().getProperty("dialect"));

       }

       Dialect dialect = null;

       switch(databaseType){

           case MYSQL:

              dialect = new MySql5Dialect();

             

       }

      

       String originalSql = (String)metaStatementHandler.getValue("delegate.boundSql.sql");

       metaStatementHandler.setValue("delegate.boundSql.sql", dialect.getLimitString(originalSql, rowBounds.getOffset(), rowBounds.getLimit()) );

       metaStatementHandler.setValue("delegate.rowBounds.offset", RowBounds.NO_ROW_OFFSET );

       metaStatementHandler.setValue("delegate.rowBounds.limit", RowBounds.NO_ROW_LIMIT );

       if(log.isDebugEnabled()){

           log.debug("生成分页SQL : " + boundSql.getSql());

       }

       return invocation.proceed();

    }

 

    @Override

    public Object plugin(Object target) {

       return Plugin.wrap(target, this);

    }

 

    @Override

    public void setProperties(Properties properties) {

    }

 

}
 
 
 

里面最重要的三条语句:

 
metaStatementHandler.setValue("delegate.boundSql.sql", dialect.getLimitString(originalSql, rowBounds.getOffset(), rowBounds.getLimit()) );

metaStatementHandler.setValue("delegate.rowBounds.offset", RowBounds.NO_ROW_OFFSET );

metaStatementHandler.setValue("delegate.rowBounds.limit", RowBounds.NO_ROW_LIMIT );
 
改别要执行的sql语句,现在新设置的sql语句是物理分页的,所以现在不再需要mybatis进行额外的操作了,所以把rowBounds的偏移量恢复为初始值(offet:0,limit:Integer.max) 为了指定数据库版本,在mybatis全局配置文件设置dialect值
<properties>

    <property name="dialect" value="mysql"/>

</properties>

<plugins>

<plugin interceptor="study.mybatis.interceptor.PaginationInterceptor">

    </plugin>

</plugins>
 
 
完整代码请用svn从下面链接检出查看:

svn checkout http://20110311start.googlecode.com/svn/trunk/

下个系列将会讲下缓存的扩展应用。     -----------------------------分隔线---------------------------------------------最近有朋友用mybatis和spring整合的时候如果按照下列方式发现dialect属性不能设置成功:
 
<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
 <property name="dataSource" ref="dataSource" />
 <property name="typeAliasesPackage" value="com.***.web.domain" />
 <property name="plugins">
 <array>
 <ref bean="paginationInterceptor"/>
 </array>
 </property>
 <property name="configurationProperties">
 <props>
 <prop key="dialect">mysql</prop>
 </props>
 </property>
 </bean>
 
 
这个问题是org.mybatis.spring.SqlSessionFactoryBean这个代码里有个bug(244行,或者不是bug,是作者不想这么做法),如果感兴趣可以看下源码。配置文件做下如下修改:
<ref bean="paginationInterceptor"/>
 </array>
 </property>
 <!-- 这里不要,注释掉
 <property name="configurationProperties">
 <props>
 <prop key="dialect">mysql</prop>
 </props>
 </property>
 -->
 <!--  加上这个属性 -->
 <property name="configLocation" value="classpath:Mybatis_Configuration.xml"/>
 </bean>
 
Mybatis_Configuration.xml的配置如下:
<?xml version="1.0" encoding="UTF-8" ?>
 <!DOCTYPE configuration
 PUBLIC "-//mybatis.org//DTD Config 3.0//EN"
 "http://mybatis.org/dtd/mybatis-3-config.dtd">
 <configuration>
 
     <properties>
         <property name="dialect" value="oracle"/>
     </properties>
 
 </configuration>
 

相关推荐