一个mybatis分页工具类

来源:互联网 发布:贝克汉姆 知乎 编辑:程序博客网 时间:2024/05/20 09:07

1.PagePlugin类

package com.tyro.plugin;import java.lang.reflect.Field;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 javax.xml.bind.PropertyException;import org.apache.ibatis.executor.ErrorContext;import org.apache.ibatis.executor.ExecutorException;import org.apache.ibatis.executor.statement.BaseStatementHandler;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.mapping.ParameterMode;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.property.PropertyTokenizer;import org.apache.ibatis.scripting.xmltags.ForEachSqlNode;import org.apache.ibatis.session.Configuration;import org.apache.ibatis.type.TypeHandler;import org.apache.ibatis.type.TypeHandlerRegistry;import com.tyro.util.ReflectHelper;import com.tyro.util.Tools;/** * * 类名称:PagePlugin.java* 类描述: * @author fuhang* 作者单位: * 联系方式:* 创建时间:2014年7月1日* @version 1.0 */@Intercepts({@Signature(type=StatementHandler.class,method="prepare",args={Connection.class})})public class PagePlugin implements Interceptor {    private static String dialect = "";    //数据库方言    private static String pageSqlId = ""; //mapper.xml中需要拦截的ID(正则匹配)    public Object intercept(Invocation ivk) throws Throwable {        if(ivk.getTarget() instanceof RoutingStatementHandler){            RoutingStatementHandler statementHandler = (RoutingStatementHandler)ivk.getTarget();            BaseStatementHandler delegate = (BaseStatementHandler) ReflectHelper.getValueByFieldName(statementHandler, "delegate");            MappedStatement mappedStatement = (MappedStatement) ReflectHelper.getValueByFieldName(delegate, "mappedStatement");            if(mappedStatement.getId().matches(pageSqlId)){ //拦截需要分页的SQL                BoundSql boundSql = delegate.getBoundSql();                Object parameterObject = boundSql.getParameterObject();//分页SQL<select>中parameterType属性对应的实体参数,即Mapper接口中执行分页方法的参数,该参数不得为空                if(parameterObject==null){                    throw new NullPointerException("parameterObject尚未实例化!");                }else{                    Connection connection = (Connection) ivk.getArgs()[0];                    String sql = boundSql.getSql();                    //String countSql = "select count(0) from (" + sql+ ") as tmp_count"; //记录统计                    String countSql = "select count(0) from (" + sql+ ")  tmp_count"; //记录统计 == oracle 加 as 报错(SQL command not properly ended)                    PreparedStatement countStmt = connection.prepareStatement(countSql);                    BoundSql countBS = new BoundSql(mappedStatement.getConfiguration(),countSql,boundSql.getParameterMappings(),parameterObject);                    setParameters(countStmt,mappedStatement,countBS,parameterObject);                    ResultSet rs = countStmt.executeQuery();                    int count = 0;                    if (rs.next()) {                        count = rs.getInt(1);                    }                    rs.close();                    countStmt.close();                    //System.out.println(count);                    Page page = null;                    if(parameterObject instanceof Page){    //参数就是Page实体                         page = (Page) parameterObject;                         page.setEntityOrField(true);                             page.setTotalResult(count);                    }else{    //参数为某个实体,该实体拥有Page属性                        Field pageField = ReflectHelper.getFieldByFieldName(parameterObject,"page");                        if(pageField!=null){                            page = (Page) ReflectHelper.getValueByFieldName(parameterObject,"page");                            if(page==null)                                page = new Page();                            page.setEntityOrField(false);                             page.setTotalResult(count);                            ReflectHelper.setValueByFieldName(parameterObject,"page", page); //通过反射,对实体对象设置分页对象                        }else{                            throw new NoSuchFieldException(parameterObject.getClass().getName()+"不存在 page 属性!");                        }                    }                    String pageSql = generatePageSql(sql,page);                    ReflectHelper.setValueByFieldName(boundSql, "sql", pageSql); //将分页sql语句反射回BoundSql.                }            }        }        return ivk.proceed();    }    /**     * 对SQL参数(?)设值,参考org.apache.ibatis.executor.parameter.DefaultParameterHandler     * @param ps     * @param mappedStatement     * @param boundSql     * @param parameterObject     * @throws SQLException     */    @SuppressWarnings({ "rawtypes", "unchecked" })    private void setParameters(PreparedStatement ps,MappedStatement mappedStatement,BoundSql boundSql,Object parameterObject) throws SQLException {        ErrorContext.instance().activity("setting parameters").object(mappedStatement.getParameterMap().getId());        List<ParameterMapping> parameterMappings = boundSql.getParameterMappings();        if (parameterMappings != null) {            Configuration configuration = mappedStatement.getConfiguration();            TypeHandlerRegistry typeHandlerRegistry = configuration.getTypeHandlerRegistry();            MetaObject metaObject = parameterObject == null ? null: configuration.newMetaObject(parameterObject);            for (int i = 0; i < parameterMappings.size(); i++) {                ParameterMapping parameterMapping = parameterMappings.get(i);                if (parameterMapping.getMode() != ParameterMode.OUT) {                    Object value;                    String propertyName = parameterMapping.getProperty();                    PropertyTokenizer prop = new PropertyTokenizer(propertyName);                    if (parameterObject == null) {                        value = null;                    } else if (typeHandlerRegistry.hasTypeHandler(parameterObject.getClass())) {                        value = parameterObject;                    } else if (boundSql.hasAdditionalParameter(propertyName)) {                        value = boundSql.getAdditionalParameter(propertyName);                    } else if (propertyName.startsWith(ForEachSqlNode.ITEM_PREFIX)&& boundSql.hasAdditionalParameter(prop.getName())) {                        value = boundSql.getAdditionalParameter(prop.getName());                        if (value != null) {                            value = configuration.newMetaObject(value).getValue(propertyName.substring(prop.getName().length()));                        }                    } else {                        value = metaObject == null ? null : metaObject.getValue(propertyName);                    }                    TypeHandler typeHandler = parameterMapping.getTypeHandler();                    if (typeHandler == null) {                        throw new ExecutorException("There was no TypeHandler found for parameter "+ propertyName + " of statement "+ mappedStatement.getId());                    }                    typeHandler.setParameter(ps, i + 1, value, parameterMapping.getJdbcType());                }            }        }    }    /**     * 根据数据库方言,生成特定的分页sql     * @param sql     * @param page     * @return     */    private String generatePageSql(String sql,Page page){        if(page!=null && Tools.notEmpty(dialect)){            StringBuffer pageSql = new StringBuffer();            if("mysql".equals(dialect)){                pageSql.append(sql);                pageSql.append(" limit "+page.getCurrentResult()+","+page.getShowCount());            }else if("oracle".equals(dialect)){                pageSql.append("select * from (select tmp_tb.*,ROWNUM row_id from (");                pageSql.append(sql);                //pageSql.append(") as tmp_tb where ROWNUM<=");                pageSql.append(") tmp_tb where ROWNUM<=");                pageSql.append(page.getCurrentResult()+page.getShowCount());                pageSql.append(") where row_id>");                pageSql.append(page.getCurrentResult());            }            return pageSql.toString();        }else{            return sql;        }    }    public Object plugin(Object arg0) {        return Plugin.wrap(arg0, this);    }    public void setProperties(Properties p) {        dialect = p.getProperty("dialect");        if (Tools.isEmpty(dialect)) {            try {                throw new PropertyException("dialect property is not found!");            } catch (PropertyException e) {                e.printStackTrace();            }        }        pageSqlId = p.getProperty("pageSqlId");        if (Tools.isEmpty(pageSqlId)) {            try {                throw new PropertyException("pageSqlId property is not found!");            } catch (PropertyException e) {                e.printStackTrace();            }        }    }}

2.Page类

package com.tyro.plugin;import com.tyro.util.Const;import com.tyro.util.ParameterMap;import com.tyro.util.Tools;public class Page {    private int showCount; //每页显示记录数    private int totalPage;        //总页数    private int totalResult;    //总记录数    private int currentPage;    //当前页    private int currentResult;    //当前记录起始索引    private boolean entityOrField;    //true:需要分页的地方,传入的参数就是Page实体;false:需要分页的地方,传入的参数所代表的实体拥有Page属性    private String pageStr;        //最终页面显示的底部翻页导航,详细见:getPageStr();    private ParameterMap pd = new ParameterMap();    public Page(){        try {            this.showCount = Integer.parseInt(Tools.readTxtFile(Const.PAGE));        } catch (Exception e) {            this.showCount = 15;        }    }    public int getTotalPage() {        if(totalResult%showCount==0)            totalPage = totalResult/showCount;        else            totalPage = totalResult/showCount+1;        return totalPage;    }    public void setTotalPage(int totalPage) {        this.totalPage = totalPage;    }    public int getTotalResult() {        return totalResult;    }    public void setTotalResult(int totalResult) {        this.totalResult = totalResult;    }    public int getCurrentPage() {        if(currentPage<=0)            currentPage = 1;        return currentPage;    }    public void setCurrentPage(int currentPage) {        this.currentPage = currentPage;    }    public String getPageStr() {        StringBuffer sb = new StringBuffer();        if(totalResult>0){            sb.append("    <ul>\n");            if(currentPage==1){                sb.append("    <li><a>共<font color=red>"+totalResult+"</font>条</a></li>\n");                sb.append("    <li><input type=\"number\" value=\"\" id=\"toGoPage\" style=\"width:50px;text-align:center;float:left\" placeholder=\"页码\"/></li>\n");                sb.append("    <li style=\"cursor:pointer;\"><a onclick=\"toTZ();\"  class=\"btn btn-mini btn-success\">跳转</a></li>\n");                sb.append("    <li><a>首页</a></li>\n");                sb.append("    <li><a>上页</a></li>\n");            }else{                sb.append("    <li><a>共<font color=red>"+totalResult+"</font>条</a></li>\n");                sb.append("    <li><input type=\"number\" value=\"\" id=\"toGoPage\" style=\"width:50px;text-align:center;float:left\" placeholder=\"页码\"/></li>\n");                sb.append("    <li style=\"cursor:pointer;\"><a onclick=\"toTZ();\"  class=\"btn btn-mini btn-success\">跳转</a></li>\n");                sb.append("    <li style=\"cursor:pointer;\"><a onclick=\"nextPage(1)\">首页</a></li>\n");                sb.append("    <li style=\"cursor:pointer;\"><a onclick=\"nextPage("+(currentPage-1)+")\">上页</a></li>\n");            }            int showTag = 5;//分页标签显示数量            int startTag = 1;            if(currentPage>showTag){                startTag = currentPage-1;            }            int endTag = startTag+showTag-1;            for(int i=startTag; i<=totalPage && i<=endTag; i++){                if(currentPage==i)                    sb.append("<li><a><font color='#808080'>"+i+"</font></a></li>\n");                else                    sb.append("    <li style=\"cursor:pointer;\"><a onclick=\"nextPage("+i+")\">"+i+"</a></li>\n");            }            if(currentPage==totalPage){                sb.append("    <li><a>下页</a></li>\n");                sb.append("    <li><a>尾页</a></li>\n");            }else{                sb.append("    <li style=\"cursor:pointer;\"><a onclick=\"nextPage("+(currentPage+1)+")\">下页</a></li>\n");                sb.append("    <li style=\"cursor:pointer;\"><a onclick=\"nextPage("+totalPage+")\">尾页</a></li>\n");            }            sb.append("    <li><a>第"+currentPage+"页</a></li>\n");            sb.append("    <li><a>共"+totalPage+"页</a></li>\n");            sb.append("    <li><select title='显示条数' style=\"width:55px;float:left;\" onchange=\"changeCount(this.value)\">\n");            sb.append("    <option value='"+showCount+"'>"+showCount+"</option>\n");            sb.append("    <option value='10'>10</option>\n");            sb.append("    <option value='20'>20</option>\n");            sb.append("    <option value='30'>30</option>\n");            sb.append("    <option value='40'>40</option>\n");            sb.append("    <option value='50'>50</option>\n");            sb.append("    <option value='60'>60</option>\n");            sb.append("    <option value='70'>70</option>\n");            sb.append("    <option value='80'>80</option>\n");            sb.append("    <option value='90'>90</option>\n");            sb.append("    <option value='99'>99</option>\n");            sb.append("    </select>\n");            sb.append("    </li>\n");            sb.append("</ul>\n");            sb.append("<script type=\"text/javascript\">\n");            //换页函数            sb.append("function nextPage(page){");            sb.append(" top.jzts();");            sb.append("    if(true && document.forms[0]){\n");            sb.append("        var url = document.forms[0].getAttribute(\"action\");\n");            sb.append("        if(url.indexOf('?')>-1){url += \"&"+(entityOrField?"currentPage":"page.currentPage")+"=\";}\n");            sb.append("        else{url += \"?"+(entityOrField?"currentPage":"page.currentPage")+"=\";}\n");            sb.append("        url = url + page + \"&" +(entityOrField?"showCount":"page.showCount")+"="+showCount+"\";\n");            sb.append("        document.forms[0].action = url;\n");            sb.append("        document.forms[0].submit();\n");            sb.append("    }else{\n");            sb.append("        var url = document.location+'';\n");            sb.append("        if(url.indexOf('?')>-1){\n");            sb.append("            if(url.indexOf('currentPage')>-1){\n");            sb.append("                var reg = /currentPage=\\d*/g;\n");            sb.append("                url = url.replace(reg,'currentPage=');\n");            sb.append("            }else{\n");            sb.append("                url += \"&"+(entityOrField?"currentPage":"page.currentPage")+"=\";\n");            sb.append("            }\n");            sb.append("        }else{url += \"?"+(entityOrField?"currentPage":"page.currentPage")+"=\";}\n");            sb.append("        url = url + page + \"&" +(entityOrField?"showCount":"page.showCount")+"="+showCount+"\";\n");            sb.append("        document.location = url;\n");            sb.append("    }\n");            sb.append("}\n");            //调整每页显示条数            sb.append("function changeCount(value){");            sb.append(" top.jzts();");            sb.append("    if(true && document.forms[0]){\n");            sb.append("        var url = document.forms[0].getAttribute(\"action\");\n");            sb.append("        if(url.indexOf('?')>-1){url += \"&"+(entityOrField?"currentPage":"page.currentPage")+"=\";}\n");            sb.append("        else{url += \"?"+(entityOrField?"currentPage":"page.currentPage")+"=\";}\n");            sb.append("        url = url + \"1&" +(entityOrField?"showCount":"page.showCount")+"=\"+value;\n");            sb.append("        document.forms[0].action = url;\n");            sb.append("        document.forms[0].submit();\n");            sb.append("    }else{\n");            sb.append("        var url = document.location+'';\n");            sb.append("        if(url.indexOf('?')>-1){\n");            sb.append("            if(url.indexOf('currentPage')>-1){\n");            sb.append("                var reg = /currentPage=\\d*/g;\n");            sb.append("                url = url.replace(reg,'currentPage=');\n");            sb.append("            }else{\n");            sb.append("                url += \"1&"+(entityOrField?"currentPage":"page.currentPage")+"=\";\n");            sb.append("            }\n");            sb.append("        }else{url += \"?"+(entityOrField?"currentPage":"page.currentPage")+"=\";}\n");            sb.append("        url = url + \"&" +(entityOrField?"showCount":"page.showCount")+"=\"+value;\n");            sb.append("        document.location = url;\n");            sb.append("    }\n");            sb.append("}\n");            //跳转函数             sb.append("function toTZ(){");            sb.append("var toPaggeVlue = document.getElementById(\"toGoPage\").value;");            sb.append("if(toPaggeVlue == ''){document.getElementById(\"toGoPage\").value=1;return;}");            sb.append("if(isNaN(Number(toPaggeVlue))){document.getElementById(\"toGoPage\").value=1;return;}");            sb.append("nextPage(toPaggeVlue);");            sb.append("}\n");            sb.append("</script>\n");        }        pageStr = sb.toString();        return pageStr;    }    public void setPageStr(String pageStr) {        this.pageStr = pageStr;    }    public int getShowCount() {        return showCount;    }    public void setShowCount(int showCount) {        this.showCount = showCount;    }    public int getCurrentResult() {        currentResult = (getCurrentPage()-1)*getShowCount();        if(currentResult<0)            currentResult = 0;        return currentResult;    }    public void setCurrentResult(int currentResult) {        this.currentResult = currentResult;    }    public boolean isEntityOrField() {        return entityOrField;    }    public void setEntityOrField(boolean entityOrField) {        this.entityOrField = entityOrField;    }    public ParameterMap getPm() {        return pd;    }    public void setPm(ParameterMap pd) {        this.pd = pd;    }}

3.ReflectHelper类

package com.tyro.util;import java.lang.reflect.Field;/** * @author Administrator *    反射工具 *  分页获取到 */public class ReflectHelper {    /**     * 获取obj对象fieldName的Field     * @param obj     * @param fieldName     * @return     */    public static Field getFieldByFieldName(Object obj, String fieldName) {        for (Class<?> superClass = obj.getClass(); superClass != Object.class; superClass = superClass                .getSuperclass()) {            try {                return superClass.getDeclaredField(fieldName);            } catch (NoSuchFieldException e) {            }        }        return null;    }    /**     * 获取obj对象fieldName的属性值     * @param obj     * @param fieldName     * @return     * @throws SecurityException     * @throws NoSuchFieldException     * @throws IllegalArgumentException     * @throws IllegalAccessException     */    public static Object getValueByFieldName(Object obj, String fieldName)            throws SecurityException, NoSuchFieldException,            IllegalArgumentException, IllegalAccessException {        Field field = getFieldByFieldName(obj, fieldName);        Object value = null;        if(field!=null){            if (field.isAccessible()) {                value = field.get(obj);            } else {                field.setAccessible(true);                value = field.get(obj);                field.setAccessible(false);            }        }        return value;    }    /**     * 设置obj对象fieldName的属性值     * @param obj     * @param fieldName     * @param value     * @throws SecurityException     * @throws NoSuchFieldException     * @throws IllegalArgumentException     * @throws IllegalAccessException     */    public static void setValueByFieldName(Object obj, String fieldName,            Object value) throws SecurityException, NoSuchFieldException,            IllegalArgumentException, IllegalAccessException {        Field field = obj.getClass().getDeclaredField(fieldName);        if (field.isAccessible()) {            field.set(obj, value);        } else {            field.setAccessible(true);            field.set(obj, value);            field.setAccessible(false);        }    }}

4.Tool类

package com.tyro.util;import java.io.BufferedReader;import java.io.BufferedWriter;import java.io.File;import java.io.FileInputStream;import java.io.FileOutputStream;import java.io.IOException;import java.io.InputStreamReader;import java.io.OutputStreamWriter;import java.text.ParseException;import java.text.SimpleDateFormat;import java.util.Date;import java.util.Random;import java.util.regex.Matcher;import java.util.regex.Pattern;public class Tools {    /**     * 随机生成六位数验证码      * @return     */    public static int getRandomNum(){         Random r = new Random();         return r.nextInt(900000)+100000;//(Math.random()*(999999-100000)+100000)    }    /**     * 检测字符串是否不为空(null,"","null")     * @param s     * @return 不为空则返回true,否则返回false     */    public static boolean notEmpty(String s){        return s!=null && !"".equals(s) && !"null".equals(s);    }    /**     * 检测字符串是否为空(null,"","null")     * @param s     * @return 为空则返回true,不否则返回false     */    public static boolean isEmpty(String s){        return s==null || "".equals(s) || "null".equals(s);    }    /**     * 写txt里的单行内容     * @param filePath  文件路径     * @param content  写入的内容     */    public static void writeFile(String fileP,String content){        String filePath = String.valueOf(Thread.currentThread().getContextClassLoader().getResource(""))+"../../";    //项目路径        filePath = (filePath.trim() + fileP.trim()).substring(6).trim();        if(filePath.indexOf(":") != 1){            filePath = File.separator + filePath;        }        try {            OutputStreamWriter write = new OutputStreamWriter(new FileOutputStream(filePath),"utf-8");                  BufferedWriter writer=new BufferedWriter(write);                      writer.write(content);                  writer.close();         } catch (IOException e) {            e.printStackTrace();        }    }    /**      * 验证邮箱      * @param email      * @return      */     public static boolean checkEmail(String email){      boolean flag = false;      try{        String check = "^([a-z0-9A-Z]+[-|_|\\.]?)+[a-z0-9A-Z]@([a-z0-9A-Z]+(-[a-z0-9A-Z]+)?\\.)+[a-zA-Z]{2,}$";        Pattern regex = Pattern.compile(check);        Matcher matcher = regex.matcher(email);        flag = matcher.matches();       }catch(Exception e){        flag = false;       }      return flag;     }     /**      * 验证手机号码      * @param mobiles      * @return      */     public static boolean checkMobileNumber(String mobileNumber){      boolean flag = false;      try{        Pattern regex = Pattern.compile("^(((13[0-9])|(15([0-3]|[5-9]))|(18[0-9]))\\d{8})|(0\\d{2}-\\d{8})|(0\\d{3}-\\d{7})$");        Matcher matcher = regex.matcher(mobileNumber);        flag = matcher.matches();       }catch(Exception e){        flag = false;       }      return flag;     }    /**     * 读取txt里的单行内容     * @param filePath  文件路径     */    public static String readTxtFile(String fileP) {        try {//            String filePath = String.valueOf(Thread.currentThread().getContextClassLoader().getResource(""))+"../../";    //项目路径            String filePath = String.valueOf(Thread.currentThread().getContextClassLoader().getResource(""));    //项目路径            filePath = filePath.replaceAll("file:/", "");            filePath = filePath.replaceAll("%20", " ");            filePath = filePath.trim() + fileP.trim();            if(filePath.indexOf(":") != 1){                filePath = File.separator + filePath;            }            String encoding = "utf-8";            System.out.println("filePath:"+filePath);            File file = new File(filePath);            if (file.isFile() && file.exists()) {         // 判断文件是否存在                InputStreamReader read = new InputStreamReader(                new FileInputStream(file), encoding);    // 考虑到编码格式                BufferedReader bufferedReader = new BufferedReader(read);                String lineTxt = null;                while ((lineTxt = bufferedReader.readLine()) != null) {                    return lineTxt;                }                read.close();            }else{                System.out.println("找不到指定的文件,查看此路径是否正确:"+filePath);            }        } catch (Exception e) {            System.out.println("读取文件内容出错");        }        return "";    }}

留着笔记,码农专用。。。。。。。。。。

原创粉丝点击