springboot学习（三）--jpa整合之处理复杂sql封装

简单粗暴，上代码，哈哈：

@Component
public class SimpleBaseRepository implements BaseRepository {

    //获取hibernate中的核心接口，hibernate中所有的数据库操作都是通过这个接口中的方法来实现的
    @Autowired
    EntityManager entityManager;

    /**
     * @param sql   查询语句
     * @param params    查询参数
     * @param pageable  分页
     * @return
     */
    @Override
    public Page<Map<String, Object>> getListMapObject(String sql, Object[] params, Pageable pageable) {

        //这个是查询的Query
        Query query = entityManager.createNativeQuery(sql);
        int i = 0;
        if(params != null) {
            for (Object param : params) {
                query.setParameter(i++, param);
            }
        }
        //query查询
        org.hibernate.Query unwrapQuery = query.unwrap(org.hibernate.Query.class);
        unwrapQuery.setResultTransformer(Criteria.ALIAS_TO_ENTITY_MAP)
                .setFirstResult(pageable.getPageNumber() * pageable.getPageSize()).setMaxResults(pageable.getPageSize());

        //获取结果，数据类型是一个map集合
        List<Map<String,Object>> resultData = unwrapQuery.list();

        //这一步是仿照jpa源码中的想法来实现的，最终获取一个PageImpl的类
        Page<?> page = getPage(resultData, pageable, ()->{
            return executeCountQuery(getCountQuerySql(sql), params);
        });

        return (PageImpl) page;
    }


    public Page<?> getPage(List<?> content, Pageable pageable, LongSupplier totalLongSupplier){
        //其实hibernate底层的求总条数和集合的思想是这样的
        if(pageable.getOffset() != 0L){
            //如果查的不是第一页的数据中的条数小于 PageSize那么就不要求这个查询的总条数了
            //直接取偏移量+这页查询到的数据条数 = 总条数
            //否则
            //说明下一页还有其它数据，这个时候我们要通过 select count(*) 来查询总条数了
            return ((content.size() != 0) && (pageable.getPageSize() > content.size())) ? new PageImpl(content,pageable,pageable.getOffset()+content.size()) :
                    new PageImpl(content,pageable,totalLongSupplier.getAsLong());
        }else{
            //如果查的是第一页，如果查询到的数据条数小于PageSize，那么说明这个 总条数 = 查询到的数据条数
            //直接取第一页的数据中的条数作为总条数 总条数 = 查询到的数据条数
            //否则
            //说明下一页还有其它数据，这个时候我们要通过 select count(*) 来查询总条数了
            return pageable.getPageSize() <= content.size() ? new PageImpl(content,pageable,totalLongSupplier.getAsLong()) :
                    new PageImpl(content,pageable,content.size());
        }
    }

    private String getCountQuerySql(String sql){
        String sqlStr = String.format("select count(*) from (%s) c",sql);
        return sqlStr;
    }

    //求总条数
    private long executeCountQuery(String sql, Object[] params){
        Query query = entityManager.createNativeQuery(sql);
        int i = 0;
        if(params != null) {
            for (Object param : params) {
                query.setParameter(i++, param);
            }
        }
        List resultList = query.getResultList();
        return Long.parseLong(resultList.get(0).toString());
    }
}

调用如下：

String sql = "SELECT u.*,g.count from user u LEFT JOIN usergrade g ON u.id = g.uid";
Page<Map<String, Object>> pageResult = simpleBaseRepository.getListMapObject(sql, null, PageRequest.of(1, 200000));
System.out.println(pageResult.getTotalElements());