【核心代碼】

/**
* 攔截后要執行的方法
*/
@SuppressWarnings({ "rawtypes", "unchecked" })
public Object intercept(Invocation invocation) throws Throwable {
Object param = invocation.getArgs()[1];
//如果參數為空，直接直接返回
if (null == param) {
return invocation.proceed();
}
//獲取分頁參數對象
PageVO pageVO = getPageParam(param);
if (null != pageVO) {
//查詢符合條件的總記錄數
int totalCount = getQueryCount(
(Executor)invocation.getTarget(), (MappedStatement)invocation.getArgs()[0], 
parameter, (RowBounds)invocation.getArgs()[2], (ResultHandler<?>)invocation.getArgs()[3]);
//設置數據總條數
pageVO.setTotalSize(totalCount);
//查詢符合條件的分頁數據行記錄
if (totalCount <= 0) {
return Arrays.asList(new PageResultVO(pageVO, new ArrayList<Object>(0)));
}
return Arrays.asList(new PageResultVO(pageVO, (List<Object>)invocation.proceed()));
}
//返回查詢結果
return invocation.proceed();
}

/**
* 獲取分頁總數據條數
* 
* @param executor
* @param ms
* @param parameter
* @param rowBounds
* @return
*/
protected int getQueryCount(Executor executor, MappedStatement ms, Object param,
RowBounds rowBounds, ResultHandler<?> resultHandler) throws Throwable {
int totalCount = 0;
//查詢分頁總記錄數
List<Object> list = executor.query(
ms.getConfiguration().getMappedStatement(
new StringBuilder(ms.getId()).append(suffix).toString()), 
param, rowBounds, resultHandler);
if (null != list 
&& list.size() == 1) {
totalCount = (Integer)list.get(0);
}
return totalCount;
}

/**
* 獲取分頁參數
* 
* @param param
* @return
*/
@SuppressWarnings({ "unchecked"})
protected PageVO getPageParam(Object param) {
PageVO pageVO = null;
//如果參數為分頁對象則直接返回
if (param instanceof PageVO) {
pageVO = (PageVO)param;
}
//如果參數不是Map對象則直接返回
if(!(param instanceof Map)){
return pageVO;
}
//迭代獲取分頁對象
ParamMap<Object> paramMap = (ParamMap<Object>)param;
if (null != paramMap) {
for(Iterator<Map.Entry<String, Object>> iterator = paramMap.entrySet().iterator(); iterator.hasNext();){
Map.Entry<String, Object> paramEentry = iterator.next(); 
//獲取到分頁對象則直接跳出迭代
if (paramEentry.getValue() instanceof PageVO) {
pageVO = (PageVO)paramEentry.getValue();
break;
}
}
}
return pageVO;
}