spring data JPA 高级动态查询, 排序加分页加多参

业务需求是根据前端传来的四个参数进行匹配查询, 并且要将总共查出来的数据总数返回至前端(因为前端有进行分页, 首尾页等), 但是前端未必会传来四个参数, 也可能是随机的其中一个,两个等. 这时候如果自己在dao层,手写查询方法或者sql语句需要大量代码, 并且非常麻烦, 在多方查阅资料,以及询问公司前辈的情况下, 终于完成了项目需求,

贴出前端查询页面:

---

贴出二元组代码:

package com.leadmap.mapservice.common;

public class TwoTuple<A, B> {

  public final A first;

    private final B second;

    public TwoTuple(A a, B b){

        first = a;

        second = b;

    }

    public A getFirst() {

        return first;

    }

    public B getSecond(){

        return second;

    }

    @Override

    public String toString(){

        return "(" + first + ", " + second + ")";

    }

}

---

贴出查询业务代码:

package com.leadmap.mapservice.service;

import com.leadmap.mapservice.common.TwoTuple;

import com.leadmap.mapservice.common.Util;

import com.leadmap.mapservice.dao.DocumentInfoDao;

import com.leadmap.mapservice.dao.OpinionFeedbackDao;

import com.leadmap.mapservice.entity.DocumentInfo;

import com.leadmap.mapservice.entity.OpinionFeedback;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.data.domain.PageRequest;

import org.springframework.data.domain.Pageable;

import org.springframework.data.jpa.domain.Specification;

import org.springframework.stereotype.Service;

import javax.persistence.criteria.CriteriaBuilder;

import javax.persistence.criteria.CriteriaQuery;

import javax.persistence.criteria.Predicate;

import javax.persistence.criteria.Root;

import java.util.ArrayList;

import java.util.Date;

import java.util.List;

@Service

public class DocumentService {

    @Autowired

    private DocumentInfoDao documentInfoDao;

    // 返回结果为实体类对象数组和匹配数据总数

    // 两个日期参数格式为 yyyy-MM-dd  数据库中对应字段格式为yyyy-mm-dd hh-mm-ss 因此下面会进行日期字符串拼接

    public TwoTuple<List<DocumentInfo>, Integer> getPageDocumentInfo(int start, int count, String beginTime, String endTime, String type, String title){

        // 创建分页参数

        Pageable page = new PageRequest(start, count);

        List<DocumentInfo> list = new ArrayList<>();

        // 新建查询参数, 重写方法

        Specification querySpecifi = new Specification<DocumentInfo>() {

            // 这几个参数算是固定用法吧 要用到就写上去

            @Override

            public Predicate toPredicate(Root root, CriteriaQuery criteriaQuery, CriteriaBuilder criteriaBuilder) {

                 // 姑且叫做 新建查询参数数组吧, 是多个查询条件组合的的数组

                List<Predicate> predicatesList = new ArrayList<>();

                // 当if成立 添加 匹配大于开始时间的查询参数

                if(!beginTime.isEmpty()){

                    Date beginDate = Util.StrToDate(beginTime+" 00:00:00");

                    predicatesList.add(criteriaBuilder.greaterThan(root.get("createTime"), beginDate));

                }

                // 同上 添加 匹配小于开始时间的查询参数  这两个参数组合也就相当于sql中的between...and...

                if(!endTime.isEmpty()){

                    Date endDate = Util.StrToDate(endTime + " 00:00:00");

                    predicatesList.add(criteriaBuilder.lessThan(root.get("createTime"), endDate));

                }

                // 添加匹配数据库中相同类型的参数

                if(!type.isEmpty()){

                    predicatesList.add(criteriaBuilder.like(root.get("type"),"%"+type+"%"));

                }

                if(!title.isEmpty()){

                    predicatesList.add(criteriaBuilder.like(root.get("title"), "%" + title + "%"));

                }

                // 设置排序条件 根据id倒序

                criteriaQuery.orderBy(criteriaBuilder.desc(root.get("id")));

                // 将排序与添加好的查询参数数组 作为返回值

                criteriaQuery.where(criteriaBuilder.and(predicatesList.toArray(new Predicate[predicatesList.size()])));

                return criteriaQuery.getRestriction();

            }

        };

        // 这个findAll是JpaSpecificationExecutor<T>接口的方法

        list = documentInfoDao.findAll(querySpecifi,page).getContent();

        // 拿到 匹配数据总数 此处以每页十条分页

        Integer counts = documentInfoDao.findAll(querySpecifi).size() / 10;

        // 返回二元组

        return new TwoTuple<List<DocumentInfo>, Integer>(list, counts);

    }

}