mybatis自定义分页拦截器

最近看了一下项目中代码，发现系统中使用的mybatis分页使用的是mybatis自带的分页，即使用RowBounds来进行分页，而这种分页是基于内存分页，即一次查出所有的数据，然后再返回分页需要的数据。断点跟踪mybatis的源码可以看到是这个方法中org.apache.ibatis.executor.resultset.FastResultSetHandler.skipRows(ResultSet, RowBounds)决定是将resultset的游标从哪个地方开始返回数据.

如果我们在mybatis执行sql之前，动态的将普通的sql语句换成分页的sql即可解决问题。那么我们就要找到mybatis是合适进行预编译sql的。

通过在程序中打断点，我们知道先是MapperProxy进行处理->然后是创建MapperMethod->然后执行mapperMethod的execute方法，在这个方法中判断我们是那种执行操作，比如增加，修改等，假如是查询操作(SELECT),然后就会根据mapperMethod中的sqlSessionTemplete参数，进行查询(selectList)操作，在该方法中进行SqlSession的获取，获取sqlSession时，会创建Exector对象（默认是SimpleExecutor）最终返回DefaultSqlSession, 然后经过一系列的操作，执行DefaultSqlSession中的selectList方法，在该方法中会执行executor的query方法，再经过一系列的操作，会执行org.apache.ibatis.executor.SimpleExecutor.doQuery(MappedStatement, Object, RowBounds, ResultHandler, BoundSql)这个方法，在该方法中会创建StatementHandler对象，跟踪断点知道创建的是RoutingStatementHandler，在RoutingStatementHandler的构造方法中，会根据StatementType判断创建那种Statement，默认是创建PreparedStatementHandler，在RoutingStatementHandler创建成功后，就会执行RoutingStatementHandler的prepare方法，在该方法中就会执行connection.prepareStatement(sql);最终返回java.sql.Statement对象。那么我们只要在connection.prepareStatement(sql)执行之前将不是分页的sql替换成分页的sql即可实现数据库的分页。

（上述流程建议，打一个断点跟踪一下mybatis的执行过程。）

     经过上述分析，我们知道sql语句是由org.apache.ibatis.executor.statement.StatementHandler.prepare(Connection)这个方法来执行预编译的。那么我们就可以写一个拦截器拦截prepare方法，在sql语句预编译之前改变sql,这正好可以借助mybatis的插件来实现。

在mybatis中编写一个插件需要实现org.apache.ibatis.plugin.Interceptor这个接口，然后在该接口上使用注解（@Intercepts）来标注要拦截的方法

解释一下插件中可能要用到的几个类：

MetaObject：mybatis提供的一个基于返回获取属性值的对象的类

    BoundSql : 在这个里面可以获取都要执行的sql和执行sql要用到的参数

    MappedStatement ： 这个可以得到当前执行的sql语句在xml文件中配置的id的值

    RowBounds : 是mybatis内存分页要用到的。

    ParameterHandler ： 是mybatis中用来替换sql中?出现的值的.

    那么我就说一下，我在我们项目中实现分页的实现：

    1.基于oracel分页

2.拦截select操作，其余的操作放行，即根据mappedStatement.getId()方法来判断

3.对分页中的参数进行一个判断，只要参数的类型是map，并且存在pageNo和pageSize这2个属性，即任务是分页sql（我们系统中的参数都是以map的方式传递的）

4.计算出总的记录数，然后放回到传递参数的map中

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

	@Override
	public Object intercept(Invocation invocation) throws Throwable {
		StatementHandler statementHandler = (StatementHandler) invocation.getTarget();
		MetaObject metaStatementHandler = MetaObject.forObject(statementHandler, SystemMetaObject.DEFAULT_OBJECT_FACTORY, SystemMetaObject.DEFAULT_OBJECT_WRAPPER_FACTORY);
		BoundSql boundSql = statementHandler.getBoundSql();
		MappedStatement mappedStatement = (MappedStatement) metaStatementHandler.getValue("delegate.mappedStatement");
		String id = mappedStatement.getId();
		System.out.println("sql语句的id : " + id);
		Object param = boundSql.getParameterObject();
		if (null != param && Map.class.isAssignableFrom(param.getClass())) {
			Map paramMap = (Map) param;
			Object _pageSize = paramMap.get("pageSize");
			Object _pageNo = paramMap.get("pageNo");
			if (_pageNo != null && _pageSize != null) {
				int pageSize = (Integer) _pageSize;
				int pageNo = (Integer) _pageNo;
				setTotalResult(boundSql, (Connection) invocation.getArgs()[0], metaStatementHandler, paramMap);
				String sql = getPageSql(pageSize, pageNo, boundSql);
				metaStatementHandler.setValue("delegate.boundSql.sql", sql);
				metaStatementHandler.setValue("delegate.rowBounds.offset", RowBounds.NO_ROW_OFFSET);
				metaStatementHandler.setValue("delegate.rowBounds.limit", RowBounds.NO_ROW_LIMIT);
			}
		}
		return invocation.proceed();
	}

        // 这个分页方法是借助于网上的
	private String getPageSql(int pageSize, int pageNo, BoundSql boundSql) {
		StringBuffer sql = new StringBuffer(boundSql.getSql());
		sql.insert(0, "select u.*, rownum r from (").append(") u where rownum < ").append(pageSize * pageNo);
		sql.insert(0, "select * from (").append(") where r >= ").append((pageSize - 1) * pageNo);
		return sql.toString();
	}
	private void setTotalResult(BoundSql boundSql, Connection conn, MetaObject metaObject, Map param) throws SQLException {
		String countSql = "select count(*) from ( " + boundSql.getSql() + " ) total";
		PreparedStatement prepareStatement = conn.prepareStatement(countSql);
		ParameterHandler parameterHandler = (ParameterHandler) metaObject.getValue("delegate.parameterHandler");
		parameterHandler.setParameters(prepareStatement); // 给sql语句设置参数
		ResultSet resultSet = prepareStatement.executeQuery();
		if (resultSet.next()) {
			param.put("total", resultSet.getObject(1));
		}
		if (resultSet != null) {
			resultSet.close();
		}
		if (prepareStatement != null) {
			prepareStatement.close();
		}
	}

	@Override
	public Object plugin(Object target) {
		return Plugin.wrap(target, this);
	}
	@Override
	public void setProperties(Properties properties) {

	}
}

插件写好之后，在配置文件中进行配置：

<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
	<property name="dataSource" ref="dataSource" />
	<property name="mapperLocations" value="sqlmap/*.xml"></property>
	<property name="plugins">
		<array>
			<bean class="com.study.mybatis.spring.interceptor.PageInterceptor" />  // 插件所在的类中的全路径
		</array>
	</property>
</bean>