自定义pageHelper方言

自定义pageHelper方言

最近遇到一个业务问题：我们的数据要去impala上去查找，但是呢，impala不支持pageHelper的分页，怎么办？，那么解决办法有两种：

一、原生分页

自己写PageUtils 。前面有讲过，这里不是重点，不赘述。

二、修改pageHelper的方言，让它支持impala的分页。

在pageHelper底层注册支持15种数据库的分页（PageAutoDialect），这里需要手动写一个impala 的方言，注册到pageHelper中就可以了。

package com.sinux.cc.utils;

import com.github.pagehelper.Page;
import com.github.pagehelper.dialect.AbstractHelperDialect;
import com.github.pagehelper.page.PageAutoDialect;
import com.github.pagehelper.util.MetaObjectUtil;
import org.apache.ibatis.cache.CacheKey;
import org.apache.ibatis.mapping.BoundSql;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.ParameterMapping;
import org.apache.ibatis.reflection.MetaObject;
import org.springframework.stereotype.Component;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
@Component
public class ImpalaDialect extends AbstractHelperDialect {
    /**
     * 注册ImpalaDialect的方言到PageAutoDialect
     */
    public ImpalaDialect(){
        PageAutoDialect.registerDialectAlias("impala",ImpalaDialect.class);
    }
    @Override
    public Object processPageParameter(MappedStatement mappedStatement, Map<String, Object> map, Page page, BoundSql boundSql, CacheKey cacheKey) {
        /**
         * 起始行 和每页条数
         */
        map.put("First_PageHelper",page.getStartRow());
        map.put("Second_PageHelper",page.getPageSize());

        cacheKey.update(page.getStartRow());
        cacheKey.update(page.getPageSize());

        if(boundSql.getParameterMappings()!=null){
            List<ParameterMapping> list = new ArrayList<>();
            if(boundSql != null && boundSql.getParameterMappings() != null){
                list.addAll(boundSql.getParameterMappings());
            }
            if(page.getStartRow() == 0){
                list.add((new ParameterMapping.Builder(mappedStatement.getConfiguration(),"Second_PageHelper",Integer.class)).build());
            }else {
                list.add((new ParameterMapping.Builder(mappedStatement.getConfiguration(),"First_PageHelper",Integer.class)).build());
                list.add((new ParameterMapping.Builder(mappedStatement.getConfiguration(),"Second_PageHelper",Integer.class)).build());
            }
            MetaObject metaObject = MetaObjectUtil.forObject(boundSql);
            metaObject.setValue("parameterMappings",list);
        }
        return map;
    }
    @Override
    public String getPageSql(String sql, Page page, CacheKey cacheKey) {
        if(!sql.toUpperCase().contains("ORDER BY")){
            throw new RuntimeException("impala需要排序后才能分页！");
        }
        StringBuilder sb = new StringBuilder(sql.length()+20);
        sb.append(sql);
        if(page.getStartRow() == 0){
            sb.append(" LIMIT ? ");
        }else {
            sb.append(" LIMIT ? OFFSET ? ");
        }
        cacheKey.update(page.getPageSize());
        return sb.toString();
    }
}

这样就可以在代码中使用pageHelper分页了。