mybatis使用拦截器实现物理分页功能

困扰我很久，为什么在使用mybatis的rowbounds不启作用了，以前做的项目都可以，后来发现原来是因为数据库换了，拦截器无法配置 导致的 

 

 

定义抽象类

 

package com.odbpo.mybatis;

 

/**

 * @Title: Dialect.java

 */

 

public abstract class Dialect

{

    public static enum Type {

        MYSQL, ORACLE

    }

 

    public abstract String getLimitString(String sql, int skipResults, int maxResults);

}

 

定义抽象类的实现类

 

package com.odbpo.mybatis;

 

import org.apache.log4j.Logger;

 

/**

 * @Title: OracleDialect.java

 * @Description:

 */

 

public class OracleDialect extends Dialect

{

    protected static Logger log = Logger.getLogger(OracleDialect.class);

    @Override

    public String getLimitString(String sql, int offset, int limit)

    {

        sql = sql.trim();

        StringBuffer pagingSelect = new StringBuffer(sql.length() + 100);

        pagingSelect.append("select * from ( select row_.*, rownum rownum_ from ( ");

        pagingSelect.append(sql);

        pagingSelect.append(" ) row_ ) where rownum_ > ").append(offset).append(" and rownum_ <= ")

                .append(offset + limit);

        log.debug(pagingSelect.toString());

        return pagingSelect.toString();

    }

}

 

定义mybatis拦截器，--关键
 

 

package com.odbpo.mybatis;

 

/**

 * @Title: PageInterceptor.java

 * @Description: myBatis 分页工具

 */

 

import java.sql.Connection;

import java.util.Map;

import java.util.Properties;

 

import org.apache.ibatis.executor.parameter.DefaultParameterHandler;

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 org.apache.log4j.Logger;

 

@Intercepts(

{ @Signature(type = StatementHandler.class, method = "prepare", args =

{ Connection.class }) })

public class PaginationInterceptor implements Interceptor

{

 

    protected static Logger log = Logger.getLogger(PaginationInterceptor.class);

 

    /* (non-Javadoc)  

     * @see org.apache.ibatis.plugin.Interceptor#intercept(org.apache.ibatis.plugin.Invocation)  

     */

    @SuppressWarnings("unchecked")

    @Override

    public Object intercept(Invocation invocation) throws Throwable

    {

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

        MetaObject metaStatementHandler = MetaObject.forObject(statementHandler);

 

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

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

        {

            return invocation.proceed();

        }

 

        DefaultParameterHandler defaultParameterHandler = (DefaultParameterHandler) metaStatementHandler

                .getValue("delegate.parameterHandler");

        Map parameterMap = (Map) defaultParameterHandler.getParameterObject();

        Object sidx = parameterMap.get("_sidx");

        Object sord = parameterMap.get("_sord");

 

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

 

        if (sidx != null && sord != null)

        {

            originalSql = originalSql + " order by " + sidx + " " + sord;

        }

 

        Configuration configuration = (Configuration) metaStatementHandler

                .getValue("delegate.configuration");

 

        Dialect.Type databaseType = null;

        try

        {

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

                    .toUpperCase());

        }

        catch (Exception e)

        {

            log.error(e);  

        }

        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 ORACLE:

                dialect = new OracleDialect();

                break;

            case MYSQL://需要实现MySQL的分页逻辑    

                break;

 

        }

 

        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())

        {

            BoundSql boundSql = statementHandler.getBoundSql();

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

        }

        return invocation.proceed();

    }

 

    /* (non-Javadoc)  

     * @see org.apache.ibatis.plugin.Interceptor#plugin(java.lang.Object)  

     */

    @Override

    public Object plugin(Object target)

    {

        return Plugin.wrap(target, this);

    }

 

    /* (non-Javadoc)  

     * @see org.apache.ibatis.plugin.Interceptor#setProperties(java.util.Properties)  

     */

    @Override

    public void setProperties(Properties arg0)

    {

 

    }

 

}

 

 

定义分页的BEAN

 

package com.odbpo.mybatis;

 

import java.util.Collections;

import java.util.List;

 

@SuppressWarnings("unchecked")

public class Page

{    

    private int page;

    

    private int limit;

    

    private int total;

    

    private List rows = Collections.emptyList();

 

    public int getPage()

    {

        return page;

    }

 

    public void setPage(int page)

    {

        this.page = page;

    }

 

    public int getLimit()

    {

        return limit;

    }

 

    public void setLimit(int limit)

    {

        this.limit = limit;

    }

 

    public int getTotal()

    {

        return total;

    }

 

    public void setTotal(int total)

    {

        this.total = total;

    }

 

    public List getRows()

    {

        return rows;

    }

 

    public void setRows(List rows)

    {

        this.rows = rows;

    }

}