根據下面分頁的思想，很容易實現Mybitas的多租戶設計。
使用Mybatis提供的攔截器。對分頁的SQL語句通過封裝處理，處理成不同的分頁sql。
本例已經實現了對Mysql和Oracle的分頁功能。注意下面的引用包，不要引用錯了。
import java.sql.Connection;import java.sql.PreparedStatement;import java.sql.ResultSet;import java.sql.SQLException;import java.util.List;import java.util.Properties;import org.apache.ibatis.executor.parameter.ParameterHandler;import org.apache.ibatis.executor.statement.RoutingStatementHandler;import org.apache.ibatis.executor.statement.StatementHandler;import org.apache.ibatis.mapping.BoundSql;import org.apache.ibatis.mapping.MappedStatement;import org.apache.ibatis.mapping.ParameterMapping;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.scripting.defaults.DefaultParameterHandler;import com.yidao.utils.Page;import com.yidao.utils.ReflectHelper;/** * * 分頁攔截器，用於攔截需要進行分頁查詢的操作，然後對其進行分頁處理。 * 利用攔截器實現Mybatis分頁的原理： * 要利用JDBC對數據庫進行操作就必須要有一個對應的Statement對象，Mybatis在執行Sql語句前就會產生一個包含Sql語句的Statement對象，而且對應的Sql語句* 是在Statement之前產生的，所以我們就可以在它生成Statement之前對用來生成Statement的Sql語句下手。在Mybatis中Statement語句是通過RoutingStatementHandler對象的* prepare方法生成的。所以利用攔截器實現Mybatis分頁的一個思路就是攔截StatementHandler接口的prepare方法，然後在攔截器方法中把Sql語句改成對應的分頁查詢Sql語句，之後再調用* StatementHandler對象的prepare方法，即調用invocation.proceed()。 * 對於分頁而言，在攔截器裡面我們還需要做的一個操作就是統計滿足當前條件的記錄一共有多少，這是通過獲取到了原始的Sql語句後，把它改為對應的統計語句再利用Mybatis封裝好的參數和設* 置參數的功能把Sql語句中的參數進行替換，之後再執行查詢記錄數的Sql語句進行總記錄數的統計。 * */ @Intercepts({@Signature(type=StatementHandler.class,method="prepare",args={Connection.class})})public class PageInterceptor implements Interceptor { private String dialect = ""; //數據庫方言private String pageSqlId = ""; //mapper.xml中需要攔截的ID(正則匹配) public Object intercept(Invocation invocation) throws Throwable { //對於StatementHandler其實只有兩個實現類，一個是RoutingStatementHandler，另一個是抽像類BaseStatementHandler， //BaseStatementHandler有三個子類，分別是SimpleStatementHandler，PreparedStatementHandler和CallableStatementHandler， //SimpleStatementHandler是用於處理Statement的，PreparedStatementHandler是處理PreparedStatement的，而CallableStatementHandler是//處理CallableStatement的。 Mybatis在進行Sql語句處理的時候都是建立的RoutingStatementHandler，而在RoutingStatementHandler裡面擁有一個//StatementHandler類型的delegate屬性，RoutingStatementHandler會依據Statement的不同建立對應的BaseStatementHandler，即SimpleStatementHandler、 //PreparedStatementHandler或CallableStatementHandler，在RoutingStatementHandler裡面所有StatementHandler接口方法的實現都是調用的delegate對應的方法。 //我們在PageInterceptor類上已經用@Signature標記了該Interceptor只攔截StatementHandler接口的prepare方法，又因為Mybatis只有在建立RoutingStatementHandler的時候//是通過Interceptor的plugin方法進行包裹的，所以我們這裡攔截到的目標對象肯定是RoutingStatementHandler對象。 if(invocation.getTarget() instanceof RoutingStatementHandler){ RoutingStatementHandler statementHandler = (RoutingStatementHandler)invocation.getTarget(); StatementHandler delegate = (StatementHandler) ReflectHelper.getFieldValue(statementHandler, "delegate"); BoundSql boundSql = delegate.getBoundSql(); Object obj = boundSql.getParameterObject(); if (obj instanceof Page<?>) { Page<?> page = (Page<?>) obj; //通過反射獲取delegate父類BaseStatementHandler的mappedStatement屬性MappedStatement mappedStatement = (MappedStatement)ReflectHelper.getFieldValue(delegate, "mappedStatement"); //攔截到的prepare方法參數是一個Connection對象Connection connection = (Connection)invocation.getArgs()[0]; //獲取當前要執行的Sql語句，也就是我們直接在Mapper映射語句中寫的Sql語句String sql = boundSql.getSql(); //給當前的page參數對象設置總記錄數this.setTotalRecord(page, mappedStatement, connection); //獲取分頁Sql語句String pageSql = this.getPageSql(page, sql); //利用反射設置當前BoundSql對應的sql屬性為我們建立好的分頁Sql語句ReflectHelper.setFieldValue(boundSql, "sql", pageSql); } } return invocation.proceed(); } /** * 給當前的參數對象page設置總記錄數* * @param page Mapper映射語句對應的參數對象* @param mappedStatement Mapper映射語句* @param connection 當前的數據庫連接*/ private void setTotalRecord(Page<?> page, MappedStatement mappedStatement, Connection connection) { //獲取對應的BoundSql，這個BoundSql其實跟我們利用StatementHandler獲取到的BoundSql是同一個對象。 //delegate裡面的boundSql也是通過mappedStatement.getBoundSql(paramObj)方法獲取到的。 BoundSql boundSql = mappedStatement.getBoundSql(page); //獲取到我們自己寫在Mapper映射語句中對應的Sql語句String sql = boundSql.getSql(); //通過查詢Sql語句獲取到對應的計算總記錄數的sql語句String countSql = this.getCountSql(sql); //通過BoundSql獲取對應的參數映射List<ParameterMapping> parameterMappings = boundSql.getParameterMappings(); //利用Configuration、查詢記錄數的Sql語句countSql、參數映射關係parameterMappings和參數對象page建立查詢記錄數對應的BoundSql對象。 BoundSql countBoundSql = new BoundSql(mappedStatement.getConfiguration(), countSql, parameterMappings, page); //通過mappedStatement、參數對象page和BoundSql對象countBoundSql建立一個用於設定參數的ParameterHandler對象ParameterHandler parameterHandler = new DefaultParameterHandler(mappedStatement, page, countBoundSql); //通過connection建立一個countSql對應的PreparedStatement對象。 PreparedStatement pstmt = null; ResultSet rs = null; try { pstmt = connection.prepareStatement(countSql); //通過parameterHandler給PreparedStatement對象設置參數parameterHandler.setParameters(pstmt); //之後就是執行獲取總記錄數的Sql語句和獲取結果了。 rs = pstmt.executeQuery(); if (rs.next()) { int totalRecord = rs.getInt(1); //給當前的參數page對象設置總記錄數page.setTotalRecord(totalRecord); } } catch (SQLException e) { e.printStackTrace(); } finally { try { if (rs != null) rs.close(); if (pstmt != null) pstmt.close(); } catch (SQLException e) { e.printStackTrace(); } } } /** * 根據原Sql語句獲取對應的查詢總記錄數的Sql語句* @param sql * @return */ private String getCountSql(String sql) { int index = sql.indexOf("from"); return "select count(*) " + sql.substring(index); } /** * 根據page對象獲取對應的分頁查詢Sql語句，這裡只做了兩種數據庫類型，Mysql和Oracle * 其它的數據庫都沒有進行分頁* * @param page 分頁對象* @param sql 原sql語句* @return */ private String getPageSql(Page<?> page, String sql) { StringBuffer sqlBuffer = new StringBuffer(sql); if ("mysql".equalsIgnoreCase(dialect)) { return getMysqlPageSql(page, sqlBuffer); } else if ("oracle".equalsIgnoreCase(dialect)) { return getOraclePageSql(page, sqlBuffer); } return sqlBuffer.toString(); } /** * 獲取Mysql數據庫的分頁查詢語句* @param page 分頁對象* @param sqlBuffer 包含原sql語句的StringBuffer對象* @return Mysql數據庫分頁語句*/ private String getMysqlPageSql(Page<?> page, StringBuffer sqlBuffer) { //計算第一條記錄的位置，Mysql中記錄的位置是從0開始的。 // System.out.println("page:"+page.getPage()+"-------"+page.getRows()); int offset = (page.getPage() - 1) * page.getRows(); sqlBuffer.append(" limit ").append(offset).append(",").append(page.getRows()); return sqlBuffer.toString(); } /** * 獲取Oracle數據庫的分頁查詢語句* @param page 分頁對象* @param sqlBuffer 包含原sql語句的StringBuffer對象* @return Oracle數據庫的分頁查詢語句*/ private String getOraclePageSql(Page<?> page, StringBuffer sqlBuffer) { //計算第一條記錄的位置，Oracle分頁是通過rownum進行的，而rownum是從1開始的int offset = (page.getPage() - 1) * page.getRows() + 1; sqlBuffer.insert(0, "select u.*, rownum r from (").append(") u where rownum < ").append(offset + page.getRows()); sqlBuffer.insert(0, "select * from (").append(") where r >= ").append(offset); //上面的Sql語句拼接之後大概是這個樣子： //select * from (select u.*, rownum r from (select * from t_user) u where rownum < 31) where r >= 16 return sqlBuffer.toString(); } /** * 攔截器對應的封裝原始對象的方法*/ public Object plugin(Object arg0) { // TODO Auto-generated method stub if (arg0 instanceof StatementHandler) { return Plugin.wrap(arg0, this); } else { return arg0; } } /** * 設置註冊攔截器時設定的屬性*/ public void setProperties(Properties p) { } public String getDialect() { return dialect; } public void setDialect(String dialect) { this.dialect = dialect; } public String getPageSqlId() { return pageSqlId; } public void setPageSqlId(String pageSqlId) { this.pageSqlId = pageSqlId; } } xml配置：
<!-- MyBatis 接口編程配置--> <bean> <!-- basePackage指定要掃描的包，在此包之下的映射器都會被搜索到,可指定多個包，包與包之間用逗號或分號分隔--> <property name="basePackage" value="com.yidao.mybatis.dao" /> <property name="sqlSessionFactoryBeanName" value="sqlSessionFactory" /> </bean> <!-- MyBatis 分頁攔截器--> <bean id="paginationInterceptor"> <property name="dialect" value="mysql"/> <!-- 攔截Mapper.xml文件中，id包含query字符的語句--> <property name="pageSqlId" value=".*query$"/> </bean>
Page類
package com.yidao.utils;/**自己看看，需要什麼字段加什麼字段吧*/public class Page { private Integer rows; private Integer page = 1; private Integer totalRecord; public Integer getRows() { return rows; } public void setRows(Integer rows) { this.rows = rows; } public Integer getPage() { return page; } public void setPage(Integer page) { this.page = page; } public Integer getTotalRecord() { return totalRecord; } public void setTotalRecord(Integer totalRecord) { this.totalRecord = totalRecord; } } ReflectHelper類
package com.yidao.utils;import java.lang.reflect.Field;import org.apache.commons.lang3.reflect.FieldUtils;public class ReflectHelper { public static Object getFieldValue(Object obj , String fieldName ){ if(obj == null){ return null ; } Field targetField = getTargetField(obj.getClass(), fieldName); try { return FieldUtils.readField(targetField, obj, true ) ; } catch (IllegalAccessException e) { e.printStackTrace(); } return null ; } public static Field getTargetField(Class<?> targetClass, String fieldName) { Field field = null; try { if (targetClass == null) { return field; } if (Object.class.equals(targetClass)) { return field; } field = FieldUtils.getDeclaredField(targetClass, fieldName, true); if (field == null) { field = getTargetField(targetClass.getSuperclass(), fieldName); } } catch (Exception e) { } return field; } public static void setFieldValue(Object obj , String fieldName , Object value ){ if(null == obj){return;} Field targetField = getTargetField(obj.getClass(), fieldName); try { FieldUtils.writeField(targetField, obj, value) ; } catch (IllegalAccessException e) { e.printStackTrace(); } } }以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持武林網。