业务需求是根据前端传来的四个参数进行匹配查询, 并且要将总共查出来的数据总数返回至前端(因为前端有进行分页, 首尾页等), 但是前端未必会传来四个参数, 也可能是随机的其中一个,两个等. 这时候如果自己在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);
}
}
