parent
ac8e453aa1
commit
b00d19ce1a
@ -0,0 +1,75 @@
|
||||
package com.platform.utils.page;
|
||||
|
||||
public class Page {
|
||||
/** 当前是第几页 （当前页） */
|
||||
// private int index = 0;
|
||||
private int currentPageNum = 1;
|
||||
|
||||
/** 每页显示多少条 （当前页前面已有多少条数据：一次查询返回记录条数）*/
|
||||
// private int pageSize = 10;
|
||||
private int limit = 20;
|
||||
|
||||
/** 总共的数据量 */
|
||||
// private int totle;
|
||||
private int totleSize;
|
||||
|
||||
/** 共有多少页 */
|
||||
private int totlePage;
|
||||
|
||||
/**
|
||||
* @return the currentPageNum
|
||||
*/
|
||||
public int getCurrentPageNum() {
|
||||
return currentPageNum;
|
||||
}
|
||||
|
||||
/**
|
||||
* @param currentPageNum the currentPageNum to set
|
||||
*/
|
||||
public void setCurrentPageNum(int currentPageNum) {
|
||||
this.currentPageNum = currentPageNum;
|
||||
}
|
||||
|
||||
/**
|
||||
* @return the limit
|
||||
*/
|
||||
public int getLimit() {
|
||||
return limit;
|
||||
}
|
||||
|
||||
/**
|
||||
* @param limit the limit to set
|
||||
*/
|
||||
public void setLimit(int limit) {
|
||||
this.limit = limit;
|
||||
}
|
||||
|
||||
/**
|
||||
* @return the totleSize
|
||||
*/
|
||||
public int getTotleSize() {
|
||||
return totleSize;
|
||||
}
|
||||
|
||||
/**
|
||||
* @param totleSize the totleSize to set
|
||||
*/
|
||||
public void setTotleSize(int totleSize) {
|
||||
this.totleSize = totleSize;
|
||||
}
|
||||
|
||||
/**
|
||||
* @return the totlePage
|
||||
*/
|
||||
public int getTotlePage() {
|
||||
return totlePage;
|
||||
}
|
||||
|
||||
/**
|
||||
* @param totlePage the totlePage to set
|
||||
*/
|
||||
public void setTotlePage(int totlePage) {
|
||||
this.totlePage = totlePage;
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,194 @@
|
||||
package com.platform.utils.page;
|
||||
|
||||
import java.util.Map;
|
||||
import java.util.Properties;
|
||||
|
||||
import org.apache.ibatis.plugin.Interceptor;
|
||||
import org.apache.ibatis.plugin.Invocation;
|
||||
|
||||
import java.sql.Connection;
|
||||
import java.sql.PreparedStatement;
|
||||
import java.sql.ResultSet;
|
||||
import java.sql.SQLException;
|
||||
import java.sql.Statement;
|
||||
import java.util.Properties;
|
||||
|
||||
import org.apache.ibatis.executor.resultset.ResultSetHandler;
|
||||
import org.apache.ibatis.executor.statement.StatementHandler;
|
||||
import org.apache.ibatis.mapping.BoundSql;
|
||||
import org.apache.ibatis.mapping.MappedStatement;
|
||||
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.reflection.SystemMetaObject;
|
||||
|
||||
import com.platform.utils.Bean2MapUtils;
|
||||
|
||||
@Intercepts({
|
||||
@Signature(type = StatementHandler.class, method = "prepare", args = {Connection.class}),
|
||||
@Signature(type = ResultSetHandler.class, method = "handleResultSets", args = {Statement.class})
|
||||
})
|
||||
public class PageInterceptor implements Interceptor {
|
||||
|
||||
|
||||
/** 存储所有语句名称 */
|
||||
/* HashMap<String, String> map_statement = new HashMap<String, String>();*/
|
||||
|
||||
/** 用户提供分页计算条数后缀 */
|
||||
private static final String SELECT_ID="page";
|
||||
|
||||
@Override
|
||||
public Object intercept(Invocation invocation) throws Throwable {
|
||||
|
||||
if (invocation.getTarget() instanceof StatementHandler) {
|
||||
StatementHandler statementHandler = (StatementHandler) invocation.getTarget();
|
||||
MetaObject metaStatementHandler = SystemMetaObject.forObject(statementHandler);
|
||||
MappedStatement mappedStatement=(MappedStatement) metaStatementHandler.getValue("delegate.mappedStatement");
|
||||
String selectId=mappedStatement.getId();
|
||||
// System.out.println(" ---------- selectId ---- : " + selectId);
|
||||
if(selectId.substring(selectId.lastIndexOf(".")+1).toLowerCase().endsWith(SELECT_ID)){
|
||||
BoundSql boundSql = (BoundSql) metaStatementHandler.getValue("delegate.boundSql");
|
||||
// 分页参数作为参数对象parameterObject的一个属性
|
||||
String sql = boundSql.getSql();
|
||||
Page co=(Page)(boundSql.getParameterObject());
|
||||
@SuppressWarnings("unchecked")
|
||||
Map<String, Object> map = Bean2MapUtils.convertBean(boundSql.getParameterObject());
|
||||
if (null == co) {
|
||||
co = new Page();
|
||||
}
|
||||
|
||||
// 重写 Sql语句
|
||||
String countSql=concatCountSql(sql);
|
||||
// while (countSql.endsWith(" ") || countSql.endsWith("\t")
|
||||
// || countSql.endsWith("\r") || countSql.endsWith("\n")) {
|
||||
// countSql = countSql.substring(0, countSql.length()-1);
|
||||
// }
|
||||
try{
|
||||
if(countSql.contains("?")) {
|
||||
StringBuffer sbr = new StringBuffer();
|
||||
String[] item = countSql.split("\\?");
|
||||
for (int i = 0; i < item.length; i++) {
|
||||
sbr.append(item[i]);
|
||||
String[] s = item[i].split("\\.");
|
||||
String[] re = s[s.length-1].split("\\=");
|
||||
if(map.keySet().contains(re[re.length-1])){
|
||||
sbr.append(map.get(re[re.length-1]));
|
||||
}
|
||||
else {
|
||||
sbr.append(" ");
|
||||
}
|
||||
}
|
||||
countSql = sbr.toString();
|
||||
}
|
||||
}catch(Exception e){
|
||||
e.printStackTrace();
|
||||
}
|
||||
String pageSql=concatPageMySql(sql,co);
|
||||
//
|
||||
// System.out.println("重写的 count sql :"+countSql);
|
||||
// System.out.println("重写的 select sql :"+pageSql);
|
||||
Connection connection = (Connection) invocation.getArgs()[0];
|
||||
|
||||
PreparedStatement countStmt = null;
|
||||
ResultSet rs = null;
|
||||
int totalCount = 0;
|
||||
try {
|
||||
countStmt = connection.prepareStatement(countSql);
|
||||
rs = countStmt.executeQuery();
|
||||
if (rs.next()) {
|
||||
totalCount = rs.getInt(1);
|
||||
}
|
||||
|
||||
} catch (Exception e) {
|
||||
e.printStackTrace();
|
||||
} finally {
|
||||
try {
|
||||
rs.close();
|
||||
countStmt.close();
|
||||
} catch (Exception e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
|
||||
metaStatementHandler.setValue("delegate.boundSql.sql", pageSql);
|
||||
//绑定count
|
||||
co.setTotleSize(totalCount);
|
||||
co.setTotlePage(totalCount/co.getLimit());
|
||||
if (co.getCurrentPageNum() > co.getTotlePage()) {
|
||||
co.setCurrentPageNum(co.getTotlePage());
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return invocation.proceed();
|
||||
}
|
||||
|
||||
/**
|
||||
* <一句话功能简述> 重写sql
|
||||
* <功能详细描述>
|
||||
* @param sql
|
||||
* @return
|
||||
* @see [类、类#方法、类#成员]
|
||||
*/
|
||||
public String concatCountSql(String sql){
|
||||
StringBuffer sb=new StringBuffer("select count(*) from ");
|
||||
// StringBuffer sb=new StringBuffer("select count(*) from (");
|
||||
// sb.append(sql).append(") as a");
|
||||
//
|
||||
sql=sql.toLowerCase();
|
||||
if(sql.lastIndexOf("order")>sql.lastIndexOf(")")){
|
||||
sb.append(sql.substring(sql.indexOf("from")+4, sql.lastIndexOf("order")));
|
||||
}else{
|
||||
sb.append(sql.substring(sql.indexOf("from")+4));
|
||||
}
|
||||
return sb.toString();
|
||||
}
|
||||
|
||||
/**
|
||||
* <一句话功能简述> 重写sql
|
||||
* <功能详细描述>
|
||||
* @param sql
|
||||
* @param page
|
||||
* @return
|
||||
* @see [类、类#方法、类#成员]
|
||||
*/
|
||||
public String concatPageMySql(String sql,Page page){
|
||||
StringBuffer sb=new StringBuffer();
|
||||
sb.append(sql);
|
||||
int size = page.getLimit();
|
||||
int index = page.getCurrentPageNum();
|
||||
if (index > 1) {
|
||||
index = index -1;
|
||||
sb.append(" limit ").append(size*index).append(" , ").append(size);
|
||||
}
|
||||
else {
|
||||
sb.append(" limit ").append(0).append(" , ").append(size);
|
||||
}
|
||||
return sb.toString();
|
||||
}
|
||||
|
||||
/**
|
||||
* 拦截类型StatementHandler
|
||||
*/
|
||||
@Override
|
||||
public Object plugin(Object target) {
|
||||
if (target instanceof StatementHandler) {
|
||||
return Plugin.wrap(target, this);
|
||||
} else {
|
||||
return target;
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void setProperties(Properties properties) {
|
||||
|
||||
}
|
||||
|
||||
public void setPageCount(){
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
Loading…
Reference in new issue