|
|
|
|
@ -1,12 +1,26 @@
|
|
|
|
|
package com.platform.utils.page;
|
|
|
|
|
import java.sql.Connection;
|
|
|
|
|
import java.sql.Connection;
|
|
|
|
|
import java.sql.PreparedStatement;
|
|
|
|
|
import java.sql.ResultSet;
|
|
|
|
|
import java.util.Properties;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
import java.util.Map;
|
|
|
|
|
import java.util.Properties;
|
|
|
|
|
|
|
|
|
|
import org.apache.ibatis.plugin.Interceptor;
|
|
|
|
|
import org.apache.ibatis.plugin.Invocation;
|
|
|
|
|
import org.apache.ibatis.executor.Executor;
|
|
|
|
|
import org.apache.ibatis.executor.parameter.ParameterHandler;
|
|
|
|
|
import org.apache.ibatis.mapping.BoundSql;
|
|
|
|
|
import org.apache.ibatis.mapping.MappedStatement;
|
|
|
|
|
import org.apache.ibatis.mapping.MappedStatement.Builder;
|
|
|
|
|
import org.apache.ibatis.mapping.ParameterMapping;
|
|
|
|
|
import org.apache.ibatis.mapping.SqlSource;
|
|
|
|
|
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.session.ResultHandler;
|
|
|
|
|
import org.apache.ibatis.session.RowBounds;
|
|
|
|
|
|
|
|
|
|
import java.sql.Connection;
|
|
|
|
|
import java.sql.PreparedStatement;
|
|
|
|
|
import java.sql.ResultSet;
|
|
|
|
|
import java.sql.SQLException;
|
|
|
|
|
@ -24,8 +38,10 @@ 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 org.apache.ibatis.scripting.defaults.DefaultParameterHandler;
|
|
|
|
|
import org.apache.log4j.Logger;
|
|
|
|
|
|
|
|
|
|
import com.platform.utils.Bean2MapUtils;
|
|
|
|
|
import com.platform.utils.Configs;
|
|
|
|
|
|
|
|
|
|
@Intercepts({
|
|
|
|
|
@Signature(type = StatementHandler.class, method = "prepare", args = {Connection.class}),
|
|
|
|
|
@ -33,6 +49,8 @@ import com.platform.utils.Bean2MapUtils;
|
|
|
|
|
})
|
|
|
|
|
public class PageInterceptor implements Interceptor {
|
|
|
|
|
|
|
|
|
|
@SuppressWarnings("static-access")
|
|
|
|
|
public static Logger log = Configs.CONSOLE_LOGGER.getLogger(PageInterceptor.class);
|
|
|
|
|
|
|
|
|
|
/** 存储所有语句名称 */
|
|
|
|
|
/* HashMap<String, String> map_statement = new HashMap<String, String>();*/
|
|
|
|
|
@ -54,49 +72,21 @@ public class PageInterceptor implements Interceptor {
|
|
|
|
|
// 分页参数作为参数对象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语句
|
|
|
|
|
log.info(sql.replaceAll("\n", "").replaceAll("\t", ""));
|
|
|
|
|
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);
|
|
|
|
|
//绑定数据
|
|
|
|
|
countStmt = connection.prepareStatement(countSql);
|
|
|
|
|
BoundSql countBS = new BoundSql(mappedStatement.getConfiguration(), countSql,
|
|
|
|
|
boundSql.getParameterMappings(), boundSql.getParameterObject());
|
|
|
|
|
setParameters(countStmt, mappedStatement, countBS, boundSql.getParameterObject());
|
|
|
|
|
//执行
|
|
|
|
|
rs = countStmt.executeQuery();
|
|
|
|
|
if (rs.next()) {
|
|
|
|
|
totalCount = rs.getInt(1);
|
|
|
|
|
@ -112,14 +102,18 @@ public class PageInterceptor implements Interceptor {
|
|
|
|
|
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());
|
|
|
|
|
int totalPage = totalCount/co.getLimit();
|
|
|
|
|
if (totalCount > totalPage*co.getLimit()) {
|
|
|
|
|
totalPage+=1;
|
|
|
|
|
}
|
|
|
|
|
co.setTotlePage(totalPage);
|
|
|
|
|
if (co.getCurrentPageNum() > co.getTotlePage()) {
|
|
|
|
|
co.setCurrentPageNum(co.getTotlePage());
|
|
|
|
|
}
|
|
|
|
|
String pageSql=concatPageMySql(sql,co);
|
|
|
|
|
metaStatementHandler.setValue("delegate.boundSql.sql", pageSql);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@ -135,9 +129,6 @@ public class PageInterceptor implements Interceptor {
|
|
|
|
|
*/
|
|
|
|
|
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")));
|
|
|
|
|
@ -162,7 +153,12 @@ public class PageInterceptor implements Interceptor {
|
|
|
|
|
int index = page.getCurrentPageNum();
|
|
|
|
|
if (index > 1) {
|
|
|
|
|
index = index -1;
|
|
|
|
|
sb.append(" limit ").append(size*index).append(" , ").append(size);
|
|
|
|
|
if (page.getTotlePage() == page.getCurrentPageNum()) {
|
|
|
|
|
sb.append(" limit ").append(size*index).append(" , ").append(page.getTotleSize()-(size*index));
|
|
|
|
|
}
|
|
|
|
|
else {
|
|
|
|
|
sb.append(" limit ").append(size*index).append(" , ").append(size);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
else {
|
|
|
|
|
sb.append(" limit ").append(0).append(" , ").append(size);
|
|
|
|
|
@ -187,8 +183,18 @@ public class PageInterceptor implements Interceptor {
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public void setPageCount(){
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
/**
|
|
|
|
|
* 对SQL参数(?)设值
|
|
|
|
|
* @param ps
|
|
|
|
|
* @param mappedStatement
|
|
|
|
|
* @param boundSql
|
|
|
|
|
* @param parameterObject
|
|
|
|
|
* @throws SQLException
|
|
|
|
|
*/
|
|
|
|
|
private void setParameters(PreparedStatement ps, MappedStatement mappedStatement, BoundSql boundSql,
|
|
|
|
|
Object parameterObject) throws SQLException {
|
|
|
|
|
ParameterHandler parameterHandler = new DefaultParameterHandler(mappedStatement, parameterObject, boundSql);
|
|
|
|
|
parameterHandler.setParameters(ps);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
